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