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