xref: /petsc/src/mat/impls/sell/seq/sell.c (revision faa75363f7f893b00eea36da2ccba8c9e6150d98)
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>
85f70456aSHong 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)
94243e2ceSHong Zhang 
10d4002b98SHong Zhang   #include <immintrin.h>
11d4002b98SHong Zhang 
12d4002b98SHong Zhang   #if !defined(_MM_SCALE_8)
13d4002b98SHong Zhang   #define _MM_SCALE_8    8
14d4002b98SHong Zhang   #endif
15d4002b98SHong Zhang 
16d4002b98SHong Zhang   #if defined(__AVX512F__)
17d4002b98SHong Zhang   /* these do not work
18d4002b98SHong Zhang    vec_idx  = _mm512_loadunpackhi_epi32(vec_idx,acolidx);
19d4002b98SHong Zhang    vec_vals = _mm512_loadunpackhi_pd(vec_vals,aval);
20d4002b98SHong Zhang   */
21d4002b98SHong Zhang     #define AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y) \
22d4002b98SHong Zhang     /* if the mask bit is set, copy from acolidx, otherwise from vec_idx */ \
23ef588d5cSRichard Tran Mills     vec_idx  = _mm256_loadu_si256((__m256i const*)acolidx); \
24ef588d5cSRichard Tran Mills     vec_vals = _mm512_loadu_pd(aval); \
25d4002b98SHong Zhang     vec_x    = _mm512_i32gather_pd(vec_idx,x,_MM_SCALE_8); \
26a48a6482SHong Zhang     vec_y    = _mm512_fmadd_pd(vec_x,vec_vals,vec_y)
275f70456aSHong Zhang   #elif defined(__AVX2__) && defined(__FMA__)
28a48a6482SHong Zhang     #define AVX2_Mult_Private(vec_idx,vec_x,vec_vals,vec_y) \
29ef588d5cSRichard Tran Mills     vec_vals = _mm256_loadu_pd(aval); \
30ef588d5cSRichard Tran Mills     vec_idx  = _mm_loadu_si128((__m128i const*)acolidx); /* SSE2 */ \
31a48a6482SHong Zhang     vec_x    = _mm256_i32gather_pd(x,vec_idx,_MM_SCALE_8); \
32a48a6482SHong Zhang     vec_y    = _mm256_fmadd_pd(vec_x,vec_vals,vec_y)
33d4002b98SHong Zhang   #endif
34d4002b98SHong Zhang #endif  /* PETSC_HAVE_IMMINTRIN_H */
35d4002b98SHong Zhang 
36d4002b98SHong Zhang /*@C
37d4002b98SHong Zhang  MatSeqSELLSetPreallocation - For good matrix assembly performance
38d4002b98SHong Zhang  the user should preallocate the matrix storage by setting the parameter nz
39d4002b98SHong Zhang  (or the array nnz).  By setting these parameters accurately, performance
40d4002b98SHong Zhang  during matrix assembly can be increased significantly.
41d4002b98SHong Zhang 
42d083f849SBarry Smith  Collective
43d4002b98SHong Zhang 
44d4002b98SHong Zhang  Input Parameters:
45d4002b98SHong Zhang  +  B - The matrix
46d4002b98SHong Zhang  .  nz - number of nonzeros per row (same for all rows)
47d4002b98SHong Zhang  -  nnz - array containing the number of nonzeros in the various rows
48d4002b98SHong Zhang  (possibly different for each row) or NULL
49d4002b98SHong Zhang 
50d4002b98SHong Zhang  Notes:
51d4002b98SHong Zhang  If nnz is given then nz is ignored.
52d4002b98SHong Zhang 
53d4002b98SHong Zhang  Specify the preallocated storage with either nz or nnz (not both).
54d4002b98SHong Zhang  Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory
55d4002b98SHong Zhang  allocation.  For large problems you MUST preallocate memory or you
56d4002b98SHong Zhang  will get TERRIBLE performance, see the users' manual chapter on matrices.
57d4002b98SHong Zhang 
58d4002b98SHong Zhang  You can call MatGetInfo() to get information on how effective the preallocation was;
59d4002b98SHong Zhang  for example the fields mallocs,nz_allocated,nz_used,nz_unneeded;
60d4002b98SHong Zhang  You can also run with the option -info and look for messages with the string
61d4002b98SHong Zhang  malloc in them to see if additional memory allocation was needed.
62d4002b98SHong Zhang 
63d4002b98SHong Zhang  Developers: Use nz of MAT_SKIP_ALLOCATION to not allocate any space for the matrix
64d4002b98SHong Zhang  entries or columns indices.
65d4002b98SHong Zhang 
66c7ee91abSRichard Tran Mills  The maximum number of nonzeos in any row should be as accurate as possible.
67c7ee91abSRichard Tran Mills  If it is underestimated, you will get bad performance due to reallocation
68d4002b98SHong Zhang  (MatSeqXSELLReallocateSELL).
69d4002b98SHong Zhang 
70d4002b98SHong Zhang  Level: intermediate
71d4002b98SHong Zhang 
72d4002b98SHong Zhang  .seealso: MatCreate(), MatCreateSELL(), MatSetValues(), MatGetInfo()
73d4002b98SHong Zhang 
74d4002b98SHong Zhang  @*/
75d4002b98SHong Zhang PetscErrorCode MatSeqSELLSetPreallocation(Mat B,PetscInt rlenmax,const PetscInt rlen[])
76d4002b98SHong Zhang {
77d4002b98SHong Zhang   PetscErrorCode ierr;
78d4002b98SHong Zhang 
79d4002b98SHong Zhang   PetscFunctionBegin;
80d4002b98SHong Zhang   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
81d4002b98SHong Zhang   PetscValidType(B,1);
82d4002b98SHong Zhang   ierr = PetscTryMethod(B,"MatSeqSELLSetPreallocation_C",(Mat,PetscInt,const PetscInt[]),(B,rlenmax,rlen));CHKERRQ(ierr);
83d4002b98SHong Zhang   PetscFunctionReturn(0);
84d4002b98SHong Zhang }
85d4002b98SHong Zhang 
86d4002b98SHong Zhang PetscErrorCode MatSeqSELLSetPreallocation_SeqSELL(Mat B,PetscInt maxallocrow,const PetscInt rlen[])
87d4002b98SHong Zhang {
88d4002b98SHong Zhang   Mat_SeqSELL    *b;
89d4002b98SHong Zhang   PetscInt       i,j,totalslices;
90d4002b98SHong Zhang   PetscBool      skipallocation=PETSC_FALSE,realalloc=PETSC_FALSE;
91d4002b98SHong Zhang   PetscErrorCode ierr;
92d4002b98SHong Zhang 
93d4002b98SHong Zhang   PetscFunctionBegin;
94d4002b98SHong Zhang   if (maxallocrow >= 0 || rlen) realalloc = PETSC_TRUE;
95d4002b98SHong Zhang   if (maxallocrow == MAT_SKIP_ALLOCATION) {
96d4002b98SHong Zhang     skipallocation = PETSC_TRUE;
97d4002b98SHong Zhang     maxallocrow    = 0;
98d4002b98SHong Zhang   }
99d4002b98SHong Zhang 
100d4002b98SHong Zhang   ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr);
101d4002b98SHong Zhang   ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr);
102d4002b98SHong Zhang 
103d4002b98SHong Zhang   /* FIXME: if one preallocates more space than needed, the matrix does not shrink automatically, but for best performance it should */
104d4002b98SHong Zhang   if (maxallocrow == PETSC_DEFAULT || maxallocrow == PETSC_DECIDE) maxallocrow = 5;
105c0aa6a63SJacob Faibussowitsch   if (maxallocrow < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"maxallocrow cannot be less than 0: value %" PetscInt_FMT,maxallocrow);
106d4002b98SHong Zhang   if (rlen) {
107d4002b98SHong Zhang     for (i=0; i<B->rmap->n; i++) {
108c0aa6a63SJacob Faibussowitsch       if (rlen[i] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"rlen cannot be less than 0: local row %" PetscInt_FMT " value %" PetscInt_FMT,i,rlen[i]);
109c0aa6a63SJacob Faibussowitsch       if (rlen[i] > B->cmap->n) SETERRQ3(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);
110d4002b98SHong Zhang     }
111d4002b98SHong Zhang   }
112d4002b98SHong Zhang 
113d4002b98SHong Zhang   B->preallocated = PETSC_TRUE;
114d4002b98SHong Zhang 
115d4002b98SHong Zhang   b = (Mat_SeqSELL*)B->data;
116d4002b98SHong Zhang 
117*faa75363SBarry Smith   totalslices = PetscCeilInt(B->rmap->n,8);
118d4002b98SHong Zhang   b->totalslices = totalslices;
119d4002b98SHong Zhang   if (!skipallocation) {
120c0aa6a63SJacob Faibussowitsch     if (B->rmap->n & 0x07) {ierr = PetscInfo1(B,"Padding rows to the SEQSELL matrix because the number of rows is not the multiple of 8 (value %" PetscInt_FMT ")\n",B->rmap->n);CHKERRQ(ierr);}
121d4002b98SHong Zhang 
122d4002b98SHong Zhang     if (!b->sliidx) { /* sliidx gives the starting index of each slice, the last element is the total space allocated */
123d4002b98SHong Zhang       ierr = PetscMalloc1(totalslices+1,&b->sliidx);CHKERRQ(ierr);
124d4002b98SHong Zhang       ierr = PetscLogObjectMemory((PetscObject)B,(totalslices+1)*sizeof(PetscInt));CHKERRQ(ierr);
125d4002b98SHong Zhang     }
126d4002b98SHong Zhang     if (!rlen) { /* if rlen is not provided, allocate same space for all the slices */
127d4002b98SHong Zhang       if (maxallocrow == PETSC_DEFAULT || maxallocrow == PETSC_DECIDE) maxallocrow = 10;
128d4002b98SHong Zhang       else if (maxallocrow < 0) maxallocrow = 1;
129d4002b98SHong Zhang       for (i=0; i<=totalslices; i++) b->sliidx[i] = i*8*maxallocrow;
130d4002b98SHong Zhang     } else {
131d4002b98SHong Zhang       maxallocrow = 0;
132d4002b98SHong Zhang       b->sliidx[0] = 0;
133d4002b98SHong Zhang       for (i=1; i<totalslices; i++) {
134d4002b98SHong Zhang         b->sliidx[i] = 0;
135d4002b98SHong Zhang         for (j=0;j<8;j++) {
136d4002b98SHong Zhang           b->sliidx[i] = PetscMax(b->sliidx[i],rlen[8*(i-1)+j]);
137d4002b98SHong Zhang         }
138d4002b98SHong Zhang         maxallocrow = PetscMax(b->sliidx[i],maxallocrow);
139c73702f5SBarry Smith         ierr = PetscIntSumError(b->sliidx[i-1],8*b->sliidx[i],&b->sliidx[i]);CHKERRQ(ierr);
140d4002b98SHong Zhang       }
141d4002b98SHong Zhang       /* last slice */
142d4002b98SHong Zhang       b->sliidx[totalslices] = 0;
143d4002b98SHong Zhang       for (j=(totalslices-1)*8;j<B->rmap->n;j++) b->sliidx[totalslices] = PetscMax(b->sliidx[totalslices],rlen[j]);
144d4002b98SHong Zhang       maxallocrow = PetscMax(b->sliidx[totalslices],maxallocrow);
145d4002b98SHong Zhang       b->sliidx[totalslices] = b->sliidx[totalslices-1] + 8*b->sliidx[totalslices];
146d4002b98SHong Zhang     }
147d4002b98SHong Zhang 
148d4002b98SHong Zhang     /* allocate space for val, colidx, rlen */
149d4002b98SHong Zhang     /* FIXME: should B's old memory be unlogged? */
150d4002b98SHong Zhang     ierr = MatSeqXSELLFreeSELL(B,&b->val,&b->colidx);CHKERRQ(ierr);
151d4002b98SHong Zhang     /* FIXME: assuming an element of the bit array takes 8 bits */
152d4002b98SHong Zhang     ierr = PetscMalloc2(b->sliidx[totalslices],&b->val,b->sliidx[totalslices],&b->colidx);CHKERRQ(ierr);
153d4002b98SHong Zhang     ierr = PetscLogObjectMemory((PetscObject)B,b->sliidx[totalslices]*(sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr);
154d4002b98SHong 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. */
155d4002b98SHong Zhang     ierr = PetscCalloc1(8*totalslices,&b->rlen);CHKERRQ(ierr);
156d4002b98SHong Zhang     ierr = PetscLogObjectMemory((PetscObject)B,8*totalslices*sizeof(PetscInt));CHKERRQ(ierr);
157d4002b98SHong Zhang 
158d4002b98SHong Zhang     b->singlemalloc = PETSC_TRUE;
159d4002b98SHong Zhang     b->free_val     = PETSC_TRUE;
160d4002b98SHong Zhang     b->free_colidx  = PETSC_TRUE;
161d4002b98SHong Zhang   } else {
162d4002b98SHong Zhang     b->free_val    = PETSC_FALSE;
163d4002b98SHong Zhang     b->free_colidx = PETSC_FALSE;
164d4002b98SHong Zhang   }
165d4002b98SHong Zhang 
166d4002b98SHong Zhang   b->nz               = 0;
167d4002b98SHong Zhang   b->maxallocrow      = maxallocrow;
168d4002b98SHong Zhang   b->rlenmax          = maxallocrow;
169d4002b98SHong Zhang   b->maxallocmat      = b->sliidx[totalslices];
170d4002b98SHong Zhang   B->info.nz_unneeded = (double)b->maxallocmat;
171d4002b98SHong Zhang   if (realalloc) {
172d4002b98SHong Zhang     ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
173d4002b98SHong Zhang   }
174d4002b98SHong Zhang   PetscFunctionReturn(0);
175d4002b98SHong Zhang }
176d4002b98SHong Zhang 
1776108893eSStefano Zampini PetscErrorCode MatGetRow_SeqSELL(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
1786108893eSStefano Zampini {
1796108893eSStefano Zampini   Mat_SeqSELL *a = (Mat_SeqSELL*)A->data;
1806108893eSStefano Zampini   PetscInt    shift;
1816108893eSStefano Zampini 
1826108893eSStefano Zampini   PetscFunctionBegin;
183c0aa6a63SJacob Faibussowitsch   if (row < 0 || row >= A->rmap->n) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %" PetscInt_FMT " out of range",row);
1846108893eSStefano Zampini   if (nz) *nz = a->rlen[row];
1856108893eSStefano Zampini   shift = a->sliidx[row>>3]+(row&0x07);
1866108893eSStefano Zampini   if (!a->getrowcols) {
1876108893eSStefano Zampini     PetscErrorCode ierr;
1886108893eSStefano Zampini 
1896108893eSStefano Zampini     ierr = PetscMalloc2(a->rlenmax,&a->getrowcols,a->rlenmax,&a->getrowvals);CHKERRQ(ierr);
1906108893eSStefano Zampini   }
1916108893eSStefano Zampini   if (idx) {
1926108893eSStefano Zampini     PetscInt j;
1936108893eSStefano Zampini     for (j=0; j<a->rlen[row]; j++) a->getrowcols[j] = a->colidx[shift+8*j];
1946108893eSStefano Zampini     *idx = a->getrowcols;
1956108893eSStefano Zampini   }
1966108893eSStefano Zampini   if (v) {
1976108893eSStefano Zampini     PetscInt j;
1986108893eSStefano Zampini     for (j=0; j<a->rlen[row]; j++) a->getrowvals[j] = a->val[shift+8*j];
1996108893eSStefano Zampini     *v = a->getrowvals;
2006108893eSStefano Zampini   }
2016108893eSStefano Zampini   PetscFunctionReturn(0);
2026108893eSStefano Zampini }
2036108893eSStefano Zampini 
2046108893eSStefano Zampini PetscErrorCode MatRestoreRow_SeqSELL(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
2056108893eSStefano Zampini {
2066108893eSStefano Zampini   PetscFunctionBegin;
2076108893eSStefano Zampini   PetscFunctionReturn(0);
2086108893eSStefano Zampini }
2096108893eSStefano Zampini 
210d4002b98SHong Zhang PetscErrorCode MatConvert_SeqSELL_SeqAIJ(Mat A, MatType newtype,MatReuse reuse,Mat *newmat)
211d4002b98SHong Zhang {
212d4002b98SHong Zhang   Mat            B;
213d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
214e3f1f374SStefano Zampini   PetscInt       i;
215d4002b98SHong Zhang   PetscErrorCode ierr;
216d4002b98SHong Zhang 
217d4002b98SHong Zhang   PetscFunctionBegin;
218ad013a7bSRichard Tran Mills   if (reuse == MAT_REUSE_MATRIX) {
219ad013a7bSRichard Tran Mills     B    = *newmat;
220e3f1f374SStefano Zampini     ierr = MatZeroEntries(B);CHKERRQ(ierr);
221ad013a7bSRichard Tran Mills   } else {
222d4002b98SHong Zhang     ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr);
223d4002b98SHong Zhang     ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr);
224d4002b98SHong Zhang     ierr = MatSetType(B,MATSEQAIJ);CHKERRQ(ierr);
225d4002b98SHong Zhang     ierr = MatSeqAIJSetPreallocation(B,0,a->rlen);CHKERRQ(ierr);
226ad013a7bSRichard Tran Mills   }
227d4002b98SHong Zhang 
228e3f1f374SStefano Zampini   for (i=0; i<A->rmap->n; i++) {
229e108cb99SStefano Zampini     PetscInt    nz = 0,*cols = NULL;
230e108cb99SStefano Zampini     PetscScalar *vals = NULL;
231e3f1f374SStefano Zampini 
232e3f1f374SStefano Zampini     ierr = MatGetRow_SeqSELL(A,i,&nz,&cols,&vals);CHKERRQ(ierr);
233e3f1f374SStefano Zampini     ierr = MatSetValues(B,1,&i,nz,cols,vals,INSERT_VALUES);CHKERRQ(ierr);
234e3f1f374SStefano Zampini     ierr = MatRestoreRow_SeqSELL(A,i,&nz,&cols,&vals);CHKERRQ(ierr);
235d4002b98SHong Zhang   }
236e3f1f374SStefano Zampini 
237d4002b98SHong Zhang   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
238d4002b98SHong Zhang   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
239d4002b98SHong Zhang   B->rmap->bs = A->rmap->bs;
240d4002b98SHong Zhang 
241d4002b98SHong Zhang   if (reuse == MAT_INPLACE_MATRIX) {
242d4002b98SHong Zhang     ierr = MatHeaderReplace(A,&B);CHKERRQ(ierr);
243d4002b98SHong Zhang   } else {
244d4002b98SHong Zhang     *newmat = B;
245d4002b98SHong Zhang   }
246d4002b98SHong Zhang   PetscFunctionReturn(0);
247d4002b98SHong Zhang }
248d4002b98SHong Zhang 
249d4002b98SHong Zhang #include <../src/mat/impls/aij/seq/aij.h>
250d4002b98SHong Zhang 
251d4002b98SHong Zhang PetscErrorCode MatConvert_SeqAIJ_SeqSELL(Mat A,MatType newtype,MatReuse reuse,Mat *newmat)
252d4002b98SHong Zhang {
253d4002b98SHong Zhang   Mat               B;
254d4002b98SHong Zhang   Mat_SeqAIJ        *a=(Mat_SeqAIJ*)A->data;
255d4002b98SHong Zhang   PetscInt          *ai=a->i,m=A->rmap->N,n=A->cmap->N,i,*rowlengths,row,ncols;
256d4002b98SHong Zhang   const PetscInt    *cols;
257d4002b98SHong Zhang   const PetscScalar *vals;
258d4002b98SHong Zhang   PetscErrorCode    ierr;
259d4002b98SHong Zhang 
260d4002b98SHong Zhang   PetscFunctionBegin;
261ad013a7bSRichard Tran Mills 
262ad013a7bSRichard Tran Mills   if (reuse == MAT_REUSE_MATRIX) {
263ad013a7bSRichard Tran Mills     B = *newmat;
264ad013a7bSRichard Tran Mills   } else {
265d5e5b2e5SBarry Smith     if (PetscDefined(USE_DEBUG) || !a->ilen) {
266d4002b98SHong Zhang       ierr = PetscMalloc1(m,&rowlengths);CHKERRQ(ierr);
267d4002b98SHong Zhang       for (i=0; i<m; i++) {
268d4002b98SHong Zhang         rowlengths[i] = ai[i+1] - ai[i];
269d4002b98SHong Zhang       }
270d5e5b2e5SBarry Smith     }
271d5e5b2e5SBarry Smith     if (PetscDefined(USE_DEBUG) && a->ilen) {
272d5e5b2e5SBarry Smith       PetscBool eq;
273d5e5b2e5SBarry Smith       ierr = PetscMemcmp(rowlengths,a->ilen,m*sizeof(PetscInt),&eq);CHKERRQ(ierr);
274d5e5b2e5SBarry Smith       if (!eq) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"SeqAIJ ilen array incorrect");
275d5e5b2e5SBarry Smith       ierr = PetscFree(rowlengths);CHKERRQ(ierr);
276d5e5b2e5SBarry Smith       rowlengths = a->ilen;
277d5e5b2e5SBarry Smith     } else if (a->ilen) rowlengths = a->ilen;
278d4002b98SHong Zhang     ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr);
279d4002b98SHong Zhang     ierr = MatSetSizes(B,m,n,m,n);CHKERRQ(ierr);
280d4002b98SHong Zhang     ierr = MatSetType(B,MATSEQSELL);CHKERRQ(ierr);
281d4002b98SHong Zhang     ierr = MatSeqSELLSetPreallocation(B,0,rowlengths);CHKERRQ(ierr);
282d5e5b2e5SBarry Smith     if (rowlengths != a->ilen) {ierr = PetscFree(rowlengths);CHKERRQ(ierr);}
283ad013a7bSRichard Tran Mills   }
284d4002b98SHong Zhang 
285d4002b98SHong Zhang   for (row=0; row<m; row++) {
286d5e5b2e5SBarry Smith     ierr = MatGetRow_SeqAIJ(A,row,&ncols,(PetscInt**)&cols,(PetscScalar**)&vals);CHKERRQ(ierr);
287d5e5b2e5SBarry Smith     ierr = MatSetValues_SeqSELL(B,1,&row,ncols,cols,vals,INSERT_VALUES);CHKERRQ(ierr);
288d5e5b2e5SBarry Smith     ierr = MatRestoreRow_SeqAIJ(A,row,&ncols,(PetscInt**)&cols,(PetscScalar**)&vals);CHKERRQ(ierr);
289d4002b98SHong Zhang   }
290d4002b98SHong Zhang   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
291d4002b98SHong Zhang   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
292d4002b98SHong Zhang   B->rmap->bs = A->rmap->bs;
293d4002b98SHong Zhang 
294d4002b98SHong Zhang   if (reuse == MAT_INPLACE_MATRIX) {
295d4002b98SHong Zhang     ierr = MatHeaderReplace(A,&B);CHKERRQ(ierr);
296d4002b98SHong Zhang   } else {
297d4002b98SHong Zhang     *newmat = B;
298d4002b98SHong Zhang   }
299d4002b98SHong Zhang   PetscFunctionReturn(0);
300d4002b98SHong Zhang }
301d4002b98SHong Zhang 
302d4002b98SHong Zhang PetscErrorCode MatMult_SeqSELL(Mat A,Vec xx,Vec yy)
303d4002b98SHong Zhang {
304d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
305d4002b98SHong Zhang   PetscScalar       *y;
306d4002b98SHong Zhang   const PetscScalar *x;
307d4002b98SHong Zhang   const MatScalar   *aval=a->val;
308d4002b98SHong Zhang   PetscInt          totalslices=a->totalslices;
309d4002b98SHong Zhang   const PetscInt    *acolidx=a->colidx;
3107285fed1SHong Zhang   PetscInt          i,j;
311d4002b98SHong Zhang   PetscErrorCode    ierr;
312d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
313d4002b98SHong Zhang   __m512d           vec_x,vec_y,vec_vals;
314d4002b98SHong Zhang   __m256i           vec_idx;
315d4002b98SHong Zhang   __mmask8          mask;
316d4002b98SHong Zhang   __m512d           vec_x2,vec_y2,vec_vals2,vec_x3,vec_y3,vec_vals3,vec_x4,vec_y4,vec_vals4;
317d4002b98SHong Zhang   __m256i           vec_idx2,vec_idx3,vec_idx4;
3185f70456aSHong 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)
319a48a6482SHong Zhang   __m128i           vec_idx;
320a48a6482SHong Zhang   __m256d           vec_x,vec_y,vec_y2,vec_vals;
321a48a6482SHong Zhang   MatScalar         yval;
322a48a6482SHong Zhang   PetscInt          r,rows_left,row,nnz_in_row;
32321cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
324d4002b98SHong Zhang   __m128d           vec_x_tmp;
325d4002b98SHong Zhang   __m256d           vec_x,vec_y,vec_y2,vec_vals;
326d4002b98SHong Zhang   MatScalar         yval;
327d4002b98SHong Zhang   PetscInt          r,rows_left,row,nnz_in_row;
328d4002b98SHong Zhang #else
329d4002b98SHong Zhang   PetscScalar       sum[8];
330d4002b98SHong Zhang #endif
331d4002b98SHong Zhang 
332d4002b98SHong Zhang #if defined(PETSC_HAVE_PRAGMA_DISJOINT)
333d4002b98SHong Zhang #pragma disjoint(*x,*y,*aval)
334d4002b98SHong Zhang #endif
335d4002b98SHong Zhang 
336d4002b98SHong Zhang   PetscFunctionBegin;
337d4002b98SHong Zhang   ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr);
338d4002b98SHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
339d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
340d4002b98SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over slices */
341d4002b98SHong Zhang     PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
342d4002b98SHong Zhang     PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
343d4002b98SHong Zhang 
344d4002b98SHong Zhang     vec_y  = _mm512_setzero_pd();
345d4002b98SHong Zhang     vec_y2 = _mm512_setzero_pd();
346d4002b98SHong Zhang     vec_y3 = _mm512_setzero_pd();
347d4002b98SHong Zhang     vec_y4 = _mm512_setzero_pd();
348d4002b98SHong Zhang 
34938efe8efSHong Zhang     j = a->sliidx[i]>>3; /* 8 bytes are read at each time, corresponding to a slice columnn */
350d4002b98SHong Zhang     switch ((a->sliidx[i+1]-a->sliidx[i])/8 & 3) {
351d4002b98SHong Zhang     case 3:
352d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
353d4002b98SHong Zhang       acolidx += 8; aval += 8;
354d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
355d4002b98SHong Zhang       acolidx += 8; aval += 8;
356d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3);
357d4002b98SHong Zhang       acolidx += 8; aval += 8;
358d4002b98SHong Zhang       j += 3;
359d4002b98SHong Zhang       break;
360d4002b98SHong Zhang     case 2:
361d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
362d4002b98SHong Zhang       acolidx += 8; aval += 8;
363d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
364d4002b98SHong Zhang       acolidx += 8; aval += 8;
365d4002b98SHong Zhang       j += 2;
366d4002b98SHong Zhang       break;
367d4002b98SHong Zhang     case 1:
368d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
369d4002b98SHong Zhang       acolidx += 8; aval += 8;
370d4002b98SHong Zhang       j += 1;
371d4002b98SHong Zhang       break;
372d4002b98SHong Zhang     }
373d4002b98SHong Zhang     #pragma novector
374d4002b98SHong Zhang     for (; j<(a->sliidx[i+1]>>3); j+=4) {
375d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
376d4002b98SHong Zhang       acolidx += 8; aval += 8;
377d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
378d4002b98SHong Zhang       acolidx += 8; aval += 8;
379d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3);
380d4002b98SHong Zhang       acolidx += 8; aval += 8;
381d4002b98SHong Zhang       AVX512_Mult_Private(vec_idx4,vec_x4,vec_vals4,vec_y4);
382d4002b98SHong Zhang       acolidx += 8; aval += 8;
383d4002b98SHong Zhang     }
384d4002b98SHong Zhang 
385d4002b98SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y2);
386d4002b98SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y3);
387d4002b98SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y4);
388d4002b98SHong Zhang     if (i == totalslices-1 && A->rmap->n & 0x07) { /* if last slice has padding rows */
389d4002b98SHong Zhang       mask = (__mmask8)(0xff >> (8-(A->rmap->n & 0x07)));
390ef588d5cSRichard Tran Mills       _mm512_mask_storeu_pd(&y[8*i],mask,vec_y);
391d4002b98SHong Zhang     } else {
392ef588d5cSRichard Tran Mills       _mm512_storeu_pd(&y[8*i],vec_y);
393d4002b98SHong Zhang     }
394d4002b98SHong Zhang   }
3955f70456aSHong 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)
396a48a6482SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over full slices */
397a48a6482SHong Zhang     PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
398a48a6482SHong Zhang     PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
399a48a6482SHong Zhang 
400a48a6482SHong Zhang     /* last slice may have padding rows. Don't use vectorization. */
401a48a6482SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) {
402a48a6482SHong Zhang       rows_left = A->rmap->n - 8*i;
403a48a6482SHong Zhang       for (r=0; r<rows_left; ++r) {
404a48a6482SHong Zhang         yval = (MatScalar)0;
405a48a6482SHong Zhang         row = 8*i + r;
406a48a6482SHong Zhang         nnz_in_row = a->rlen[row];
407a48a6482SHong Zhang         for (j=0; j<nnz_in_row; ++j) yval += aval[8*j+r] * x[acolidx[8*j+r]];
408a48a6482SHong Zhang         y[row] = yval;
409a48a6482SHong Zhang       }
410a48a6482SHong Zhang       break;
411a48a6482SHong Zhang     }
412a48a6482SHong Zhang 
413a48a6482SHong Zhang     vec_y  = _mm256_setzero_pd();
414a48a6482SHong Zhang     vec_y2 = _mm256_setzero_pd();
415a48a6482SHong Zhang 
416a48a6482SHong Zhang     /* Process slice of height 8 (512 bits) via two subslices of height 4 (256 bits) via AVX */
417a48a6482SHong Zhang     #pragma novector
418a48a6482SHong Zhang     #pragma unroll(2)
419a48a6482SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
420a48a6482SHong Zhang       AVX2_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
421a48a6482SHong Zhang       aval += 4; acolidx += 4;
422a48a6482SHong Zhang       AVX2_Mult_Private(vec_idx,vec_x,vec_vals,vec_y2);
423a48a6482SHong Zhang       aval += 4; acolidx += 4;
424a48a6482SHong Zhang     }
425a48a6482SHong Zhang 
426ef588d5cSRichard Tran Mills     _mm256_storeu_pd(y+i*8,vec_y);
427ef588d5cSRichard Tran Mills     _mm256_storeu_pd(y+i*8+4,vec_y2);
428a48a6482SHong Zhang   }
42921cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
430d4002b98SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over full slices */
431d4002b98SHong Zhang     PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
432d4002b98SHong Zhang     PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
433d4002b98SHong Zhang 
434d4002b98SHong Zhang     vec_y  = _mm256_setzero_pd();
435d4002b98SHong Zhang     vec_y2 = _mm256_setzero_pd();
436d4002b98SHong Zhang 
437d4002b98SHong Zhang     /* last slice may have padding rows. Don't use vectorization. */
438d4002b98SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) {
439d4002b98SHong Zhang       rows_left = A->rmap->n - 8*i;
440d4002b98SHong Zhang       for (r=0; r<rows_left; ++r) {
441d4002b98SHong Zhang         yval = (MatScalar)0;
442d4002b98SHong Zhang         row = 8*i + r;
443d4002b98SHong Zhang         nnz_in_row = a->rlen[row];
444d4002b98SHong Zhang         for (j=0; j<nnz_in_row; ++j) yval += aval[8*j + r] * x[acolidx[8*j + r]];
445d4002b98SHong Zhang         y[row] = yval;
446d4002b98SHong Zhang       }
447d4002b98SHong Zhang       break;
448d4002b98SHong Zhang     }
449d4002b98SHong Zhang 
450d4002b98SHong Zhang     /* Process slice of height 8 (512 bits) via two subslices of height 4 (256 bits) via AVX */
451a48a6482SHong Zhang     #pragma novector
452a48a6482SHong Zhang     #pragma unroll(2)
4537285fed1SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
454d4002b98SHong Zhang       vec_vals  = _mm256_loadu_pd(aval);
455d4002b98SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
456d4002b98SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
457d4002b98SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,0);
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,1);
461d4002b98SHong Zhang       vec_y     = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y);
462d4002b98SHong Zhang       aval     += 4;
463d4002b98SHong Zhang 
464d4002b98SHong Zhang       vec_vals  = _mm256_loadu_pd(aval);
465d4002b98SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
466d4002b98SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
467d4002b98SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,0);
468d4002b98SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
469d4002b98SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
470d4002b98SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,1);
471d4002b98SHong Zhang       vec_y2    = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y2);
472d4002b98SHong Zhang       aval     += 4;
473d4002b98SHong Zhang     }
474d4002b98SHong Zhang 
475d4002b98SHong Zhang     _mm256_storeu_pd(y + i*8,     vec_y);
476d4002b98SHong Zhang     _mm256_storeu_pd(y + i*8 + 4, vec_y2);
477d4002b98SHong Zhang   }
478d4002b98SHong Zhang #else
479d4002b98SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over slices */
480d4002b98SHong Zhang     for (j=0; j<8; j++) sum[j] = 0.0;
481d4002b98SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
482d4002b98SHong Zhang       sum[0] += aval[j] * x[acolidx[j]];
483d4002b98SHong Zhang       sum[1] += aval[j+1] * x[acolidx[j+1]];
484d4002b98SHong Zhang       sum[2] += aval[j+2] * x[acolidx[j+2]];
485d4002b98SHong Zhang       sum[3] += aval[j+3] * x[acolidx[j+3]];
486d4002b98SHong Zhang       sum[4] += aval[j+4] * x[acolidx[j+4]];
487d4002b98SHong Zhang       sum[5] += aval[j+5] * x[acolidx[j+5]];
488d4002b98SHong Zhang       sum[6] += aval[j+6] * x[acolidx[j+6]];
489d4002b98SHong Zhang       sum[7] += aval[j+7] * x[acolidx[j+7]];
490d4002b98SHong Zhang     }
491d4002b98SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) { /* if last slice has padding rows */
492d4002b98SHong Zhang       for (j=0; j<(A->rmap->n & 0x07); j++) y[8*i+j] = sum[j];
493d4002b98SHong Zhang     } else {
4947285fed1SHong Zhang       for (j=0; j<8; j++) y[8*i+j] = sum[j];
495d4002b98SHong Zhang     }
496d4002b98SHong Zhang   }
497d4002b98SHong Zhang #endif
498d4002b98SHong Zhang 
499d4002b98SHong Zhang   ierr = PetscLogFlops(2.0*a->nz-a->nonzerorowcnt);CHKERRQ(ierr); /* theoretical minimal FLOPs */
500d4002b98SHong Zhang   ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr);
501d4002b98SHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
502d4002b98SHong Zhang   PetscFunctionReturn(0);
503d4002b98SHong Zhang }
504d4002b98SHong Zhang 
505d4002b98SHong Zhang #include <../src/mat/impls/aij/seq/ftn-kernels/fmultadd.h>
506d4002b98SHong Zhang PetscErrorCode MatMultAdd_SeqSELL(Mat A,Vec xx,Vec yy,Vec zz)
507d4002b98SHong Zhang {
508d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
509d4002b98SHong Zhang   PetscScalar       *y,*z;
510d4002b98SHong Zhang   const PetscScalar *x;
511d4002b98SHong Zhang   const MatScalar   *aval=a->val;
512d4002b98SHong Zhang   PetscInt          totalslices=a->totalslices;
513d4002b98SHong Zhang   const PetscInt    *acolidx=a->colidx;
514d4002b98SHong Zhang   PetscInt          i,j;
515d4002b98SHong Zhang   PetscErrorCode    ierr;
516d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
5177285fed1SHong Zhang   __m512d           vec_x,vec_y,vec_vals;
518d4002b98SHong Zhang   __m256i           vec_idx;
519d4002b98SHong Zhang   __mmask8          mask;
5207285fed1SHong Zhang   __m512d           vec_x2,vec_y2,vec_vals2,vec_x3,vec_y3,vec_vals3,vec_x4,vec_y4,vec_vals4;
5217285fed1SHong Zhang   __m256i           vec_idx2,vec_idx3,vec_idx4;
52221cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
5237285fed1SHong Zhang   __m128d           vec_x_tmp;
5247285fed1SHong Zhang   __m256d           vec_x,vec_y,vec_y2,vec_vals;
5257285fed1SHong Zhang   MatScalar         yval;
5267285fed1SHong Zhang   PetscInt          r,row,nnz_in_row;
527d4002b98SHong Zhang #else
528d4002b98SHong Zhang   PetscScalar       sum[8];
529d4002b98SHong Zhang #endif
530d4002b98SHong Zhang 
531d4002b98SHong Zhang #if defined(PETSC_HAVE_PRAGMA_DISJOINT)
532d4002b98SHong Zhang #pragma disjoint(*x,*y,*aval)
533d4002b98SHong Zhang #endif
534d4002b98SHong Zhang 
535d4002b98SHong Zhang   PetscFunctionBegin;
536d4002b98SHong Zhang   ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr);
537d4002b98SHong Zhang   ierr = VecGetArrayPair(yy,zz,&y,&z);CHKERRQ(ierr);
538d4002b98SHong Zhang #if defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX512F__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
5397285fed1SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over slices */
5407285fed1SHong Zhang     PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
5417285fed1SHong Zhang     PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
5427285fed1SHong Zhang 
543d4002b98SHong Zhang     if (i == totalslices-1 && A->rmap->n & 0x07) { /* if last slice has padding rows */
544d4002b98SHong Zhang       mask   = (__mmask8)(0xff >> (8-(A->rmap->n & 0x07)));
545ef588d5cSRichard Tran Mills       vec_y  = _mm512_mask_loadu_pd(vec_y,mask,&y[8*i]);
5467285fed1SHong Zhang     } else {
547ef588d5cSRichard Tran Mills       vec_y  = _mm512_loadu_pd(&y[8*i]);
5487285fed1SHong Zhang     }
5497285fed1SHong Zhang     vec_y2 = _mm512_setzero_pd();
5507285fed1SHong Zhang     vec_y3 = _mm512_setzero_pd();
5517285fed1SHong Zhang     vec_y4 = _mm512_setzero_pd();
5527285fed1SHong Zhang 
5537285fed1SHong Zhang     j = a->sliidx[i]>>3; /* 8 bytes are read at each time, corresponding to a slice columnn */
5547285fed1SHong Zhang     switch ((a->sliidx[i+1]-a->sliidx[i])/8 & 3) {
5557285fed1SHong Zhang     case 3:
5567285fed1SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
5577285fed1SHong Zhang       acolidx += 8; aval += 8;
5587285fed1SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
5597285fed1SHong Zhang       acolidx += 8; aval += 8;
5607285fed1SHong Zhang       AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3);
5617285fed1SHong Zhang       acolidx += 8; aval += 8;
5627285fed1SHong Zhang       j += 3;
5637285fed1SHong Zhang       break;
5647285fed1SHong Zhang     case 2:
5657285fed1SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
5667285fed1SHong Zhang       acolidx += 8; aval += 8;
5677285fed1SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
5687285fed1SHong Zhang       acolidx += 8; aval += 8;
5697285fed1SHong Zhang       j += 2;
5707285fed1SHong Zhang       break;
5717285fed1SHong Zhang     case 1:
5727285fed1SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
5737285fed1SHong Zhang       acolidx += 8; aval += 8;
5747285fed1SHong Zhang       j += 1;
5757285fed1SHong Zhang       break;
5767285fed1SHong Zhang     }
5777285fed1SHong Zhang     #pragma novector
5787285fed1SHong Zhang     for (; j<(a->sliidx[i+1]>>3); j+=4) {
5797285fed1SHong Zhang       AVX512_Mult_Private(vec_idx,vec_x,vec_vals,vec_y);
5807285fed1SHong Zhang       acolidx += 8; aval += 8;
5817285fed1SHong Zhang       AVX512_Mult_Private(vec_idx2,vec_x2,vec_vals2,vec_y2);
5827285fed1SHong Zhang       acolidx += 8; aval += 8;
5837285fed1SHong Zhang       AVX512_Mult_Private(vec_idx3,vec_x3,vec_vals3,vec_y3);
5847285fed1SHong Zhang       acolidx += 8; aval += 8;
5857285fed1SHong Zhang       AVX512_Mult_Private(vec_idx4,vec_x4,vec_vals4,vec_y4);
5867285fed1SHong Zhang       acolidx += 8; aval += 8;
5877285fed1SHong Zhang     }
5887285fed1SHong Zhang 
5897285fed1SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y2);
5907285fed1SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y3);
5917285fed1SHong Zhang     vec_y = _mm512_add_pd(vec_y,vec_y4);
5927285fed1SHong Zhang     if (i == totalslices-1 && A->rmap->n & 0x07) { /* if last slice has padding rows */
593ef588d5cSRichard Tran Mills       _mm512_mask_storeu_pd(&z[8*i],mask,vec_y);
594d4002b98SHong Zhang     } else {
595ef588d5cSRichard Tran Mills       _mm512_storeu_pd(&z[8*i],vec_y);
596d4002b98SHong Zhang     }
5977285fed1SHong Zhang   }
59821cec45eSHong Zhang #elif defined(PETSC_HAVE_IMMINTRIN_H) && defined(__AVX__) && defined(PETSC_USE_REAL_DOUBLE) && !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_64BIT_INDICES)
5997285fed1SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over full slices */
6007285fed1SHong Zhang     PetscPrefetchBlock(acolidx,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
6017285fed1SHong Zhang     PetscPrefetchBlock(aval,a->sliidx[i+1]-a->sliidx[i],0,PETSC_PREFETCH_HINT_T0);
6027285fed1SHong Zhang 
6037285fed1SHong Zhang     /* last slice may have padding rows. Don't use vectorization. */
6047285fed1SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) {
6057285fed1SHong Zhang       for (r=0; r<(A->rmap->n & 0x07); ++r) {
6067285fed1SHong Zhang         row        = 8*i + r;
6077285fed1SHong Zhang         yval       = (MatScalar)0.0;
6087285fed1SHong Zhang         nnz_in_row = a->rlen[row];
6097285fed1SHong Zhang         for (j=0; j<nnz_in_row; ++j) yval += aval[8*j+r] * x[acolidx[8*j+r]];
6107285fed1SHong Zhang         z[row] = y[row] + yval;
6117285fed1SHong Zhang       }
6127285fed1SHong Zhang       break;
6137285fed1SHong Zhang     }
6147285fed1SHong Zhang 
6157285fed1SHong Zhang     vec_y  = _mm256_loadu_pd(y+8*i);
6167285fed1SHong Zhang     vec_y2 = _mm256_loadu_pd(y+8*i+4);
6177285fed1SHong Zhang 
6187285fed1SHong Zhang     /* Process slice of height 8 (512 bits) via two subslices of height 4 (256 bits) via AVX */
6197285fed1SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
6207285fed1SHong Zhang       vec_vals  = _mm256_loadu_pd(aval);
6217285fed1SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
6227285fed1SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
6237285fed1SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,0);
6247285fed1SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
6257285fed1SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
6267285fed1SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,1);
6277285fed1SHong Zhang       vec_y     = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y);
6287285fed1SHong Zhang       aval     += 4;
6297285fed1SHong Zhang 
6307285fed1SHong Zhang       vec_vals  = _mm256_loadu_pd(aval);
6317285fed1SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
6327285fed1SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
6337285fed1SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,0);
6347285fed1SHong Zhang       vec_x_tmp = _mm_loadl_pd(vec_x_tmp, x + *acolidx++);
6357285fed1SHong Zhang       vec_x_tmp = _mm_loadh_pd(vec_x_tmp, x + *acolidx++);
6367285fed1SHong Zhang       vec_x     = _mm256_insertf128_pd(vec_x,vec_x_tmp,1);
6377285fed1SHong Zhang       vec_y2    = _mm256_add_pd(_mm256_mul_pd(vec_x,vec_vals),vec_y2);
6387285fed1SHong Zhang       aval     += 4;
6397285fed1SHong Zhang     }
6407285fed1SHong Zhang 
6417285fed1SHong Zhang     _mm256_storeu_pd(z+i*8,vec_y);
6427285fed1SHong Zhang     _mm256_storeu_pd(z+i*8+4,vec_y2);
6437285fed1SHong Zhang   }
644d4002b98SHong Zhang #else
6457285fed1SHong Zhang   for (i=0; i<totalslices; i++) { /* loop over slices */
6467285fed1SHong Zhang     for (j=0; j<8; j++) sum[j] = 0.0;
647d4002b98SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
648d4002b98SHong Zhang       sum[0] += aval[j] * x[acolidx[j]];
649d4002b98SHong Zhang       sum[1] += aval[j+1] * x[acolidx[j+1]];
650d4002b98SHong Zhang       sum[2] += aval[j+2] * x[acolidx[j+2]];
651d4002b98SHong Zhang       sum[3] += aval[j+3] * x[acolidx[j+3]];
652d4002b98SHong Zhang       sum[4] += aval[j+4] * x[acolidx[j+4]];
653d4002b98SHong Zhang       sum[5] += aval[j+5] * x[acolidx[j+5]];
654d4002b98SHong Zhang       sum[6] += aval[j+6] * x[acolidx[j+6]];
655d4002b98SHong Zhang       sum[7] += aval[j+7] * x[acolidx[j+7]];
656d4002b98SHong Zhang     }
6577285fed1SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) {
6587285fed1SHong Zhang       for (j=0; j<(A->rmap->n & 0x07); j++) z[8*i+j] = y[8*i+j] + sum[j];
659d4002b98SHong Zhang     } else {
6607285fed1SHong Zhang       for (j=0; j<8; j++) z[8*i+j] = y[8*i+j] + sum[j];
6617285fed1SHong Zhang     }
662d4002b98SHong Zhang   }
663d4002b98SHong Zhang #endif
664d4002b98SHong Zhang 
665d4002b98SHong Zhang   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
666d4002b98SHong Zhang   ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr);
667d4002b98SHong Zhang   ierr = VecRestoreArrayPair(yy,zz,&y,&z);CHKERRQ(ierr);
668d4002b98SHong Zhang   PetscFunctionReturn(0);
669d4002b98SHong Zhang }
670d4002b98SHong Zhang 
671d4002b98SHong Zhang PetscErrorCode MatMultTransposeAdd_SeqSELL(Mat A,Vec xx,Vec zz,Vec yy)
672d4002b98SHong Zhang {
673d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
674d4002b98SHong Zhang   PetscScalar       *y;
675d4002b98SHong Zhang   const PetscScalar *x;
676d4002b98SHong Zhang   const MatScalar   *aval=a->val;
677d4002b98SHong Zhang   const PetscInt    *acolidx=a->colidx;
6787285fed1SHong Zhang   PetscInt          i,j,r,row,nnz_in_row,totalslices=a->totalslices;
679d4002b98SHong Zhang   PetscErrorCode    ierr;
680d4002b98SHong Zhang 
681d4002b98SHong Zhang #if defined(PETSC_HAVE_PRAGMA_DISJOINT)
682d4002b98SHong Zhang #pragma disjoint(*x,*y,*aval)
683d4002b98SHong Zhang #endif
684d4002b98SHong Zhang 
685d4002b98SHong Zhang   PetscFunctionBegin;
6869fc32365SStefano Zampini   if (A->symmetric) {
6879fc32365SStefano Zampini     ierr = MatMultAdd_SeqSELL(A,xx,zz,yy);CHKERRQ(ierr);
6889fc32365SStefano Zampini     PetscFunctionReturn(0);
6899fc32365SStefano Zampini   }
690d4002b98SHong Zhang   if (zz != yy) { ierr = VecCopy(zz,yy);CHKERRQ(ierr); }
691d4002b98SHong Zhang   ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr);
692d4002b98SHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
693d4002b98SHong Zhang   for (i=0; i<a->totalslices; i++) { /* loop over slices */
6947285fed1SHong Zhang     if (i == totalslices-1 && (A->rmap->n & 0x07)) {
6957285fed1SHong Zhang       for (r=0; r<(A->rmap->n & 0x07); ++r) {
6967285fed1SHong Zhang         row        = 8*i + r;
6977285fed1SHong Zhang         nnz_in_row = a->rlen[row];
6987285fed1SHong Zhang         for (j=0; j<nnz_in_row; ++j) y[acolidx[8*j+r]] += aval[8*j+r] * x[row];
6997285fed1SHong Zhang       }
7007285fed1SHong Zhang       break;
7017285fed1SHong Zhang     }
7027285fed1SHong Zhang     for (j=a->sliidx[i]; j<a->sliidx[i+1]; j+=8) {
7037285fed1SHong Zhang       y[acolidx[j]]   += aval[j] * x[8*i];
7047285fed1SHong Zhang       y[acolidx[j+1]] += aval[j+1] * x[8*i+1];
7057285fed1SHong Zhang       y[acolidx[j+2]] += aval[j+2] * x[8*i+2];
7067285fed1SHong Zhang       y[acolidx[j+3]] += aval[j+3] * x[8*i+3];
7077285fed1SHong Zhang       y[acolidx[j+4]] += aval[j+4] * x[8*i+4];
7087285fed1SHong Zhang       y[acolidx[j+5]] += aval[j+5] * x[8*i+5];
7097285fed1SHong Zhang       y[acolidx[j+6]] += aval[j+6] * x[8*i+6];
7107285fed1SHong Zhang       y[acolidx[j+7]] += aval[j+7] * x[8*i+7];
711d4002b98SHong Zhang     }
712d4002b98SHong Zhang   }
713d4002b98SHong Zhang   ierr = PetscLogFlops(2.0*a->sliidx[a->totalslices]);CHKERRQ(ierr);
714d4002b98SHong Zhang   ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr);
715d4002b98SHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
716d4002b98SHong Zhang   PetscFunctionReturn(0);
717d4002b98SHong Zhang }
718d4002b98SHong Zhang 
719d4002b98SHong Zhang PetscErrorCode MatMultTranspose_SeqSELL(Mat A,Vec xx,Vec yy)
720d4002b98SHong Zhang {
721d4002b98SHong Zhang   PetscErrorCode ierr;
722d4002b98SHong Zhang 
723d4002b98SHong Zhang   PetscFunctionBegin;
7249fc32365SStefano Zampini   if (A->symmetric) {
7259fc32365SStefano Zampini     ierr = MatMult_SeqSELL(A,xx,yy);CHKERRQ(ierr);
7269fc32365SStefano Zampini   } else {
727d4002b98SHong Zhang     ierr = VecSet(yy,0.0);CHKERRQ(ierr);
728d4002b98SHong Zhang     ierr = MatMultTransposeAdd_SeqSELL(A,xx,yy,yy);CHKERRQ(ierr);
7299fc32365SStefano Zampini   }
730d4002b98SHong Zhang   PetscFunctionReturn(0);
731d4002b98SHong Zhang }
732d4002b98SHong Zhang 
733d4002b98SHong Zhang /*
734d4002b98SHong Zhang      Checks for missing diagonals
735d4002b98SHong Zhang */
736d4002b98SHong Zhang PetscErrorCode MatMissingDiagonal_SeqSELL(Mat A,PetscBool  *missing,PetscInt *d)
737d4002b98SHong Zhang {
738d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
739d4002b98SHong Zhang   PetscInt       *diag,i;
740994fe344SLisandro Dalcin   PetscErrorCode ierr;
741d4002b98SHong Zhang 
742d4002b98SHong Zhang   PetscFunctionBegin;
743d4002b98SHong Zhang   *missing = PETSC_FALSE;
744d4002b98SHong Zhang   if (A->rmap->n > 0 && !(a->colidx)) {
745d4002b98SHong Zhang     *missing = PETSC_TRUE;
746d4002b98SHong Zhang     if (d) *d = 0;
747994fe344SLisandro Dalcin     ierr = PetscInfo(A,"Matrix has no entries therefore is missing diagonal\n");CHKERRQ(ierr);
748d4002b98SHong Zhang   } else {
749d4002b98SHong Zhang     diag = a->diag;
750d4002b98SHong Zhang     for (i=0; i<A->rmap->n; i++) {
751d4002b98SHong Zhang       if (diag[i] == -1) {
752d4002b98SHong Zhang         *missing = PETSC_TRUE;
753d4002b98SHong Zhang         if (d) *d = i;
754c0aa6a63SJacob Faibussowitsch         ierr = PetscInfo1(A,"Matrix is missing diagonal number %" PetscInt_FMT "\n",i);CHKERRQ(ierr);
755d4002b98SHong Zhang         break;
756d4002b98SHong Zhang       }
757d4002b98SHong Zhang     }
758d4002b98SHong Zhang   }
759d4002b98SHong Zhang   PetscFunctionReturn(0);
760d4002b98SHong Zhang }
761d4002b98SHong Zhang 
762d4002b98SHong Zhang PetscErrorCode MatMarkDiagonal_SeqSELL(Mat A)
763d4002b98SHong Zhang {
764d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
765d4002b98SHong Zhang   PetscInt       i,j,m=A->rmap->n,shift;
766d4002b98SHong Zhang   PetscErrorCode ierr;
767d4002b98SHong Zhang 
768d4002b98SHong Zhang   PetscFunctionBegin;
769d4002b98SHong Zhang   if (!a->diag) {
770d4002b98SHong Zhang     ierr         = PetscMalloc1(m,&a->diag);CHKERRQ(ierr);
771d4002b98SHong Zhang     ierr         = PetscLogObjectMemory((PetscObject)A,m*sizeof(PetscInt));CHKERRQ(ierr);
772d4002b98SHong Zhang     a->free_diag = PETSC_TRUE;
773d4002b98SHong Zhang   }
774d4002b98SHong Zhang   for (i=0; i<m; i++) { /* loop over rows */
775d4002b98SHong Zhang     shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
776d4002b98SHong Zhang     a->diag[i] = -1;
777d4002b98SHong Zhang     for (j=0; j<a->rlen[i]; j++) {
778d4002b98SHong Zhang       if (a->colidx[shift+j*8] == i) {
779d4002b98SHong Zhang         a->diag[i] = shift+j*8;
780d4002b98SHong Zhang         break;
781d4002b98SHong Zhang       }
782d4002b98SHong Zhang     }
783d4002b98SHong Zhang   }
784d4002b98SHong Zhang   PetscFunctionReturn(0);
785d4002b98SHong Zhang }
786d4002b98SHong Zhang 
787d4002b98SHong Zhang /*
788d4002b98SHong Zhang   Negative shift indicates do not generate an error if there is a zero diagonal, just invert it anyways
789d4002b98SHong Zhang */
790d4002b98SHong Zhang PetscErrorCode MatInvertDiagonal_SeqSELL(Mat A,PetscScalar omega,PetscScalar fshift)
791d4002b98SHong Zhang {
792d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*) A->data;
793d4002b98SHong Zhang   PetscInt       i,*diag,m = A->rmap->n;
794d4002b98SHong Zhang   MatScalar      *val = a->val;
795d4002b98SHong Zhang   PetscScalar    *idiag,*mdiag;
796d4002b98SHong Zhang   PetscErrorCode ierr;
797d4002b98SHong Zhang 
798d4002b98SHong Zhang   PetscFunctionBegin;
799d4002b98SHong Zhang   if (a->idiagvalid) PetscFunctionReturn(0);
800d4002b98SHong Zhang   ierr = MatMarkDiagonal_SeqSELL(A);CHKERRQ(ierr);
801d4002b98SHong Zhang   diag = a->diag;
802d4002b98SHong Zhang   if (!a->idiag) {
803d4002b98SHong Zhang     ierr = PetscMalloc3(m,&a->idiag,m,&a->mdiag,m,&a->ssor_work);CHKERRQ(ierr);
804d4002b98SHong Zhang     ierr = PetscLogObjectMemory((PetscObject)A, 3*m*sizeof(PetscScalar));CHKERRQ(ierr);
805d4002b98SHong Zhang     val  = a->val;
806d4002b98SHong Zhang   }
807d4002b98SHong Zhang   mdiag = a->mdiag;
808d4002b98SHong Zhang   idiag = a->idiag;
809d4002b98SHong Zhang 
810d4002b98SHong Zhang   if (omega == 1.0 && PetscRealPart(fshift) <= 0.0) {
811d4002b98SHong Zhang     for (i=0; i<m; i++) {
812d4002b98SHong Zhang       mdiag[i] = val[diag[i]];
813d4002b98SHong Zhang       if (!PetscAbsScalar(mdiag[i])) { /* zero diagonal */
814d4002b98SHong Zhang         if (PetscRealPart(fshift)) {
815c0aa6a63SJacob Faibussowitsch           ierr = PetscInfo1(A,"Zero diagonal on row %" PetscInt_FMT "\n",i);CHKERRQ(ierr);
816d4002b98SHong Zhang           A->factorerrortype             = MAT_FACTOR_NUMERIC_ZEROPIVOT;
817d4002b98SHong Zhang           A->factorerror_zeropivot_value = 0.0;
818d4002b98SHong Zhang           A->factorerror_zeropivot_row   = i;
819c0aa6a63SJacob Faibussowitsch         } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Zero diagonal on row %" PetscInt_FMT,i);
820d4002b98SHong Zhang       }
821d4002b98SHong Zhang       idiag[i] = 1.0/val[diag[i]];
822d4002b98SHong Zhang     }
823d4002b98SHong Zhang     ierr = PetscLogFlops(m);CHKERRQ(ierr);
824d4002b98SHong Zhang   } else {
825d4002b98SHong Zhang     for (i=0; i<m; i++) {
826d4002b98SHong Zhang       mdiag[i] = val[diag[i]];
827d4002b98SHong Zhang       idiag[i] = omega/(fshift + val[diag[i]]);
828d4002b98SHong Zhang     }
829d4002b98SHong Zhang     ierr = PetscLogFlops(2.0*m);CHKERRQ(ierr);
830d4002b98SHong Zhang   }
831d4002b98SHong Zhang   a->idiagvalid = PETSC_TRUE;
832d4002b98SHong Zhang   PetscFunctionReturn(0);
833d4002b98SHong Zhang }
834d4002b98SHong Zhang 
835d4002b98SHong Zhang PetscErrorCode MatZeroEntries_SeqSELL(Mat A)
836d4002b98SHong Zhang {
837d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
838d4002b98SHong Zhang   PetscErrorCode ierr;
839d4002b98SHong Zhang 
840d4002b98SHong Zhang   PetscFunctionBegin;
841580bdb30SBarry Smith   ierr = PetscArrayzero(a->val,a->sliidx[a->totalslices]);CHKERRQ(ierr);
842d4002b98SHong Zhang   ierr = MatSeqSELLInvalidateDiagonal(A);CHKERRQ(ierr);
843d4002b98SHong Zhang   PetscFunctionReturn(0);
844d4002b98SHong Zhang }
845d4002b98SHong Zhang 
846d4002b98SHong Zhang PetscErrorCode MatDestroy_SeqSELL(Mat A)
847d4002b98SHong Zhang {
848d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
849d4002b98SHong Zhang   PetscErrorCode ierr;
850d4002b98SHong Zhang 
851d4002b98SHong Zhang   PetscFunctionBegin;
852d4002b98SHong Zhang #if defined(PETSC_USE_LOG)
853c0aa6a63SJacob Faibussowitsch   PetscLogObjectState((PetscObject)A,"Rows=%" PetscInt_FMT ", Cols=%" PetscInt_FMT ", NZ=%" PetscInt_FMT,A->rmap->n,A->cmap->n,a->nz);
854d4002b98SHong Zhang #endif
855d4002b98SHong Zhang   ierr = MatSeqXSELLFreeSELL(A,&a->val,&a->colidx);CHKERRQ(ierr);
856d4002b98SHong Zhang   ierr = ISDestroy(&a->row);CHKERRQ(ierr);
857d4002b98SHong Zhang   ierr = ISDestroy(&a->col);CHKERRQ(ierr);
858d4002b98SHong Zhang   ierr = PetscFree(a->diag);CHKERRQ(ierr);
859d4002b98SHong Zhang   ierr = PetscFree(a->rlen);CHKERRQ(ierr);
860d4002b98SHong Zhang   ierr = PetscFree(a->sliidx);CHKERRQ(ierr);
861d4002b98SHong Zhang   ierr = PetscFree3(a->idiag,a->mdiag,a->ssor_work);CHKERRQ(ierr);
862d4002b98SHong Zhang   ierr = PetscFree(a->solve_work);CHKERRQ(ierr);
863d4002b98SHong Zhang   ierr = ISDestroy(&a->icol);CHKERRQ(ierr);
864d4002b98SHong Zhang   ierr = PetscFree(a->saved_values);CHKERRQ(ierr);
8656108893eSStefano Zampini   ierr = PetscFree2(a->getrowcols,a->getrowvals);CHKERRQ(ierr);
866d4002b98SHong Zhang 
867d4002b98SHong Zhang   ierr = PetscFree(A->data);CHKERRQ(ierr);
868d4002b98SHong Zhang 
869f4259b30SLisandro Dalcin   ierr = PetscObjectChangeTypeName((PetscObject)A,NULL);CHKERRQ(ierr);
870d4002b98SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C",NULL);CHKERRQ(ierr);
871d4002b98SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C",NULL);CHKERRQ(ierr);
872d4002b98SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSELLSetPreallocation_C",NULL);CHKERRQ(ierr);
873d4002b98SHong Zhang   PetscFunctionReturn(0);
874d4002b98SHong Zhang }
875d4002b98SHong Zhang 
876d4002b98SHong Zhang PetscErrorCode MatSetOption_SeqSELL(Mat A,MatOption op,PetscBool flg)
877d4002b98SHong Zhang {
878d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
879d4002b98SHong Zhang   PetscErrorCode ierr;
880d4002b98SHong Zhang 
881d4002b98SHong Zhang   PetscFunctionBegin;
882d4002b98SHong Zhang   switch (op) {
883d4002b98SHong Zhang   case MAT_ROW_ORIENTED:
884d4002b98SHong Zhang     a->roworiented = flg;
885d4002b98SHong Zhang     break;
886d4002b98SHong Zhang   case MAT_KEEP_NONZERO_PATTERN:
887d4002b98SHong Zhang     a->keepnonzeropattern = flg;
888d4002b98SHong Zhang     break;
889d4002b98SHong Zhang   case MAT_NEW_NONZERO_LOCATIONS:
890d4002b98SHong Zhang     a->nonew = (flg ? 0 : 1);
891d4002b98SHong Zhang     break;
892d4002b98SHong Zhang   case MAT_NEW_NONZERO_LOCATION_ERR:
893d4002b98SHong Zhang     a->nonew = (flg ? -1 : 0);
894d4002b98SHong Zhang     break;
895d4002b98SHong Zhang   case MAT_NEW_NONZERO_ALLOCATION_ERR:
896d4002b98SHong Zhang     a->nonew = (flg ? -2 : 0);
897d4002b98SHong Zhang     break;
898d4002b98SHong Zhang   case MAT_UNUSED_NONZERO_LOCATION_ERR:
899d4002b98SHong Zhang     a->nounused = (flg ? -1 : 0);
900d4002b98SHong Zhang     break;
9018c78258cSHong Zhang   case MAT_FORCE_DIAGONAL_ENTRIES:
902d4002b98SHong Zhang   case MAT_IGNORE_OFF_PROC_ENTRIES:
903d4002b98SHong Zhang   case MAT_USE_HASH_TABLE:
904071fcb05SBarry Smith   case MAT_SORTED_FULL:
905d4002b98SHong Zhang     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
906d4002b98SHong Zhang     break;
907d4002b98SHong Zhang   case MAT_SPD:
908d4002b98SHong Zhang   case MAT_SYMMETRIC:
909d4002b98SHong Zhang   case MAT_STRUCTURALLY_SYMMETRIC:
910d4002b98SHong Zhang   case MAT_HERMITIAN:
911d4002b98SHong Zhang   case MAT_SYMMETRY_ETERNAL:
912d4002b98SHong Zhang     /* These options are handled directly by MatSetOption() */
913d4002b98SHong Zhang     break;
914d4002b98SHong Zhang   default:
915d4002b98SHong Zhang     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op);
916d4002b98SHong Zhang   }
917d4002b98SHong Zhang   PetscFunctionReturn(0);
918d4002b98SHong Zhang }
919d4002b98SHong Zhang 
920d4002b98SHong Zhang PetscErrorCode MatGetDiagonal_SeqSELL(Mat A,Vec v)
921d4002b98SHong Zhang {
922d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
923d4002b98SHong Zhang   PetscInt       i,j,n,shift;
924d4002b98SHong Zhang   PetscScalar    *x,zero=0.0;
925d4002b98SHong Zhang   PetscErrorCode ierr;
926d4002b98SHong Zhang 
927d4002b98SHong Zhang   PetscFunctionBegin;
928d4002b98SHong Zhang   ierr = VecGetLocalSize(v,&n);CHKERRQ(ierr);
929d4002b98SHong Zhang   if (n != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector");
930d4002b98SHong Zhang 
931d4002b98SHong Zhang   if (A->factortype == MAT_FACTOR_ILU || A->factortype == MAT_FACTOR_LU) {
932d4002b98SHong Zhang     PetscInt *diag=a->diag;
933d4002b98SHong Zhang     ierr = VecGetArray(v,&x);CHKERRQ(ierr);
934d4002b98SHong Zhang     for (i=0; i<n; i++) x[i] = 1.0/a->val[diag[i]];
935d4002b98SHong Zhang     ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
936d4002b98SHong Zhang     PetscFunctionReturn(0);
937d4002b98SHong Zhang   }
938d4002b98SHong Zhang 
939d4002b98SHong Zhang   ierr = VecSet(v,zero);CHKERRQ(ierr);
940d4002b98SHong Zhang   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
941d4002b98SHong Zhang   for (i=0; i<n; i++) { /* loop over rows */
942d4002b98SHong Zhang     shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
943d4002b98SHong Zhang     x[i] = 0;
944d4002b98SHong Zhang     for (j=0; j<a->rlen[i]; j++) {
945d4002b98SHong Zhang       if (a->colidx[shift+j*8] == i) {
946d4002b98SHong Zhang         x[i] = a->val[shift+j*8];
947d4002b98SHong Zhang         break;
948d4002b98SHong Zhang       }
949d4002b98SHong Zhang     }
950d4002b98SHong Zhang   }
951d4002b98SHong Zhang   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
952d4002b98SHong Zhang   PetscFunctionReturn(0);
953d4002b98SHong Zhang }
954d4002b98SHong Zhang 
955d4002b98SHong Zhang PetscErrorCode MatDiagonalScale_SeqSELL(Mat A,Vec ll,Vec rr)
956d4002b98SHong Zhang {
957d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
958d4002b98SHong Zhang   const PetscScalar *l,*r;
959d4002b98SHong Zhang   PetscInt          i,j,m,n,row;
960d4002b98SHong Zhang   PetscErrorCode    ierr;
961d4002b98SHong Zhang 
962d4002b98SHong Zhang   PetscFunctionBegin;
963d4002b98SHong Zhang   if (ll) {
964d4002b98SHong Zhang     /* The local size is used so that VecMPI can be passed to this routine
965d4002b98SHong Zhang        by MatDiagonalScale_MPISELL */
966d4002b98SHong Zhang     ierr = VecGetLocalSize(ll,&m);CHKERRQ(ierr);
967d4002b98SHong Zhang     if (m != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Left scaling vector wrong length");
968d4002b98SHong Zhang     ierr = VecGetArrayRead(ll,&l);CHKERRQ(ierr);
969d4002b98SHong Zhang     for (i=0; i<a->totalslices; i++) { /* loop over slices */
970dab86139SHong Zhang       if (i == a->totalslices-1 && (A->rmap->n & 0x07)) { /* if last slice has padding rows */
971dab86139SHong Zhang         for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) {
972dab86139SHong Zhang           if (row < (A->rmap->n & 0x07)) a->val[j] *= l[8*i+row];
973dab86139SHong Zhang         }
974dab86139SHong Zhang       } else {
975d4002b98SHong Zhang         for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) {
976d4002b98SHong Zhang           a->val[j] *= l[8*i+row];
977d4002b98SHong Zhang         }
978d4002b98SHong Zhang       }
979dab86139SHong Zhang     }
980d4002b98SHong Zhang     ierr = VecRestoreArrayRead(ll,&l);CHKERRQ(ierr);
981d4002b98SHong Zhang     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
982d4002b98SHong Zhang   }
983d4002b98SHong Zhang   if (rr) {
984d4002b98SHong Zhang     ierr = VecGetLocalSize(rr,&n);CHKERRQ(ierr);
985d4002b98SHong Zhang     if (n != A->cmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Right scaling vector wrong length");
986d4002b98SHong Zhang     ierr = VecGetArrayRead(rr,&r);CHKERRQ(ierr);
987d4002b98SHong Zhang     for (i=0; i<a->totalslices; i++) { /* loop over slices */
988dab86139SHong Zhang       if (i == a->totalslices-1 && (A->rmap->n & 0x07)) { /* if last slice has padding rows */
989dab86139SHong Zhang         for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) {
990dab86139SHong Zhang           if (row < (A->rmap->n & 0x07)) a->val[j] *= r[a->colidx[j]];
991dab86139SHong Zhang         }
992dab86139SHong Zhang       } else {
993d4002b98SHong Zhang         for (j=a->sliidx[i]; j<a->sliidx[i+1]; j++) {
994d4002b98SHong Zhang           a->val[j] *= r[a->colidx[j]];
995d4002b98SHong Zhang         }
996d4002b98SHong Zhang       }
997dab86139SHong Zhang     }
998d4002b98SHong Zhang     ierr = VecRestoreArrayRead(rr,&r);CHKERRQ(ierr);
999d4002b98SHong Zhang     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
1000d4002b98SHong Zhang   }
1001d4002b98SHong Zhang   ierr = MatSeqSELLInvalidateDiagonal(A);CHKERRQ(ierr);
1002d4002b98SHong Zhang   PetscFunctionReturn(0);
1003d4002b98SHong Zhang }
1004d4002b98SHong Zhang 
1005d4002b98SHong Zhang extern PetscErrorCode MatSetValues_SeqSELL(Mat,PetscInt,const PetscInt[],PetscInt,const PetscInt[],const PetscScalar[],InsertMode);
1006d4002b98SHong Zhang 
1007d4002b98SHong Zhang PetscErrorCode MatGetValues_SeqSELL(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[])
1008d4002b98SHong Zhang {
1009d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
1010d4002b98SHong Zhang   PetscInt    *cp,i,k,low,high,t,row,col,l;
1011d4002b98SHong Zhang   PetscInt    shift;
1012d4002b98SHong Zhang   MatScalar   *vp;
1013d4002b98SHong Zhang 
1014d4002b98SHong Zhang   PetscFunctionBegin;
101568aafef3SStefano Zampini   for (k=0; k<m; k++) { /* loop over requested rows */
1016d4002b98SHong Zhang     row = im[k];
1017d4002b98SHong Zhang     if (row<0) continue;
1018c0aa6a63SJacob Faibussowitsch     if (PetscUnlikelyDebug(row >= A->rmap->n)) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %" PetscInt_FMT " max %" PetscInt_FMT,row,A->rmap->n-1);
1019d4002b98SHong Zhang     shift = a->sliidx[row>>3]+(row&0x07); /* starting index of the row */
1020d4002b98SHong Zhang     cp = a->colidx+shift; /* pointer to the row */
1021d4002b98SHong Zhang     vp = a->val+shift; /* pointer to the row */
102268aafef3SStefano Zampini     for (l=0; l<n; l++) { /* loop over requested columns */
1023d4002b98SHong Zhang       col = in[l];
1024d4002b98SHong Zhang       if (col<0) continue;
1025c0aa6a63SJacob Faibussowitsch       if (PetscUnlikelyDebug(col >= A->cmap->n)) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: row %" PetscInt_FMT " max %" PetscInt_FMT,col,A->cmap->n-1);
1026d4002b98SHong Zhang       high = a->rlen[row]; low = 0; /* assume unsorted */
1027d4002b98SHong Zhang       while (high-low > 5) {
1028d4002b98SHong Zhang         t = (low+high)/2;
1029d4002b98SHong Zhang         if (*(cp+t*8) > col) high = t;
1030d4002b98SHong Zhang         else low = t;
1031d4002b98SHong Zhang       }
1032d4002b98SHong Zhang       for (i=low; i<high; i++) {
1033d4002b98SHong Zhang         if (*(cp+8*i) > col) break;
1034d4002b98SHong Zhang         if (*(cp+8*i) == col) {
1035d4002b98SHong Zhang           *v++ = *(vp+8*i);
1036d4002b98SHong Zhang           goto finished;
1037d4002b98SHong Zhang         }
1038d4002b98SHong Zhang       }
1039d4002b98SHong Zhang       *v++ = 0.0;
1040d4002b98SHong Zhang     finished:;
1041d4002b98SHong Zhang     }
1042d4002b98SHong Zhang   }
1043d4002b98SHong Zhang   PetscFunctionReturn(0);
1044d4002b98SHong Zhang }
1045d4002b98SHong Zhang 
1046d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL_ASCII(Mat A,PetscViewer viewer)
1047d4002b98SHong Zhang {
1048d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
1049d4002b98SHong Zhang   PetscInt          i,j,m=A->rmap->n,shift;
1050d4002b98SHong Zhang   const char        *name;
1051d4002b98SHong Zhang   PetscViewerFormat format;
1052d4002b98SHong Zhang   PetscErrorCode    ierr;
1053d4002b98SHong Zhang 
1054d4002b98SHong Zhang   PetscFunctionBegin;
1055d4002b98SHong Zhang   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
1056d4002b98SHong Zhang   if (format == PETSC_VIEWER_ASCII_MATLAB) {
1057d4002b98SHong Zhang     PetscInt nofinalvalue = 0;
1058d4002b98SHong Zhang     /*
1059d4002b98SHong Zhang     if (m && ((a->i[m] == a->i[m-1]) || (a->j[a->nz-1] != A->cmap->n-1))) {
1060d4002b98SHong Zhang       nofinalvalue = 1;
1061d4002b98SHong Zhang     }
1062d4002b98SHong Zhang     */
1063d4002b98SHong Zhang     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
1064c0aa6a63SJacob Faibussowitsch     ierr = PetscViewerASCIIPrintf(viewer,"%% Size = %" PetscInt_FMT " %" PetscInt_FMT " \n",m,A->cmap->n);CHKERRQ(ierr);
1065c0aa6a63SJacob Faibussowitsch     ierr = PetscViewerASCIIPrintf(viewer,"%% Nonzeros = %" PetscInt_FMT " \n",a->nz);CHKERRQ(ierr);
1066d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1067c0aa6a63SJacob Faibussowitsch     ierr = PetscViewerASCIIPrintf(viewer,"zzz = zeros(%" PetscInt_FMT ",4);\n",a->nz+nofinalvalue);CHKERRQ(ierr);
1068d4002b98SHong Zhang #else
1069c0aa6a63SJacob Faibussowitsch     ierr = PetscViewerASCIIPrintf(viewer,"zzz = zeros(%" PetscInt_FMT ",3);\n",a->nz+nofinalvalue);CHKERRQ(ierr);
1070d4002b98SHong Zhang #endif
1071d4002b98SHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"zzz = [\n");CHKERRQ(ierr);
1072d4002b98SHong Zhang 
1073d4002b98SHong Zhang     for (i=0; i<m; i++) {
1074d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1075d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1076d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1077c0aa6a63SJacob Faibussowitsch         ierr = PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT "  %18.16e %18.16e\n",i+1,a->colidx[shift+8*j]+1,(double)PetscRealPart(a->val[shift+8*j]),(double)PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr);
1078d4002b98SHong Zhang #else
1079c0aa6a63SJacob Faibussowitsch         ierr = PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT "  %18.16e\n",i+1,a->colidx[shift+8*j]+1,(double)a->val[shift+8*j]);CHKERRQ(ierr);
1080d4002b98SHong Zhang #endif
1081d4002b98SHong Zhang       }
1082d4002b98SHong Zhang     }
1083d4002b98SHong Zhang     /*
1084d4002b98SHong Zhang     if (nofinalvalue) {
1085d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1086c0aa6a63SJacob Faibussowitsch       ierr = PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT "  %18.16e %18.16e\n",m,A->cmap->n,0.,0.);CHKERRQ(ierr);
1087d4002b98SHong Zhang #else
1088c0aa6a63SJacob Faibussowitsch       ierr = PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT "  %18.16e\n",m,A->cmap->n,0.0);CHKERRQ(ierr);
1089d4002b98SHong Zhang #endif
1090d4002b98SHong Zhang     }
1091d4002b98SHong Zhang     */
1092d4002b98SHong Zhang     ierr = PetscObjectGetName((PetscObject)A,&name);CHKERRQ(ierr);
1093d4002b98SHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"];\n %s = spconvert(zzz);\n",name);CHKERRQ(ierr);
1094d4002b98SHong Zhang     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
1095d4002b98SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO || format == PETSC_VIEWER_ASCII_INFO) {
1096d4002b98SHong Zhang     PetscFunctionReturn(0);
1097d4002b98SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_COMMON) {
1098d4002b98SHong Zhang     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
1099d4002b98SHong Zhang     for (i=0; i<m; i++) {
1100c0aa6a63SJacob Faibussowitsch       ierr = PetscViewerASCIIPrintf(viewer,"row %" PetscInt_FMT ":",i);CHKERRQ(ierr);
1101d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1102d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1103d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1104d4002b98SHong Zhang         if (PetscImaginaryPart(a->val[shift+8*j]) > 0.0 && PetscRealPart(a->val[shift+8*j]) != 0.0) {
1105c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]),(double)PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr);
1106d4002b98SHong Zhang         } else if (PetscImaginaryPart(a->val[shift+8*j]) < 0.0 && PetscRealPart(a->val[shift+8*j]) != 0.0) {
1107c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]),(double)-PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr);
1108d4002b98SHong Zhang         } else if (PetscRealPart(a->val[shift+8*j]) != 0.0) {
1109c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]));CHKERRQ(ierr);
1110d4002b98SHong Zhang         }
1111d4002b98SHong Zhang #else
1112c0aa6a63SJacob Faibussowitsch         if (a->val[shift+8*j] != 0.0) {ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)a->val[shift+8*j]);CHKERRQ(ierr);}
1113d4002b98SHong Zhang #endif
1114d4002b98SHong Zhang       }
1115d4002b98SHong Zhang       ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
1116d4002b98SHong Zhang     }
1117d4002b98SHong Zhang     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
1118d4002b98SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_DENSE) {
1119d4002b98SHong Zhang     PetscInt    cnt=0,jcnt;
1120d4002b98SHong Zhang     PetscScalar value;
1121d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1122d4002b98SHong Zhang     PetscBool   realonly=PETSC_TRUE;
1123d4002b98SHong Zhang     for (i=0; i<a->sliidx[a->totalslices]; i++) {
1124d4002b98SHong Zhang       if (PetscImaginaryPart(a->val[i]) != 0.0) {
1125d4002b98SHong Zhang         realonly = PETSC_FALSE;
1126d4002b98SHong Zhang         break;
1127d4002b98SHong Zhang       }
1128d4002b98SHong Zhang     }
1129d4002b98SHong Zhang #endif
1130d4002b98SHong Zhang 
1131d4002b98SHong Zhang     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
1132d4002b98SHong Zhang     for (i=0; i<m; i++) {
1133d4002b98SHong Zhang       jcnt = 0;
1134d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1135d4002b98SHong Zhang       for (j=0; j<A->cmap->n; j++) {
1136d4002b98SHong Zhang         if (jcnt < a->rlen[i] && j == a->colidx[shift+8*j]) {
1137d4002b98SHong Zhang           value = a->val[cnt++];
1138d4002b98SHong Zhang           jcnt++;
1139d4002b98SHong Zhang         } else {
1140d4002b98SHong Zhang           value = 0.0;
1141d4002b98SHong Zhang         }
1142d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1143d4002b98SHong Zhang         if (realonly) {
1144d4002b98SHong Zhang           ierr = PetscViewerASCIIPrintf(viewer," %7.5e ",(double)PetscRealPart(value));CHKERRQ(ierr);
1145d4002b98SHong Zhang         } else {
1146d4002b98SHong Zhang           ierr = PetscViewerASCIIPrintf(viewer," %7.5e+%7.5e i ",(double)PetscRealPart(value),(double)PetscImaginaryPart(value));CHKERRQ(ierr);
1147d4002b98SHong Zhang         }
1148d4002b98SHong Zhang #else
1149d4002b98SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer," %7.5e ",(double)value);CHKERRQ(ierr);
1150d4002b98SHong Zhang #endif
1151d4002b98SHong Zhang       }
1152d4002b98SHong Zhang       ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
1153d4002b98SHong Zhang     }
1154d4002b98SHong Zhang     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
1155d4002b98SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_MATRIXMARKET) {
1156d4002b98SHong Zhang     PetscInt fshift=1;
1157d4002b98SHong Zhang     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
1158d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1159d4002b98SHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"%%%%MatrixMarket matrix coordinate complex general\n");CHKERRQ(ierr);
1160d4002b98SHong Zhang #else
1161d4002b98SHong Zhang     ierr = PetscViewerASCIIPrintf(viewer,"%%%%MatrixMarket matrix coordinate real general\n");CHKERRQ(ierr);
1162d4002b98SHong Zhang #endif
1163c0aa6a63SJacob Faibussowitsch     ierr = PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %" PetscInt_FMT "\n", m, A->cmap->n, a->nz);CHKERRQ(ierr);
1164d4002b98SHong Zhang     for (i=0; i<m; i++) {
1165d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1166d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1167d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1168c0aa6a63SJacob Faibussowitsch         ierr = PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %g %g\n",i+fshift,a->colidx[shift+8*j]+fshift,(double)PetscRealPart(a->val[shift+8*j]),(double)PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr);
1169d4002b98SHong Zhang #else
1170c0aa6a63SJacob Faibussowitsch         ierr = PetscViewerASCIIPrintf(viewer,"%" PetscInt_FMT " %" PetscInt_FMT " %g\n",i+fshift,a->colidx[shift+8*j]+fshift,(double)a->val[shift+8*j]);CHKERRQ(ierr);
1171d4002b98SHong Zhang #endif
1172d4002b98SHong Zhang       }
1173d4002b98SHong Zhang     }
1174d4002b98SHong Zhang     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
117568aafef3SStefano Zampini   } else if (format == PETSC_VIEWER_NATIVE) {
117668aafef3SStefano Zampini     for (i=0; i<a->totalslices; i++) { /* loop over slices */
117768aafef3SStefano Zampini       PetscInt row;
1178c0aa6a63SJacob Faibussowitsch       ierr = PetscViewerASCIIPrintf(viewer,"slice %" PetscInt_FMT ": %" PetscInt_FMT " %" PetscInt_FMT "\n",i,a->sliidx[i],a->sliidx[i+1]);CHKERRQ(ierr);
117968aafef3SStefano Zampini       for (j=a->sliidx[i],row=0; j<a->sliidx[i+1]; j++,row=((row+1)&0x07)) {
118068aafef3SStefano Zampini #if defined(PETSC_USE_COMPLEX)
118168aafef3SStefano Zampini         if (PetscImaginaryPart(a->val[j]) > 0.0) {
1182c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer,"  %" PetscInt_FMT " %" PetscInt_FMT " %g + %g i\n",8*i+row,a->colidx[j],(double)PetscRealPart(a->val[j]),(double)PetscImaginaryPart(a->val[j]));CHKERRQ(ierr);
118368aafef3SStefano Zampini         } else if (PetscImaginaryPart(a->val[j]) < 0.0) {
1184c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer,"  %" PetscInt_FMT " %" PetscInt_FMT " %g - %g i\n",8*i+row,a->colidx[j],(double)PetscRealPart(a->val[j]),-(double)PetscImaginaryPart(a->val[j]));CHKERRQ(ierr);
118568aafef3SStefano Zampini         } else {
1186c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer,"  %" PetscInt_FMT " %" PetscInt_FMT " %g\n",8*i+row,a->colidx[j],(double)PetscRealPart(a->val[j]));CHKERRQ(ierr);
118768aafef3SStefano Zampini         }
118868aafef3SStefano Zampini #else
1189c0aa6a63SJacob Faibussowitsch         ierr = PetscViewerASCIIPrintf(viewer,"  %" PetscInt_FMT " %" PetscInt_FMT " %g\n",8*i+row,a->colidx[j],(double)a->val[j]);CHKERRQ(ierr);
119068aafef3SStefano Zampini #endif
119168aafef3SStefano Zampini       }
119268aafef3SStefano Zampini     }
1193d4002b98SHong Zhang   } else {
1194d4002b98SHong Zhang     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
1195d4002b98SHong Zhang     if (A->factortype) {
1196d4002b98SHong Zhang       for (i=0; i<m; i++) {
1197d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07);
1198c0aa6a63SJacob Faibussowitsch         ierr = PetscViewerASCIIPrintf(viewer,"row %" PetscInt_FMT ":",i);CHKERRQ(ierr);
1199d4002b98SHong Zhang         /* L part */
1200d4002b98SHong Zhang         for (j=shift; j<a->diag[i]; j+=8) {
1201d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1202d4002b98SHong Zhang           if (PetscImaginaryPart(a->val[shift+8*j]) > 0.0) {
1203c0aa6a63SJacob Faibussowitsch             ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)PetscImaginaryPart(a->val[j]));CHKERRQ(ierr);
1204d4002b98SHong Zhang           } else if (PetscImaginaryPart(a->val[shift+8*j]) < 0.0) {
1205c0aa6a63SJacob Faibussowitsch             ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)(-PetscImaginaryPart(a->val[j])));CHKERRQ(ierr);
1206d4002b98SHong Zhang           } else {
1207c0aa6a63SJacob Faibussowitsch             ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)PetscRealPart(a->val[j]));CHKERRQ(ierr);
1208d4002b98SHong Zhang           }
1209d4002b98SHong Zhang #else
1210c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)a->val[j]);CHKERRQ(ierr);
1211d4002b98SHong Zhang #endif
1212d4002b98SHong Zhang         }
1213d4002b98SHong Zhang         /* diagonal */
1214d4002b98SHong Zhang         j = a->diag[i];
1215d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1216d4002b98SHong Zhang         if (PetscImaginaryPart(a->val[j]) > 0.0) {
1217c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[j],(double)PetscRealPart(1.0/a->val[j]),(double)PetscImaginaryPart(1.0/a->val[j]));CHKERRQ(ierr);
1218d4002b98SHong Zhang         } else if (PetscImaginaryPart(a->val[j]) < 0.0) {
1219c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[j],(double)PetscRealPart(1.0/a->val[j]),(double)(-PetscImaginaryPart(1.0/a->val[j])));CHKERRQ(ierr);
1220d4002b98SHong Zhang         } else {
1221c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)PetscRealPart(1.0/a->val[j]));CHKERRQ(ierr);
1222d4002b98SHong Zhang         }
1223d4002b98SHong Zhang #else
1224c0aa6a63SJacob Faibussowitsch         ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)(1.0/a->val[j]));CHKERRQ(ierr);
1225d4002b98SHong Zhang #endif
1226d4002b98SHong Zhang 
1227d4002b98SHong Zhang         /* U part */
1228d4002b98SHong Zhang         for (j=a->diag[i]+1; j<shift+8*a->rlen[i]; j+=8) {
1229d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1230d4002b98SHong Zhang           if (PetscImaginaryPart(a->val[j]) > 0.0) {
1231c0aa6a63SJacob Faibussowitsch             ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)PetscImaginaryPart(a->val[j]));CHKERRQ(ierr);
1232d4002b98SHong Zhang           } else if (PetscImaginaryPart(a->val[j]) < 0.0) {
1233c0aa6a63SJacob Faibussowitsch             ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[j],(double)PetscRealPart(a->val[j]),(double)(-PetscImaginaryPart(a->val[j])));CHKERRQ(ierr);
1234d4002b98SHong Zhang           } else {
1235c0aa6a63SJacob Faibussowitsch             ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)PetscRealPart(a->val[j]));CHKERRQ(ierr);
1236d4002b98SHong Zhang           }
1237d4002b98SHong Zhang #else
1238c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[j],(double)a->val[j]);CHKERRQ(ierr);
1239d4002b98SHong Zhang #endif
1240d4002b98SHong Zhang         }
1241d4002b98SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
1242d4002b98SHong Zhang       }
1243d4002b98SHong Zhang     } else {
1244d4002b98SHong Zhang       for (i=0; i<m; i++) {
1245d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07);
1246c0aa6a63SJacob Faibussowitsch         ierr = PetscViewerASCIIPrintf(viewer,"row %" PetscInt_FMT ":",i);CHKERRQ(ierr);
1247d4002b98SHong Zhang         for (j=0; j<a->rlen[i]; j++) {
1248d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
1249d4002b98SHong Zhang           if (PetscImaginaryPart(a->val[j]) > 0.0) {
1250c0aa6a63SJacob Faibussowitsch             ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i)",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]),(double)PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr);
1251d4002b98SHong Zhang           } else if (PetscImaginaryPart(a->val[j]) < 0.0) {
1252c0aa6a63SJacob Faibussowitsch             ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i)",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]),(double)-PetscImaginaryPart(a->val[shift+8*j]));CHKERRQ(ierr);
1253d4002b98SHong Zhang           } else {
1254c0aa6a63SJacob Faibussowitsch             ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)PetscRealPart(a->val[shift+8*j]));CHKERRQ(ierr);
1255d4002b98SHong Zhang           }
1256d4002b98SHong Zhang #else
1257c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->colidx[shift+8*j],(double)a->val[shift+8*j]);CHKERRQ(ierr);
1258d4002b98SHong Zhang #endif
1259d4002b98SHong Zhang         }
1260d4002b98SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
1261d4002b98SHong Zhang       }
1262d4002b98SHong Zhang     }
1263d4002b98SHong Zhang     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
1264d4002b98SHong Zhang   }
1265d4002b98SHong Zhang   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
1266d4002b98SHong Zhang   PetscFunctionReturn(0);
1267d4002b98SHong Zhang }
1268d4002b98SHong Zhang 
1269d4002b98SHong Zhang #include <petscdraw.h>
1270d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL_Draw_Zoom(PetscDraw draw,void *Aa)
1271d4002b98SHong Zhang {
1272d4002b98SHong Zhang   Mat               A=(Mat)Aa;
1273d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
1274d4002b98SHong Zhang   PetscInt          i,j,m=A->rmap->n,shift;
1275d4002b98SHong Zhang   int               color;
1276d4002b98SHong Zhang   PetscReal         xl,yl,xr,yr,x_l,x_r,y_l,y_r;
1277d4002b98SHong Zhang   PetscViewer       viewer;
1278d4002b98SHong Zhang   PetscViewerFormat format;
1279d4002b98SHong Zhang   PetscErrorCode    ierr;
1280d4002b98SHong Zhang 
1281d4002b98SHong Zhang   PetscFunctionBegin;
1282d4002b98SHong Zhang   ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr);
1283d4002b98SHong Zhang   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
1284d4002b98SHong Zhang   ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr);
1285d4002b98SHong Zhang 
1286d4002b98SHong Zhang   /* loop over matrix elements drawing boxes */
1287d4002b98SHong Zhang 
1288d4002b98SHong Zhang   if (format != PETSC_VIEWER_DRAW_CONTOUR) {
1289d4002b98SHong Zhang     ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
1290d4002b98SHong Zhang     /* Blue for negative, Cyan for zero and  Red for positive */
1291d4002b98SHong Zhang     color = PETSC_DRAW_BLUE;
1292d4002b98SHong Zhang     for (i=0; i<m; i++) {
1293d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
1294d4002b98SHong Zhang       y_l = m - i - 1.0; y_r = y_l + 1.0;
1295d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1296d4002b98SHong Zhang         x_l = a->colidx[shift+j*8]; x_r = x_l + 1.0;
1297d4002b98SHong Zhang         if (PetscRealPart(a->val[shift+8*j]) >=  0.) continue;
1298d4002b98SHong Zhang         ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr);
1299d4002b98SHong Zhang       }
1300d4002b98SHong Zhang     }
1301d4002b98SHong Zhang     color = PETSC_DRAW_CYAN;
1302d4002b98SHong Zhang     for (i=0; i<m; i++) {
1303d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1304d4002b98SHong Zhang       y_l = m - i - 1.0; y_r = y_l + 1.0;
1305d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1306d4002b98SHong Zhang         x_l = a->colidx[shift+j*8]; x_r = x_l + 1.0;
1307d4002b98SHong Zhang         if (a->val[shift+8*j] !=  0.) continue;
1308d4002b98SHong Zhang         ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr);
1309d4002b98SHong Zhang       }
1310d4002b98SHong Zhang     }
1311d4002b98SHong Zhang     color = PETSC_DRAW_RED;
1312d4002b98SHong Zhang     for (i=0; i<m; i++) {
1313d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1314d4002b98SHong Zhang       y_l = m - i - 1.0; y_r = y_l + 1.0;
1315d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1316d4002b98SHong Zhang         x_l = a->colidx[shift+j*8]; x_r = x_l + 1.0;
1317d4002b98SHong Zhang         if (PetscRealPart(a->val[shift+8*j]) <=  0.) continue;
1318d4002b98SHong Zhang         ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr);
1319d4002b98SHong Zhang       }
1320d4002b98SHong Zhang     }
1321d4002b98SHong Zhang     ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
1322d4002b98SHong Zhang   } else {
1323d4002b98SHong Zhang     /* use contour shading to indicate magnitude of values */
1324d4002b98SHong Zhang     /* first determine max of all nonzero values */
1325d4002b98SHong Zhang     PetscReal minv=0.0,maxv=0.0;
1326d4002b98SHong Zhang     PetscInt  count=0;
1327d4002b98SHong Zhang     PetscDraw popup;
1328d4002b98SHong Zhang     for (i=0; i<a->sliidx[a->totalslices]; i++) {
1329d4002b98SHong Zhang       if (PetscAbsScalar(a->val[i]) > maxv) maxv = PetscAbsScalar(a->val[i]);
1330d4002b98SHong Zhang     }
1331d4002b98SHong Zhang     if (minv >= maxv) maxv = minv + PETSC_SMALL;
1332d4002b98SHong Zhang     ierr = PetscDrawGetPopup(draw,&popup);CHKERRQ(ierr);
1333d4002b98SHong Zhang     ierr = PetscDrawScalePopup(popup,minv,maxv);CHKERRQ(ierr);
1334d4002b98SHong Zhang 
1335d4002b98SHong Zhang     ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
1336d4002b98SHong Zhang     for (i=0; i<m; i++) {
1337d4002b98SHong Zhang       shift = a->sliidx[i>>3]+(i&0x07);
1338d4002b98SHong Zhang       y_l = m - i - 1.0;
1339d4002b98SHong Zhang       y_r = y_l + 1.0;
1340d4002b98SHong Zhang       for (j=0; j<a->rlen[i]; j++) {
1341d4002b98SHong Zhang         x_l = a->colidx[shift+j*8];
1342d4002b98SHong Zhang         x_r = x_l + 1.0;
1343d4002b98SHong Zhang         color = PetscDrawRealToColor(PetscAbsScalar(a->val[count]),minv,maxv);
1344d4002b98SHong Zhang         ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr);
1345d4002b98SHong Zhang         count++;
1346d4002b98SHong Zhang       }
1347d4002b98SHong Zhang     }
1348d4002b98SHong Zhang     ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
1349d4002b98SHong Zhang   }
1350d4002b98SHong Zhang   PetscFunctionReturn(0);
1351d4002b98SHong Zhang }
1352d4002b98SHong Zhang 
1353d4002b98SHong Zhang #include <petscdraw.h>
1354d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL_Draw(Mat A,PetscViewer viewer)
1355d4002b98SHong Zhang {
1356d4002b98SHong Zhang   PetscDraw      draw;
1357d4002b98SHong Zhang   PetscReal      xr,yr,xl,yl,h,w;
1358d4002b98SHong Zhang   PetscBool      isnull;
1359d4002b98SHong Zhang   PetscErrorCode ierr;
1360d4002b98SHong Zhang 
1361d4002b98SHong Zhang   PetscFunctionBegin;
1362d4002b98SHong Zhang   ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
1363d4002b98SHong Zhang   ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr);
1364d4002b98SHong Zhang   if (isnull) PetscFunctionReturn(0);
1365d4002b98SHong Zhang 
1366d4002b98SHong Zhang   xr   = A->cmap->n; yr  = A->rmap->n; h = yr/10.0; w = xr/10.0;
1367d4002b98SHong Zhang   xr  += w;          yr += h;         xl = -w;     yl = -h;
1368d4002b98SHong Zhang   ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr);
1369d4002b98SHong Zhang   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr);
1370d4002b98SHong Zhang   ierr = PetscDrawZoom(draw,MatView_SeqSELL_Draw_Zoom,A);CHKERRQ(ierr);
1371d4002b98SHong Zhang   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL);CHKERRQ(ierr);
1372d4002b98SHong Zhang   ierr = PetscDrawSave(draw);CHKERRQ(ierr);
1373d4002b98SHong Zhang   PetscFunctionReturn(0);
1374d4002b98SHong Zhang }
1375d4002b98SHong Zhang 
1376d4002b98SHong Zhang PetscErrorCode MatView_SeqSELL(Mat A,PetscViewer viewer)
1377d4002b98SHong Zhang {
1378d4002b98SHong Zhang   PetscBool      iascii,isbinary,isdraw;
1379d4002b98SHong Zhang   PetscErrorCode ierr;
1380d4002b98SHong Zhang 
1381d4002b98SHong Zhang   PetscFunctionBegin;
1382d4002b98SHong Zhang   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
1383d4002b98SHong Zhang   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
1384d4002b98SHong Zhang   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
1385d4002b98SHong Zhang   if (iascii) {
1386d4002b98SHong Zhang     ierr = MatView_SeqSELL_ASCII(A,viewer);CHKERRQ(ierr);
1387d4002b98SHong Zhang   } else if (isbinary) {
1388d4002b98SHong Zhang     /* ierr = MatView_SeqSELL_Binary(A,viewer);CHKERRQ(ierr); */
1389d4002b98SHong Zhang   } else if (isdraw) {
1390d4002b98SHong Zhang     ierr = MatView_SeqSELL_Draw(A,viewer);CHKERRQ(ierr);
1391d4002b98SHong Zhang   }
1392d4002b98SHong Zhang   PetscFunctionReturn(0);
1393d4002b98SHong Zhang }
1394d4002b98SHong Zhang 
1395d4002b98SHong Zhang PetscErrorCode MatAssemblyEnd_SeqSELL(Mat A,MatAssemblyType mode)
1396d4002b98SHong Zhang {
1397d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
1398d4002b98SHong Zhang   PetscInt       i,shift,row_in_slice,row,nrow,*cp,lastcol,j,k;
1399d4002b98SHong Zhang   MatScalar      *vp;
1400d4002b98SHong Zhang   PetscErrorCode ierr;
1401d4002b98SHong Zhang 
1402d4002b98SHong Zhang   PetscFunctionBegin;
1403d4002b98SHong Zhang   if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0);
1404d4002b98SHong Zhang   /* To do: compress out the unused elements */
1405d4002b98SHong Zhang   ierr = MatMarkDiagonal_SeqSELL(A);CHKERRQ(ierr);
1406c0aa6a63SJacob Faibussowitsch   ierr = PetscInfo6(A,"Matrix size: %" PetscInt_FMT " X %" PetscInt_FMT "; storage space: %" PetscInt_FMT " allocated %" PetscInt_FMT " used (%" PetscInt_FMT " nonzeros+%" PetscInt_FMT " paddedzeros)\n",A->rmap->n,A->cmap->n,a->maxallocmat,a->sliidx[a->totalslices],a->nz,a->sliidx[a->totalslices]-a->nz);CHKERRQ(ierr);
1407c0aa6a63SJacob Faibussowitsch   ierr = PetscInfo1(A,"Number of mallocs during MatSetValues() is %" PetscInt_FMT "\n",a->reallocs);CHKERRQ(ierr);
1408c0aa6a63SJacob Faibussowitsch   ierr = PetscInfo1(A,"Maximum nonzeros in any row is %" PetscInt_FMT "\n",a->rlenmax);CHKERRQ(ierr);
1409d4002b98SHong 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 */
1410d4002b98SHong Zhang   for (i=0; i<a->totalslices; ++i) {
1411d4002b98SHong Zhang     shift = a->sliidx[i];    /* starting index of the slice */
1412d4002b98SHong Zhang     cp    = a->colidx+shift; /* pointer to the column indices of the slice */
1413d4002b98SHong Zhang     vp    = a->val+shift;    /* pointer to the nonzero values of the slice */
1414d4002b98SHong Zhang     for (row_in_slice=0; row_in_slice<8; ++row_in_slice) { /* loop over rows in the slice */
1415d4002b98SHong Zhang       row  = 8*i + row_in_slice;
1416d4002b98SHong Zhang       nrow = a->rlen[row]; /* number of nonzeros in row */
1417d4002b98SHong Zhang       /*
1418d4002b98SHong Zhang         Search for the nearest nonzero. Normally setting the index to zero may cause extra communication.
1419d4002b98SHong Zhang         But if the entire slice are empty, it is fine to use 0 since the index will not be loaded.
1420d4002b98SHong Zhang       */
1421d4002b98SHong Zhang       lastcol = 0;
1422d4002b98SHong Zhang       if (nrow>0) { /* nonempty row */
1423d4002b98SHong Zhang         lastcol = cp[8*(nrow-1)+row_in_slice]; /* use the index from the last nonzero at current row */
1424d4002b98SHong Zhang       } else if (!row_in_slice) { /* first row of the currect slice is empty */
1425d4002b98SHong Zhang         for (j=1;j<8;j++) {
1426d4002b98SHong Zhang           if (a->rlen[8*i+j]) {
1427d4002b98SHong Zhang             lastcol = cp[j];
1428d4002b98SHong Zhang             break;
1429d4002b98SHong Zhang           }
1430d4002b98SHong Zhang         }
1431d4002b98SHong Zhang       } else {
1432d4002b98SHong Zhang         if (a->sliidx[i+1] != shift) lastcol = cp[row_in_slice-1]; /* use the index from the previous row */
1433d4002b98SHong Zhang       }
1434d4002b98SHong Zhang 
1435d4002b98SHong Zhang       for (k=nrow; k<(a->sliidx[i+1]-shift)/8; ++k) {
1436d4002b98SHong Zhang         cp[8*k+row_in_slice] = lastcol;
1437d4002b98SHong Zhang         vp[8*k+row_in_slice] = (MatScalar)0;
1438d4002b98SHong Zhang       }
1439d4002b98SHong Zhang     }
1440d4002b98SHong Zhang   }
1441d4002b98SHong Zhang 
1442d4002b98SHong Zhang   A->info.mallocs += a->reallocs;
1443d4002b98SHong Zhang   a->reallocs      = 0;
1444d4002b98SHong Zhang 
1445d4002b98SHong Zhang   ierr = MatSeqSELLInvalidateDiagonal(A);CHKERRQ(ierr);
1446d4002b98SHong Zhang   PetscFunctionReturn(0);
1447d4002b98SHong Zhang }
1448d4002b98SHong Zhang 
1449d4002b98SHong Zhang PetscErrorCode MatGetInfo_SeqSELL(Mat A,MatInfoType flag,MatInfo *info)
1450d4002b98SHong Zhang {
1451d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
1452d4002b98SHong Zhang 
1453d4002b98SHong Zhang   PetscFunctionBegin;
1454d4002b98SHong Zhang   info->block_size   = 1.0;
14553966268fSBarry Smith   info->nz_allocated = a->maxallocmat;
14563966268fSBarry Smith   info->nz_used      = a->sliidx[a->totalslices]; /* include padding zeros */
14573966268fSBarry Smith   info->nz_unneeded  = (a->maxallocmat-a->sliidx[a->totalslices]);
14583966268fSBarry Smith   info->assemblies   = A->num_ass;
14593966268fSBarry Smith   info->mallocs      = A->info.mallocs;
1460d4002b98SHong Zhang   info->memory       = ((PetscObject)A)->mem;
1461d4002b98SHong Zhang   if (A->factortype) {
1462d4002b98SHong Zhang     info->fill_ratio_given  = A->info.fill_ratio_given;
1463d4002b98SHong Zhang     info->fill_ratio_needed = A->info.fill_ratio_needed;
1464d4002b98SHong Zhang     info->factor_mallocs    = A->info.factor_mallocs;
1465d4002b98SHong Zhang   } else {
1466d4002b98SHong Zhang     info->fill_ratio_given  = 0;
1467d4002b98SHong Zhang     info->fill_ratio_needed = 0;
1468d4002b98SHong Zhang     info->factor_mallocs    = 0;
1469d4002b98SHong Zhang   }
1470d4002b98SHong Zhang   PetscFunctionReturn(0);
1471d4002b98SHong Zhang }
1472d4002b98SHong Zhang 
1473d4002b98SHong Zhang PetscErrorCode MatSetValues_SeqSELL(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
1474d4002b98SHong Zhang {
1475d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
1476d4002b98SHong Zhang   PetscInt       shift,i,k,l,low,high,t,ii,row,col,nrow;
1477d4002b98SHong Zhang   PetscInt       *cp,nonew=a->nonew,lastcol=-1;
1478d4002b98SHong Zhang   MatScalar      *vp,value;
1479d4002b98SHong Zhang   PetscErrorCode ierr;
1480d4002b98SHong Zhang 
1481d4002b98SHong Zhang   PetscFunctionBegin;
1482d4002b98SHong Zhang   for (k=0; k<m; k++) { /* loop over added rows */
1483d4002b98SHong Zhang     row = im[k];
1484d4002b98SHong Zhang     if (row < 0) continue;
1485c0aa6a63SJacob Faibussowitsch     if (PetscUnlikelyDebug(row >= A->rmap->n)) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %" PetscInt_FMT " max %" PetscInt_FMT,row,A->rmap->n-1);
1486d4002b98SHong Zhang     shift = a->sliidx[row>>3]+(row&0x07); /* starting index of the row */
1487d4002b98SHong Zhang     cp    = a->colidx+shift; /* pointer to the row */
1488d4002b98SHong Zhang     vp    = a->val+shift; /* pointer to the row */
1489d4002b98SHong Zhang     nrow  = a->rlen[row];
1490d4002b98SHong Zhang     low   = 0;
1491d4002b98SHong Zhang     high  = nrow;
1492d4002b98SHong Zhang 
1493d4002b98SHong Zhang     for (l=0; l<n; l++) { /* loop over added columns */
1494d4002b98SHong Zhang       col = in[l];
1495d4002b98SHong Zhang       if (col<0) continue;
1496c0aa6a63SJacob Faibussowitsch       if (PetscUnlikelyDebug(col >= A->cmap->n)) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Col too large: row %" PetscInt_FMT " max %" PetscInt_FMT,col,A->cmap->n-1);
1497d4002b98SHong Zhang       if (a->roworiented) {
1498d4002b98SHong Zhang         value = v[l+k*n];
1499d4002b98SHong Zhang       } else {
1500d4002b98SHong Zhang         value = v[k+l*m];
1501d4002b98SHong Zhang       }
1502d4002b98SHong Zhang       if ((value == 0.0 && a->ignorezeroentries) && (is == ADD_VALUES)) continue;
1503d4002b98SHong Zhang 
1504d4002b98SHong Zhang       /* search in this row for the specified colmun, i indicates the column to be set */
1505d4002b98SHong Zhang       if (col <= lastcol) low = 0;
1506d4002b98SHong Zhang       else high = nrow;
1507d4002b98SHong Zhang       lastcol = col;
1508d4002b98SHong Zhang       while (high-low > 5) {
1509d4002b98SHong Zhang         t = (low+high)/2;
1510d4002b98SHong Zhang         if (*(cp+t*8) > col) high = t;
1511d4002b98SHong Zhang         else low = t;
1512d4002b98SHong Zhang       }
1513d4002b98SHong Zhang       for (i=low; i<high; i++) {
1514d4002b98SHong Zhang         if (*(cp+i*8) > col) break;
1515d4002b98SHong Zhang         if (*(cp+i*8) == col) {
1516d4002b98SHong Zhang           if (is == ADD_VALUES) *(vp+i*8) += value;
1517d4002b98SHong Zhang           else *(vp+i*8) = value;
1518d4002b98SHong Zhang           low = i + 1;
1519d4002b98SHong Zhang           goto noinsert;
1520d4002b98SHong Zhang         }
1521d4002b98SHong Zhang       }
1522d4002b98SHong Zhang       if (value == 0.0 && a->ignorezeroentries) goto noinsert;
1523d4002b98SHong Zhang       if (nonew == 1) goto noinsert;
1524c0aa6a63SJacob Faibussowitsch       if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%" PetscInt_FMT ", %" PetscInt_FMT ") in the matrix", row, col);
1525d4002b98SHong Zhang       /* If the current row length exceeds the slice width (e.g. nrow==slice_width), allocate a new space, otherwise do nothing */
1526d4002b98SHong Zhang       MatSeqXSELLReallocateSELL(A,A->rmap->n,1,nrow,a->sliidx,row/8,row,col,a->colidx,a->val,cp,vp,nonew,MatScalar);
1527d4002b98SHong Zhang       /* add the new nonzero to the high position, shift the remaining elements in current row to the right by one slot */
1528d4002b98SHong Zhang       for (ii=nrow-1; ii>=i; ii--) {
1529d4002b98SHong Zhang         *(cp+(ii+1)*8) = *(cp+ii*8);
1530d4002b98SHong Zhang         *(vp+(ii+1)*8) = *(vp+ii*8);
1531d4002b98SHong Zhang       }
1532d4002b98SHong Zhang       a->rlen[row]++;
1533d4002b98SHong Zhang       *(cp+i*8) = col;
1534d4002b98SHong Zhang       *(vp+i*8) = value;
1535d4002b98SHong Zhang       a->nz++;
1536d4002b98SHong Zhang       A->nonzerostate++;
1537d4002b98SHong Zhang       low = i+1; high++; nrow++;
1538d4002b98SHong Zhang noinsert:;
1539d4002b98SHong Zhang     }
1540d4002b98SHong Zhang     a->rlen[row] = nrow;
1541d4002b98SHong Zhang   }
1542d4002b98SHong Zhang   PetscFunctionReturn(0);
1543d4002b98SHong Zhang }
1544d4002b98SHong Zhang 
1545d4002b98SHong Zhang PetscErrorCode MatCopy_SeqSELL(Mat A,Mat B,MatStructure str)
1546d4002b98SHong Zhang {
1547d4002b98SHong Zhang   PetscErrorCode ierr;
1548d4002b98SHong Zhang 
1549d4002b98SHong Zhang   PetscFunctionBegin;
1550d4002b98SHong Zhang   /* If the two matrices have the same copy implementation, use fast copy. */
1551d4002b98SHong Zhang   if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) {
1552d4002b98SHong Zhang     Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
1553d4002b98SHong Zhang     Mat_SeqSELL *b=(Mat_SeqSELL*)B->data;
1554d4002b98SHong Zhang 
1555d4002b98SHong Zhang     if (a->sliidx[a->totalslices] != b->sliidx[b->totalslices]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Number of nonzeros in two matrices are different");
1556580bdb30SBarry Smith     ierr = PetscArraycpy(b->val,a->val,a->sliidx[a->totalslices]);CHKERRQ(ierr);
1557d4002b98SHong Zhang   } else {
1558d4002b98SHong Zhang     ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr);
1559d4002b98SHong Zhang   }
1560d4002b98SHong Zhang   PetscFunctionReturn(0);
1561d4002b98SHong Zhang }
1562d4002b98SHong Zhang 
1563d4002b98SHong Zhang PetscErrorCode MatSetUp_SeqSELL(Mat A)
1564d4002b98SHong Zhang {
1565d4002b98SHong Zhang   PetscErrorCode ierr;
1566d4002b98SHong Zhang 
1567d4002b98SHong Zhang   PetscFunctionBegin;
1568f4259b30SLisandro Dalcin   ierr = MatSeqSELLSetPreallocation(A,PETSC_DEFAULT,NULL);CHKERRQ(ierr);
1569d4002b98SHong Zhang   PetscFunctionReturn(0);
1570d4002b98SHong Zhang }
1571d4002b98SHong Zhang 
1572d4002b98SHong Zhang PetscErrorCode MatSeqSELLGetArray_SeqSELL(Mat A,PetscScalar *array[])
1573d4002b98SHong Zhang {
1574d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
1575d4002b98SHong Zhang 
1576d4002b98SHong Zhang   PetscFunctionBegin;
1577d4002b98SHong Zhang   *array = a->val;
1578d4002b98SHong Zhang   PetscFunctionReturn(0);
1579d4002b98SHong Zhang }
1580d4002b98SHong Zhang 
1581d4002b98SHong Zhang PetscErrorCode MatSeqSELLRestoreArray_SeqSELL(Mat A,PetscScalar *array[])
1582d4002b98SHong Zhang {
1583d4002b98SHong Zhang   PetscFunctionBegin;
1584d4002b98SHong Zhang   PetscFunctionReturn(0);
1585d4002b98SHong Zhang }
1586d4002b98SHong Zhang 
1587d4002b98SHong Zhang PetscErrorCode MatRealPart_SeqSELL(Mat A)
1588d4002b98SHong Zhang {
1589d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
1590d4002b98SHong Zhang   PetscInt    i;
1591d4002b98SHong Zhang   MatScalar   *aval=a->val;
1592d4002b98SHong Zhang 
1593d4002b98SHong Zhang   PetscFunctionBegin;
1594d4002b98SHong Zhang   for (i=0; i<a->sliidx[a->totalslices]; i++) aval[i]=PetscRealPart(aval[i]);
1595d4002b98SHong Zhang   PetscFunctionReturn(0);
1596d4002b98SHong Zhang }
1597d4002b98SHong Zhang 
1598d4002b98SHong Zhang PetscErrorCode MatImaginaryPart_SeqSELL(Mat A)
1599d4002b98SHong Zhang {
1600d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)A->data;
1601d4002b98SHong Zhang   PetscInt       i;
1602d4002b98SHong Zhang   MatScalar      *aval=a->val;
1603d4002b98SHong Zhang   PetscErrorCode ierr;
1604d4002b98SHong Zhang 
1605d4002b98SHong Zhang   PetscFunctionBegin;
1606d4002b98SHong Zhang   for (i=0; i<a->sliidx[a->totalslices]; i++) aval[i] = PetscImaginaryPart(aval[i]);
1607d4002b98SHong Zhang   ierr = MatSeqSELLInvalidateDiagonal(A);CHKERRQ(ierr);
1608d4002b98SHong Zhang   PetscFunctionReturn(0);
1609d4002b98SHong Zhang }
1610d4002b98SHong Zhang 
1611d4002b98SHong Zhang PetscErrorCode MatScale_SeqSELL(Mat inA,PetscScalar alpha)
1612d4002b98SHong Zhang {
1613d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)inA->data;
1614d4002b98SHong Zhang   MatScalar      *aval=a->val;
1615d4002b98SHong Zhang   PetscScalar    oalpha=alpha;
1616d4002b98SHong Zhang   PetscBLASInt   one=1,size;
1617d4002b98SHong Zhang   PetscErrorCode ierr;
1618d4002b98SHong Zhang 
1619d4002b98SHong Zhang   PetscFunctionBegin;
1620d4002b98SHong Zhang   ierr = PetscBLASIntCast(a->sliidx[a->totalslices],&size);CHKERRQ(ierr);
1621d4002b98SHong Zhang   PetscStackCallBLAS("BLASscal",BLASscal_(&size,&oalpha,aval,&one));
1622d4002b98SHong Zhang   ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
1623d4002b98SHong Zhang   ierr = MatSeqSELLInvalidateDiagonal(inA);CHKERRQ(ierr);
1624d4002b98SHong Zhang   PetscFunctionReturn(0);
1625d4002b98SHong Zhang }
1626d4002b98SHong Zhang 
1627d4002b98SHong Zhang PetscErrorCode MatShift_SeqSELL(Mat Y,PetscScalar a)
1628d4002b98SHong Zhang {
1629d4002b98SHong Zhang   Mat_SeqSELL    *y=(Mat_SeqSELL*)Y->data;
1630d4002b98SHong Zhang   PetscErrorCode ierr;
1631d4002b98SHong Zhang 
1632d4002b98SHong Zhang   PetscFunctionBegin;
1633d4002b98SHong Zhang   if (!Y->preallocated || !y->nz) {
1634d4002b98SHong Zhang     ierr = MatSeqSELLSetPreallocation(Y,1,NULL);CHKERRQ(ierr);
1635d4002b98SHong Zhang   }
1636d4002b98SHong Zhang   ierr = MatShift_Basic(Y,a);CHKERRQ(ierr);
1637d4002b98SHong Zhang   PetscFunctionReturn(0);
1638d4002b98SHong Zhang }
1639d4002b98SHong Zhang 
1640d4002b98SHong Zhang PetscErrorCode MatSOR_SeqSELL(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx)
1641d4002b98SHong Zhang {
1642d4002b98SHong Zhang   Mat_SeqSELL       *a=(Mat_SeqSELL*)A->data;
1643d4002b98SHong Zhang   PetscScalar       *x,sum,*t;
1644f4259b30SLisandro Dalcin   const MatScalar   *idiag=NULL,*mdiag;
1645d4002b98SHong Zhang   const PetscScalar *b,*xb;
1646d4002b98SHong Zhang   PetscInt          n,m=A->rmap->n,i,j,shift;
1647d4002b98SHong Zhang   const PetscInt    *diag;
1648d4002b98SHong Zhang   PetscErrorCode    ierr;
1649d4002b98SHong Zhang 
1650d4002b98SHong Zhang   PetscFunctionBegin;
1651d4002b98SHong Zhang   its = its*lits;
1652d4002b98SHong Zhang 
1653d4002b98SHong Zhang   if (fshift != a->fshift || omega != a->omega) a->idiagvalid = PETSC_FALSE; /* must recompute idiag[] */
1654d4002b98SHong Zhang   if (!a->idiagvalid) {ierr = MatInvertDiagonal_SeqSELL(A,omega,fshift);CHKERRQ(ierr);}
1655d4002b98SHong Zhang   a->fshift = fshift;
1656d4002b98SHong Zhang   a->omega  = omega;
1657d4002b98SHong Zhang 
1658d4002b98SHong Zhang   diag  = a->diag;
1659d4002b98SHong Zhang   t     = a->ssor_work;
1660d4002b98SHong Zhang   idiag = a->idiag;
1661d4002b98SHong Zhang   mdiag = a->mdiag;
1662d4002b98SHong Zhang 
1663d4002b98SHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1664d4002b98SHong Zhang   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
1665d4002b98SHong Zhang   /* We count flops by assuming the upper triangular and lower triangular parts have the same number of nonzeros */
1666d4002b98SHong Zhang   if (flag == SOR_APPLY_UPPER) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"SOR_APPLY_UPPER is not implemented");
1667d4002b98SHong Zhang   if (flag == SOR_APPLY_LOWER) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"SOR_APPLY_LOWER is not implemented");
1668d4002b98SHong Zhang   if (flag & SOR_EISENSTAT) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support yet for Eisenstat");
1669d4002b98SHong Zhang 
1670d4002b98SHong Zhang   if (flag & SOR_ZERO_INITIAL_GUESS) {
1671d4002b98SHong Zhang     if ((flag & SOR_FORWARD_SWEEP) || (flag & SOR_LOCAL_FORWARD_SWEEP)) {
1672d4002b98SHong Zhang       for (i=0; i<m; i++) {
1673d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
1674d4002b98SHong Zhang         sum   = b[i];
1675d4002b98SHong Zhang         n     = (diag[i]-shift)/8;
1676d4002b98SHong Zhang         for (j=0; j<n; j++) sum -= a->val[shift+j*8]*x[a->colidx[shift+j*8]];
1677d4002b98SHong Zhang         t[i]  = sum;
1678d4002b98SHong Zhang         x[i]  = sum*idiag[i];
1679d4002b98SHong Zhang       }
1680d4002b98SHong Zhang       xb   = t;
1681d4002b98SHong Zhang       ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
1682d4002b98SHong Zhang     } else xb = b;
1683d4002b98SHong Zhang     if ((flag & SOR_BACKWARD_SWEEP) || (flag & SOR_LOCAL_BACKWARD_SWEEP)) {
1684d4002b98SHong Zhang       for (i=m-1; i>=0; i--) {
1685d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
1686d4002b98SHong Zhang         sum   = xb[i];
1687d4002b98SHong Zhang         n     = a->rlen[i]-(diag[i]-shift)/8-1;
1688d4002b98SHong Zhang         for (j=1; j<=n; j++) sum -= a->val[diag[i]+j*8]*x[a->colidx[diag[i]+j*8]];
1689d4002b98SHong Zhang         if (xb == b) {
1690d4002b98SHong Zhang           x[i] = sum*idiag[i];
1691d4002b98SHong Zhang         } else {
1692d4002b98SHong Zhang           x[i] = (1.-omega)*x[i]+sum*idiag[i];  /* omega in idiag */
1693d4002b98SHong Zhang         }
1694d4002b98SHong Zhang       }
1695d4002b98SHong Zhang       ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); /* assumes 1/2 in upper */
1696d4002b98SHong Zhang     }
1697d4002b98SHong Zhang     its--;
1698d4002b98SHong Zhang   }
1699d4002b98SHong Zhang   while (its--) {
1700d4002b98SHong Zhang     if ((flag & SOR_FORWARD_SWEEP) || (flag & SOR_LOCAL_FORWARD_SWEEP)) {
1701d4002b98SHong Zhang       for (i=0; i<m; i++) {
1702d4002b98SHong Zhang         /* lower */
1703d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
1704d4002b98SHong Zhang         sum   = b[i];
1705d4002b98SHong Zhang         n     = (diag[i]-shift)/8;
1706d4002b98SHong Zhang         for (j=0; j<n; j++) sum -= a->val[shift+j*8]*x[a->colidx[shift+j*8]];
1707d4002b98SHong Zhang         t[i]  = sum;             /* save application of the lower-triangular part */
1708d4002b98SHong Zhang         /* upper */
1709d4002b98SHong Zhang         n     = a->rlen[i]-(diag[i]-shift)/8-1;
1710d4002b98SHong Zhang         for (j=1; j<=n; j++) sum -= a->val[diag[i]+j*8]*x[a->colidx[diag[i]+j*8]];
1711d4002b98SHong Zhang         x[i]  = (1.-omega)*x[i]+sum*idiag[i];  /* omega in idiag */
1712d4002b98SHong Zhang       }
1713d4002b98SHong Zhang       xb   = t;
1714d4002b98SHong Zhang       ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
1715d4002b98SHong Zhang     } else xb = b;
1716d4002b98SHong Zhang     if ((flag & SOR_BACKWARD_SWEEP) || (flag & SOR_LOCAL_BACKWARD_SWEEP)) {
1717d4002b98SHong Zhang       for (i=m-1; i>=0; i--) {
1718d4002b98SHong Zhang         shift = a->sliidx[i>>3]+(i&0x07); /* starting index of the row i */
1719d4002b98SHong Zhang         sum = xb[i];
1720d4002b98SHong Zhang         if (xb == b) {
1721d4002b98SHong Zhang           /* whole matrix (no checkpointing available) */
1722d4002b98SHong Zhang           n     = a->rlen[i];
1723d4002b98SHong Zhang           for (j=0; j<n; j++) sum -= a->val[shift+j*8]*x[a->colidx[shift+j*8]];
1724d4002b98SHong Zhang           x[i] = (1.-omega)*x[i]+(sum+mdiag[i]*x[i])*idiag[i];
1725d4002b98SHong Zhang         } else { /* lower-triangular part has been saved, so only apply upper-triangular */
1726d4002b98SHong Zhang           n     = a->rlen[i]-(diag[i]-shift)/8-1;
1727d4002b98SHong Zhang           for (j=1; j<=n; j++) sum -= a->val[diag[i]+j*8]*x[a->colidx[diag[i]+j*8]];
1728d4002b98SHong Zhang           x[i]  = (1.-omega)*x[i]+sum*idiag[i];  /* omega in idiag */
1729d4002b98SHong Zhang         }
1730d4002b98SHong Zhang       }
1731d4002b98SHong Zhang       if (xb == b) {
1732d4002b98SHong Zhang         ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
1733d4002b98SHong Zhang       } else {
1734d4002b98SHong Zhang         ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); /* assumes 1/2 in upper */
1735d4002b98SHong Zhang       }
1736d4002b98SHong Zhang     }
1737d4002b98SHong Zhang   }
1738d4002b98SHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1739d4002b98SHong Zhang   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1740d4002b98SHong Zhang   PetscFunctionReturn(0);
1741d4002b98SHong Zhang }
1742d4002b98SHong Zhang 
1743d4002b98SHong Zhang /* -------------------------------------------------------------------*/
1744d4002b98SHong Zhang static struct _MatOps MatOps_Values = {MatSetValues_SeqSELL,
17456108893eSStefano Zampini                                        MatGetRow_SeqSELL,
17466108893eSStefano Zampini                                        MatRestoreRow_SeqSELL,
1747d4002b98SHong Zhang                                        MatMult_SeqSELL,
1748d4002b98SHong Zhang                                /* 4*/  MatMultAdd_SeqSELL,
1749d4002b98SHong Zhang                                        MatMultTranspose_SeqSELL,
1750d4002b98SHong Zhang                                        MatMultTransposeAdd_SeqSELL,
1751f4259b30SLisandro Dalcin                                        NULL,
1752f4259b30SLisandro Dalcin                                        NULL,
1753f4259b30SLisandro Dalcin                                        NULL,
1754f4259b30SLisandro Dalcin                                /* 10*/ NULL,
1755f4259b30SLisandro Dalcin                                        NULL,
1756f4259b30SLisandro Dalcin                                        NULL,
1757d4002b98SHong Zhang                                        MatSOR_SeqSELL,
1758f4259b30SLisandro Dalcin                                        NULL,
1759d4002b98SHong Zhang                                /* 15*/ MatGetInfo_SeqSELL,
1760d4002b98SHong Zhang                                        MatEqual_SeqSELL,
1761d4002b98SHong Zhang                                        MatGetDiagonal_SeqSELL,
1762d4002b98SHong Zhang                                        MatDiagonalScale_SeqSELL,
1763f4259b30SLisandro Dalcin                                        NULL,
1764f4259b30SLisandro Dalcin                                /* 20*/ NULL,
1765d4002b98SHong Zhang                                        MatAssemblyEnd_SeqSELL,
1766d4002b98SHong Zhang                                        MatSetOption_SeqSELL,
1767d4002b98SHong Zhang                                        MatZeroEntries_SeqSELL,
1768f4259b30SLisandro Dalcin                                /* 24*/ NULL,
1769f4259b30SLisandro Dalcin                                        NULL,
1770f4259b30SLisandro Dalcin                                        NULL,
1771f4259b30SLisandro Dalcin                                        NULL,
1772f4259b30SLisandro Dalcin                                        NULL,
1773d4002b98SHong Zhang                                /* 29*/ MatSetUp_SeqSELL,
1774f4259b30SLisandro Dalcin                                        NULL,
1775f4259b30SLisandro Dalcin                                        NULL,
1776f4259b30SLisandro Dalcin                                        NULL,
1777f4259b30SLisandro Dalcin                                        NULL,
1778d4002b98SHong Zhang                                /* 34*/ MatDuplicate_SeqSELL,
1779f4259b30SLisandro Dalcin                                        NULL,
1780f4259b30SLisandro Dalcin                                        NULL,
1781f4259b30SLisandro Dalcin                                        NULL,
1782f4259b30SLisandro Dalcin                                        NULL,
1783f4259b30SLisandro Dalcin                                /* 39*/ NULL,
1784f4259b30SLisandro Dalcin                                        NULL,
1785f4259b30SLisandro Dalcin                                        NULL,
1786d4002b98SHong Zhang                                        MatGetValues_SeqSELL,
1787d4002b98SHong Zhang                                        MatCopy_SeqSELL,
1788f4259b30SLisandro Dalcin                                /* 44*/ NULL,
1789d4002b98SHong Zhang                                        MatScale_SeqSELL,
1790d4002b98SHong Zhang                                        MatShift_SeqSELL,
1791f4259b30SLisandro Dalcin                                        NULL,
1792f4259b30SLisandro Dalcin                                        NULL,
1793f4259b30SLisandro Dalcin                                /* 49*/ NULL,
1794f4259b30SLisandro Dalcin                                        NULL,
1795f4259b30SLisandro Dalcin                                        NULL,
1796f4259b30SLisandro Dalcin                                        NULL,
1797f4259b30SLisandro Dalcin                                        NULL,
1798d4002b98SHong Zhang                                /* 54*/ MatFDColoringCreate_SeqXAIJ,
1799f4259b30SLisandro Dalcin                                        NULL,
1800f4259b30SLisandro Dalcin                                        NULL,
1801f4259b30SLisandro Dalcin                                        NULL,
1802f4259b30SLisandro Dalcin                                        NULL,
1803f4259b30SLisandro Dalcin                                /* 59*/ NULL,
1804d4002b98SHong Zhang                                        MatDestroy_SeqSELL,
1805d4002b98SHong Zhang                                        MatView_SeqSELL,
1806f4259b30SLisandro Dalcin                                        NULL,
1807f4259b30SLisandro Dalcin                                        NULL,
1808f4259b30SLisandro Dalcin                                /* 64*/ NULL,
1809f4259b30SLisandro Dalcin                                        NULL,
1810f4259b30SLisandro Dalcin                                        NULL,
1811f4259b30SLisandro Dalcin                                        NULL,
1812f4259b30SLisandro Dalcin                                        NULL,
1813f4259b30SLisandro Dalcin                                /* 69*/ NULL,
1814f4259b30SLisandro Dalcin                                        NULL,
1815f4259b30SLisandro Dalcin                                        NULL,
1816f4259b30SLisandro Dalcin                                        NULL,
1817f4259b30SLisandro Dalcin                                        NULL,
1818f4259b30SLisandro Dalcin                                /* 74*/ NULL,
1819d4002b98SHong Zhang                                        MatFDColoringApply_AIJ, /* reuse the FDColoring function for AIJ */
1820f4259b30SLisandro Dalcin                                        NULL,
1821f4259b30SLisandro Dalcin                                        NULL,
1822f4259b30SLisandro Dalcin                                        NULL,
1823f4259b30SLisandro Dalcin                                /* 79*/ NULL,
1824f4259b30SLisandro Dalcin                                        NULL,
1825f4259b30SLisandro Dalcin                                        NULL,
1826f4259b30SLisandro Dalcin                                        NULL,
1827f4259b30SLisandro Dalcin                                        NULL,
1828f4259b30SLisandro Dalcin                                /* 84*/ NULL,
1829f4259b30SLisandro Dalcin                                        NULL,
1830f4259b30SLisandro Dalcin                                        NULL,
1831f4259b30SLisandro Dalcin                                        NULL,
1832f4259b30SLisandro Dalcin                                        NULL,
1833f4259b30SLisandro Dalcin                                /* 89*/ NULL,
1834f4259b30SLisandro Dalcin                                        NULL,
1835f4259b30SLisandro Dalcin                                        NULL,
1836f4259b30SLisandro Dalcin                                        NULL,
1837f4259b30SLisandro Dalcin                                        NULL,
1838f4259b30SLisandro Dalcin                                /* 94*/ NULL,
1839f4259b30SLisandro Dalcin                                        NULL,
1840f4259b30SLisandro Dalcin                                        NULL,
1841f4259b30SLisandro Dalcin                                        NULL,
1842f4259b30SLisandro Dalcin                                        NULL,
1843f4259b30SLisandro Dalcin                                /* 99*/ NULL,
1844f4259b30SLisandro Dalcin                                        NULL,
1845f4259b30SLisandro Dalcin                                        NULL,
1846d4002b98SHong Zhang                                        MatConjugate_SeqSELL,
1847f4259b30SLisandro Dalcin                                        NULL,
1848f4259b30SLisandro Dalcin                                /*104*/ NULL,
1849f4259b30SLisandro Dalcin                                        NULL,
1850f4259b30SLisandro Dalcin                                        NULL,
1851f4259b30SLisandro Dalcin                                        NULL,
1852f4259b30SLisandro Dalcin                                        NULL,
1853f4259b30SLisandro Dalcin                                /*109*/ NULL,
1854f4259b30SLisandro Dalcin                                        NULL,
1855f4259b30SLisandro Dalcin                                        NULL,
1856f4259b30SLisandro Dalcin                                        NULL,
1857d4002b98SHong Zhang                                        MatMissingDiagonal_SeqSELL,
1858f4259b30SLisandro Dalcin                                /*114*/ NULL,
1859f4259b30SLisandro Dalcin                                        NULL,
1860f4259b30SLisandro Dalcin                                        NULL,
1861f4259b30SLisandro Dalcin                                        NULL,
1862f4259b30SLisandro Dalcin                                        NULL,
1863f4259b30SLisandro Dalcin                                /*119*/ NULL,
1864f4259b30SLisandro Dalcin                                        NULL,
1865f4259b30SLisandro Dalcin                                        NULL,
1866f4259b30SLisandro Dalcin                                        NULL,
1867f4259b30SLisandro Dalcin                                        NULL,
1868f4259b30SLisandro Dalcin                                /*124*/ NULL,
1869f4259b30SLisandro Dalcin                                        NULL,
1870f4259b30SLisandro Dalcin                                        NULL,
1871f4259b30SLisandro Dalcin                                        NULL,
1872f4259b30SLisandro Dalcin                                        NULL,
1873f4259b30SLisandro Dalcin                                /*129*/ NULL,
1874f4259b30SLisandro Dalcin                                        NULL,
1875f4259b30SLisandro Dalcin                                        NULL,
1876f4259b30SLisandro Dalcin                                        NULL,
1877f4259b30SLisandro Dalcin                                        NULL,
1878f4259b30SLisandro Dalcin                                /*134*/ NULL,
1879f4259b30SLisandro Dalcin                                        NULL,
1880f4259b30SLisandro Dalcin                                        NULL,
1881f4259b30SLisandro Dalcin                                        NULL,
1882f4259b30SLisandro Dalcin                                        NULL,
1883f4259b30SLisandro Dalcin                                /*139*/ NULL,
1884f4259b30SLisandro Dalcin                                        NULL,
1885f4259b30SLisandro Dalcin                                        NULL,
1886d4002b98SHong Zhang                                        MatFDColoringSetUp_SeqXAIJ,
1887f4259b30SLisandro Dalcin                                        NULL,
1888f4259b30SLisandro Dalcin                                /*144*/ NULL
1889d4002b98SHong Zhang };
1890d4002b98SHong Zhang 
1891d4002b98SHong Zhang PetscErrorCode MatStoreValues_SeqSELL(Mat mat)
1892d4002b98SHong Zhang {
1893d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)mat->data;
1894d4002b98SHong Zhang   PetscErrorCode ierr;
1895d4002b98SHong Zhang 
1896d4002b98SHong Zhang   PetscFunctionBegin;
1897d4002b98SHong Zhang   if (!a->nonew) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
1898d4002b98SHong Zhang 
1899d4002b98SHong Zhang   /* allocate space for values if not already there */
1900d4002b98SHong Zhang   if (!a->saved_values) {
1901d4002b98SHong Zhang     ierr = PetscMalloc1(a->sliidx[a->totalslices]+1,&a->saved_values);CHKERRQ(ierr);
1902d4002b98SHong Zhang     ierr = PetscLogObjectMemory((PetscObject)mat,(a->sliidx[a->totalslices]+1)*sizeof(PetscScalar));CHKERRQ(ierr);
1903d4002b98SHong Zhang   }
1904d4002b98SHong Zhang 
1905d4002b98SHong Zhang   /* copy values over */
1906580bdb30SBarry Smith   ierr = PetscArraycpy(a->saved_values,a->val,a->sliidx[a->totalslices]);CHKERRQ(ierr);
1907d4002b98SHong Zhang   PetscFunctionReturn(0);
1908d4002b98SHong Zhang }
1909d4002b98SHong Zhang 
1910d4002b98SHong Zhang PetscErrorCode MatRetrieveValues_SeqSELL(Mat mat)
1911d4002b98SHong Zhang {
1912d4002b98SHong Zhang   Mat_SeqSELL    *a=(Mat_SeqSELL*)mat->data;
1913d4002b98SHong Zhang   PetscErrorCode ierr;
1914d4002b98SHong Zhang 
1915d4002b98SHong Zhang   PetscFunctionBegin;
1916d4002b98SHong Zhang   if (!a->nonew) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
1917d4002b98SHong Zhang   if (!a->saved_values) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first");
1918580bdb30SBarry Smith   ierr = PetscArraycpy(a->val,a->saved_values,a->sliidx[a->totalslices]);CHKERRQ(ierr);
1919d4002b98SHong Zhang   PetscFunctionReturn(0);
1920d4002b98SHong Zhang }
1921d4002b98SHong Zhang 
1922d4002b98SHong Zhang /*@C
1923d4002b98SHong Zhang  MatSeqSELLRestoreArray - returns access to the array where the data for a MATSEQSELL matrix is stored obtained by MatSeqSELLGetArray()
1924d4002b98SHong Zhang 
1925d4002b98SHong Zhang  Not Collective
1926d4002b98SHong Zhang 
1927d4002b98SHong Zhang  Input Parameters:
1928d4002b98SHong Zhang  .  mat - a MATSEQSELL matrix
1929d4002b98SHong Zhang  .  array - pointer to the data
1930d4002b98SHong Zhang 
1931d4002b98SHong Zhang  Level: intermediate
1932d4002b98SHong Zhang 
1933d4002b98SHong Zhang  .seealso: MatSeqSELLGetArray(), MatSeqSELLRestoreArrayF90()
1934d4002b98SHong Zhang  @*/
1935d4002b98SHong Zhang PetscErrorCode MatSeqSELLRestoreArray(Mat A,PetscScalar **array)
1936d4002b98SHong Zhang {
1937d4002b98SHong Zhang   PetscErrorCode ierr;
1938d4002b98SHong Zhang 
1939d4002b98SHong Zhang   PetscFunctionBegin;
1940d4002b98SHong Zhang   ierr = PetscUseMethod(A,"MatSeqSELLRestoreArray_C",(Mat,PetscScalar**),(A,array));CHKERRQ(ierr);
1941d4002b98SHong Zhang   PetscFunctionReturn(0);
1942d4002b98SHong Zhang }
1943d4002b98SHong Zhang 
1944d4002b98SHong Zhang PETSC_EXTERN PetscErrorCode MatCreate_SeqSELL(Mat B)
1945d4002b98SHong Zhang {
1946d4002b98SHong Zhang   Mat_SeqSELL    *b;
1947d4002b98SHong Zhang   PetscMPIInt    size;
1948d4002b98SHong Zhang   PetscErrorCode ierr;
1949d4002b98SHong Zhang 
1950d4002b98SHong Zhang   PetscFunctionBegin;
1951ffc4695bSBarry Smith   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)B),&size);CHKERRMPI(ierr);
1952d4002b98SHong Zhang   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Comm must be of size 1");
1953d4002b98SHong Zhang 
1954d4002b98SHong Zhang   ierr = PetscNewLog(B,&b);CHKERRQ(ierr);
1955d4002b98SHong Zhang 
1956d4002b98SHong Zhang   B->data = (void*)b;
1957d4002b98SHong Zhang 
1958d4002b98SHong Zhang   ierr = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
1959d4002b98SHong Zhang 
1960f4259b30SLisandro Dalcin   b->row                = NULL;
1961f4259b30SLisandro Dalcin   b->col                = NULL;
1962f4259b30SLisandro Dalcin   b->icol               = NULL;
1963d4002b98SHong Zhang   b->reallocs           = 0;
1964d4002b98SHong Zhang   b->ignorezeroentries  = PETSC_FALSE;
1965d4002b98SHong Zhang   b->roworiented        = PETSC_TRUE;
1966d4002b98SHong Zhang   b->nonew              = 0;
1967f4259b30SLisandro Dalcin   b->diag               = NULL;
1968f4259b30SLisandro Dalcin   b->solve_work         = NULL;
1969f4259b30SLisandro Dalcin   B->spptr              = NULL;
1970f4259b30SLisandro Dalcin   b->saved_values       = NULL;
1971f4259b30SLisandro Dalcin   b->idiag              = NULL;
1972f4259b30SLisandro Dalcin   b->mdiag              = NULL;
1973f4259b30SLisandro Dalcin   b->ssor_work          = NULL;
1974d4002b98SHong Zhang   b->omega              = 1.0;
1975d4002b98SHong Zhang   b->fshift             = 0.0;
1976d4002b98SHong Zhang   b->idiagvalid         = PETSC_FALSE;
1977d4002b98SHong Zhang   b->keepnonzeropattern = PETSC_FALSE;
1978d4002b98SHong Zhang 
1979d4002b98SHong Zhang   ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSELL);CHKERRQ(ierr);
1980d4002b98SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSELLGetArray_C",MatSeqSELLGetArray_SeqSELL);CHKERRQ(ierr);
1981d4002b98SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSELLRestoreArray_C",MatSeqSELLRestoreArray_SeqSELL);CHKERRQ(ierr);
1982d4002b98SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_SeqSELL);CHKERRQ(ierr);
1983d4002b98SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_SeqSELL);CHKERRQ(ierr);
1984d4002b98SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSELLSetPreallocation_C",MatSeqSELLSetPreallocation_SeqSELL);CHKERRQ(ierr);
1985d4002b98SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsell_seqaij_C",MatConvert_SeqSELL_SeqAIJ);CHKERRQ(ierr);
1986d4002b98SHong Zhang   PetscFunctionReturn(0);
1987d4002b98SHong Zhang }
1988d4002b98SHong Zhang 
1989d4002b98SHong Zhang /*
1990d4002b98SHong Zhang  Given a matrix generated with MatGetFactor() duplicates all the information in A into B
1991d4002b98SHong Zhang  */
1992d4002b98SHong Zhang PetscErrorCode MatDuplicateNoCreate_SeqSELL(Mat C,Mat A,MatDuplicateOption cpvalues,PetscBool mallocmatspace)
1993d4002b98SHong Zhang {
1994d4002b98SHong Zhang   Mat_SeqSELL    *c,*a=(Mat_SeqSELL*)A->data;
1995d4002b98SHong Zhang   PetscInt       i,m=A->rmap->n;
1996d4002b98SHong Zhang   PetscInt       totalslices=a->totalslices;
1997d4002b98SHong Zhang   PetscErrorCode ierr;
1998d4002b98SHong Zhang 
1999d4002b98SHong Zhang   PetscFunctionBegin;
2000d4002b98SHong Zhang   c = (Mat_SeqSELL*)C->data;
2001d4002b98SHong Zhang 
2002d4002b98SHong Zhang   C->factortype = A->factortype;
2003f4259b30SLisandro Dalcin   c->row        = NULL;
2004f4259b30SLisandro Dalcin   c->col        = NULL;
2005f4259b30SLisandro Dalcin   c->icol       = NULL;
2006d4002b98SHong Zhang   c->reallocs   = 0;
2007d4002b98SHong Zhang 
2008d4002b98SHong Zhang   C->assembled = PETSC_TRUE;
2009d4002b98SHong Zhang 
2010d4002b98SHong Zhang   ierr = PetscLayoutReference(A->rmap,&C->rmap);CHKERRQ(ierr);
2011d4002b98SHong Zhang   ierr = PetscLayoutReference(A->cmap,&C->cmap);CHKERRQ(ierr);
2012d4002b98SHong Zhang 
2013d4002b98SHong Zhang   ierr = PetscMalloc1(8*totalslices,&c->rlen);CHKERRQ(ierr);
2014d4002b98SHong Zhang   ierr = PetscLogObjectMemory((PetscObject)C,m*sizeof(PetscInt));CHKERRQ(ierr);
2015d4002b98SHong Zhang   ierr = PetscMalloc1(totalslices+1,&c->sliidx);CHKERRQ(ierr);
2016d4002b98SHong Zhang   ierr = PetscLogObjectMemory((PetscObject)C, (totalslices+1)*sizeof(PetscInt));CHKERRQ(ierr);
2017d4002b98SHong Zhang 
2018d4002b98SHong Zhang   for (i=0; i<m; i++) c->rlen[i] = a->rlen[i];
2019d4002b98SHong Zhang   for (i=0; i<totalslices+1; i++) c->sliidx[i] = a->sliidx[i];
2020d4002b98SHong Zhang 
2021d4002b98SHong Zhang   /* allocate the matrix space */
2022d4002b98SHong Zhang   if (mallocmatspace) {
2023d4002b98SHong Zhang     ierr = PetscMalloc2(a->maxallocmat,&c->val,a->maxallocmat,&c->colidx);CHKERRQ(ierr);
2024d4002b98SHong Zhang     ierr = PetscLogObjectMemory((PetscObject)C,a->maxallocmat*(sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr);
2025d4002b98SHong Zhang 
2026d4002b98SHong Zhang     c->singlemalloc = PETSC_TRUE;
2027d4002b98SHong Zhang 
2028d4002b98SHong Zhang     if (m > 0) {
2029580bdb30SBarry Smith       ierr = PetscArraycpy(c->colidx,a->colidx,a->maxallocmat);CHKERRQ(ierr);
2030d4002b98SHong Zhang       if (cpvalues == MAT_COPY_VALUES) {
2031580bdb30SBarry Smith         ierr = PetscArraycpy(c->val,a->val,a->maxallocmat);CHKERRQ(ierr);
2032d4002b98SHong Zhang       } else {
2033580bdb30SBarry Smith         ierr = PetscArrayzero(c->val,a->maxallocmat);CHKERRQ(ierr);
2034d4002b98SHong Zhang       }
2035d4002b98SHong Zhang     }
2036d4002b98SHong Zhang   }
2037d4002b98SHong Zhang 
2038d4002b98SHong Zhang   c->ignorezeroentries = a->ignorezeroentries;
2039d4002b98SHong Zhang   c->roworiented       = a->roworiented;
2040d4002b98SHong Zhang   c->nonew             = a->nonew;
2041d4002b98SHong Zhang   if (a->diag) {
2042d4002b98SHong Zhang     ierr = PetscMalloc1(m,&c->diag);CHKERRQ(ierr);
2043d4002b98SHong Zhang     ierr = PetscLogObjectMemory((PetscObject)C,m*sizeof(PetscInt));CHKERRQ(ierr);
2044d4002b98SHong Zhang     for (i=0; i<m; i++) {
2045d4002b98SHong Zhang       c->diag[i] = a->diag[i];
2046d4002b98SHong Zhang     }
2047f4259b30SLisandro Dalcin   } else c->diag = NULL;
2048d4002b98SHong Zhang 
2049f4259b30SLisandro Dalcin   c->solve_work         = NULL;
2050f4259b30SLisandro Dalcin   c->saved_values       = NULL;
2051f4259b30SLisandro Dalcin   c->idiag              = NULL;
2052f4259b30SLisandro Dalcin   c->ssor_work          = NULL;
2053d4002b98SHong Zhang   c->keepnonzeropattern = a->keepnonzeropattern;
2054d4002b98SHong Zhang   c->free_val           = PETSC_TRUE;
2055d4002b98SHong Zhang   c->free_colidx        = PETSC_TRUE;
2056d4002b98SHong Zhang 
2057d4002b98SHong Zhang   c->maxallocmat  = a->maxallocmat;
2058d4002b98SHong Zhang   c->maxallocrow  = a->maxallocrow;
2059d4002b98SHong Zhang   c->rlenmax      = a->rlenmax;
2060d4002b98SHong Zhang   c->nz           = a->nz;
2061d4002b98SHong Zhang   C->preallocated = PETSC_TRUE;
2062d4002b98SHong Zhang 
2063d4002b98SHong Zhang   c->nonzerorowcnt = a->nonzerorowcnt;
2064d4002b98SHong Zhang   C->nonzerostate  = A->nonzerostate;
2065d4002b98SHong Zhang 
2066d4002b98SHong Zhang   ierr = PetscFunctionListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist);CHKERRQ(ierr);
2067d4002b98SHong Zhang   PetscFunctionReturn(0);
2068d4002b98SHong Zhang }
2069d4002b98SHong Zhang 
2070d4002b98SHong Zhang PetscErrorCode MatDuplicate_SeqSELL(Mat A,MatDuplicateOption cpvalues,Mat *B)
2071d4002b98SHong Zhang {
2072d4002b98SHong Zhang   PetscErrorCode ierr;
2073d4002b98SHong Zhang 
2074d4002b98SHong Zhang   PetscFunctionBegin;
2075d4002b98SHong Zhang   ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr);
2076d4002b98SHong Zhang   ierr = MatSetSizes(*B,A->rmap->n,A->cmap->n,A->rmap->n,A->cmap->n);CHKERRQ(ierr);
2077d4002b98SHong Zhang   if (!(A->rmap->n % A->rmap->bs) && !(A->cmap->n % A->cmap->bs)) {
2078d4002b98SHong Zhang     ierr = MatSetBlockSizesFromMats(*B,A,A);CHKERRQ(ierr);
2079d4002b98SHong Zhang   }
2080d4002b98SHong Zhang   ierr = MatSetType(*B,((PetscObject)A)->type_name);CHKERRQ(ierr);
2081d4002b98SHong Zhang   ierr = MatDuplicateNoCreate_SeqSELL(*B,A,cpvalues,PETSC_TRUE);CHKERRQ(ierr);
2082d4002b98SHong Zhang   PetscFunctionReturn(0);
2083d4002b98SHong Zhang }
2084d4002b98SHong Zhang 
2085d4002b98SHong Zhang /*@C
2086d4002b98SHong Zhang  MatCreateSeqSELL - Creates a sparse matrix in SELL format.
2087d4002b98SHong Zhang 
2088d083f849SBarry Smith  Collective
2089d4002b98SHong Zhang 
2090d4002b98SHong Zhang  Input Parameters:
2091d4002b98SHong Zhang +  comm - MPI communicator, set to PETSC_COMM_SELF
2092d4002b98SHong Zhang .  m - number of rows
2093d4002b98SHong Zhang .  n - number of columns
2094d4002b98SHong Zhang .  rlenmax - maximum number of nonzeros in a row
2095d4002b98SHong Zhang -  rlen - array containing the number of nonzeros in the various rows
2096d4002b98SHong Zhang  (possibly different for each row) or NULL
2097d4002b98SHong Zhang 
2098d4002b98SHong Zhang  Output Parameter:
2099d4002b98SHong Zhang .  A - the matrix
2100d4002b98SHong Zhang 
2101d4002b98SHong Zhang  It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(),
2102f6f02116SRichard Tran Mills  MatXXXXSetPreallocation() paradigm instead of this routine directly.
2103d4002b98SHong Zhang  [MatXXXXSetPreallocation() is, for example, MatSeqSELLSetPreallocation]
2104d4002b98SHong Zhang 
2105d4002b98SHong Zhang  Notes:
2106d4002b98SHong Zhang  If nnz is given then nz is ignored
2107d4002b98SHong Zhang 
2108d4002b98SHong Zhang  Specify the preallocated storage with either rlenmax or rlen (not both).
2109d4002b98SHong Zhang  Set rlenmax=PETSC_DEFAULT and rlen=NULL for PETSc to control dynamic memory
2110d4002b98SHong Zhang  allocation.  For large problems you MUST preallocate memory or you
2111d4002b98SHong Zhang  will get TERRIBLE performance, see the users' manual chapter on matrices.
2112d4002b98SHong Zhang 
2113d4002b98SHong Zhang  Level: intermediate
2114d4002b98SHong Zhang 
21153eec2f6aSPierre Jolivet  .seealso: MatCreate(), MatCreateSELL(), MatSetValues()
2116d4002b98SHong Zhang 
2117d4002b98SHong Zhang  @*/
2118d4002b98SHong Zhang PetscErrorCode MatCreateSeqSELL(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt maxallocrow,const PetscInt rlen[],Mat *A)
2119d4002b98SHong Zhang {
2120d4002b98SHong Zhang   PetscErrorCode ierr;
2121d4002b98SHong Zhang 
2122d4002b98SHong Zhang   PetscFunctionBegin;
2123d4002b98SHong Zhang   ierr = MatCreate(comm,A);CHKERRQ(ierr);
2124d4002b98SHong Zhang   ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr);
2125d4002b98SHong Zhang   ierr = MatSetType(*A,MATSEQSELL);CHKERRQ(ierr);
2126d4002b98SHong Zhang   ierr = MatSeqSELLSetPreallocation_SeqSELL(*A,maxallocrow,rlen);CHKERRQ(ierr);
2127d4002b98SHong Zhang   PetscFunctionReturn(0);
2128d4002b98SHong Zhang }
2129d4002b98SHong Zhang 
2130d4002b98SHong Zhang PetscErrorCode MatEqual_SeqSELL(Mat A,Mat B,PetscBool * flg)
2131d4002b98SHong Zhang {
2132d4002b98SHong Zhang   Mat_SeqSELL     *a=(Mat_SeqSELL*)A->data,*b=(Mat_SeqSELL*)B->data;
2133d4002b98SHong Zhang   PetscInt       totalslices=a->totalslices;
2134d4002b98SHong Zhang   PetscErrorCode ierr;
2135d4002b98SHong Zhang 
2136d4002b98SHong Zhang   PetscFunctionBegin;
2137d4002b98SHong Zhang   /* If the  matrix dimensions are not equal,or no of nonzeros */
2138d4002b98SHong Zhang   if ((A->rmap->n != B->rmap->n) || (A->cmap->n != B->cmap->n) ||(a->nz != b->nz) || (a->rlenmax != b->rlenmax)) {
2139d4002b98SHong Zhang     *flg = PETSC_FALSE;
2140d4002b98SHong Zhang     PetscFunctionReturn(0);
2141d4002b98SHong Zhang   }
2142d4002b98SHong Zhang   /* if the a->colidx are the same */
2143580bdb30SBarry Smith   ierr = PetscArraycmp(a->colidx,b->colidx,a->sliidx[totalslices],flg);CHKERRQ(ierr);
2144d4002b98SHong Zhang   if (!*flg) PetscFunctionReturn(0);
2145d4002b98SHong Zhang   /* if a->val are the same */
2146580bdb30SBarry Smith   ierr = PetscArraycmp(a->val,b->val,a->sliidx[totalslices],flg);CHKERRQ(ierr);
2147d4002b98SHong Zhang   PetscFunctionReturn(0);
2148d4002b98SHong Zhang }
2149d4002b98SHong Zhang 
2150d4002b98SHong Zhang PetscErrorCode MatSeqSELLInvalidateDiagonal(Mat A)
2151d4002b98SHong Zhang {
2152d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
2153d4002b98SHong Zhang 
2154d4002b98SHong Zhang   PetscFunctionBegin;
2155d4002b98SHong Zhang   a->idiagvalid  = PETSC_FALSE;
2156d4002b98SHong Zhang   PetscFunctionReturn(0);
2157d4002b98SHong Zhang }
2158d4002b98SHong Zhang 
2159d4002b98SHong Zhang PetscErrorCode MatConjugate_SeqSELL(Mat A)
2160d4002b98SHong Zhang {
2161d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX)
2162d4002b98SHong Zhang   Mat_SeqSELL *a=(Mat_SeqSELL*)A->data;
2163d4002b98SHong Zhang   PetscInt    i;
2164d4002b98SHong Zhang   PetscScalar *val = a->val;
2165d4002b98SHong Zhang 
2166d4002b98SHong Zhang   PetscFunctionBegin;
2167d4002b98SHong Zhang   for (i=0; i<a->sliidx[a->totalslices]; i++) {
2168d4002b98SHong Zhang     val[i] = PetscConj(val[i]);
2169d4002b98SHong Zhang   }
2170d4002b98SHong Zhang #else
2171d4002b98SHong Zhang   PetscFunctionBegin;
2172d4002b98SHong Zhang #endif
2173d4002b98SHong Zhang   PetscFunctionReturn(0);
2174d4002b98SHong Zhang }
2175