xref: /petsc/src/mat/impls/aij/seq/aijfact.c (revision f73b041577f311cd4e522d727240702cd4268ffe)
1b377110cSBarry Smith 
2c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/aij.h>
3c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/sbaij.h>
4c6db04a5SJed Brown #include <petscbt.h>
5c6db04a5SJed Brown #include <../src/mat/utils/freespace.h>
63b2fbd54SBarry Smith 
78fa24674SBarry Smith /*
88fa24674SBarry Smith       Computes an ordering to get most of the large numerical values in the lower triangular part of the matrix
9312e372aSBarry Smith 
10312e372aSBarry Smith       This code does not work and is not called anywhere. It would be registered with MatOrderingRegisterAll()
118fa24674SBarry Smith */
1219fd82e9SBarry Smith PetscErrorCode MatGetOrdering_Flow_SeqAIJ(Mat mat,MatOrderingType type,IS *irow,IS *icol)
133b2fbd54SBarry Smith {
148fa24674SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)mat->data;
158fa24674SBarry Smith   PetscErrorCode    ierr;
16889a8dedSBarry Smith   PetscInt          i,j,jj,k, kk,n = mat->rmap->n, current = 0, newcurrent = 0,*order;
178fa24674SBarry Smith   const PetscInt    *ai = a->i, *aj = a->j;
188fa24674SBarry Smith   const PetscScalar *aa = a->a;
19ace3abfcSBarry Smith   PetscBool         *done;
20889a8dedSBarry Smith   PetscReal         best,past = 0,future;
213a40ed3dSBarry Smith 
228fa24674SBarry Smith   PetscFunctionBegin;
238fa24674SBarry Smith   /* pick initial row */
248fa24674SBarry Smith   best = -1;
258fa24674SBarry Smith   for (i=0; i<n; i++) {
2675567043SBarry Smith     future = 0.0;
278fa24674SBarry Smith     for (j=ai[i]; j<ai[i+1]; j++) {
28db4deed7SKarl Rupp       if (aj[j] != i) future += PetscAbsScalar(aa[j]);
29db4deed7SKarl Rupp       else              past  = PetscAbsScalar(aa[j]);
308fa24674SBarry Smith     }
318fa24674SBarry Smith     if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */
328fa24674SBarry Smith     if (past/future > best) {
338fa24674SBarry Smith       best    = past/future;
348fa24674SBarry Smith       current = i;
358fa24674SBarry Smith     }
368fa24674SBarry Smith   }
378fa24674SBarry Smith 
38785e854fSJed Brown   ierr     = PetscMalloc1(n,&done);CHKERRQ(ierr);
39580bdb30SBarry Smith   ierr     = PetscArrayzero(done,n);CHKERRQ(ierr);
40785e854fSJed Brown   ierr     = PetscMalloc1(n,&order);CHKERRQ(ierr);
418fa24674SBarry Smith   order[0] = current;
428fa24674SBarry Smith   for (i=0; i<n-1; i++) {
438fa24674SBarry Smith     done[current] = PETSC_TRUE;
448fa24674SBarry Smith     best          = -1;
458fa24674SBarry Smith     /* loop over all neighbors of current pivot */
468fa24674SBarry Smith     for (j=ai[current]; j<ai[current+1]; j++) {
478fa24674SBarry Smith       jj = aj[j];
488fa24674SBarry Smith       if (done[jj]) continue;
498fa24674SBarry Smith       /* loop over columns of potential next row computing weights for below and above diagonal */
508fa24674SBarry Smith       past = future = 0.0;
518fa24674SBarry Smith       for (k=ai[jj]; k<ai[jj+1]; k++) {
528fa24674SBarry Smith         kk = aj[k];
538fa24674SBarry Smith         if (done[kk]) past += PetscAbsScalar(aa[k]);
548fa24674SBarry Smith         else if (kk != jj) future += PetscAbsScalar(aa[k]);
558fa24674SBarry Smith       }
568fa24674SBarry Smith       if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */
578fa24674SBarry Smith       if (past/future > best) {
588fa24674SBarry Smith         best       = past/future;
598fa24674SBarry Smith         newcurrent = jj;
608fa24674SBarry Smith       }
618fa24674SBarry Smith     }
628fa24674SBarry Smith     if (best == -1) { /* no neighbors to select from so select best of all that remain */
638fa24674SBarry Smith       best = -1;
648fa24674SBarry Smith       for (k=0; k<n; k++) {
658fa24674SBarry Smith         if (done[k]) continue;
6675567043SBarry Smith         future = 0.0;
6775567043SBarry Smith         past   = 0.0;
688fa24674SBarry Smith         for (j=ai[k]; j<ai[k+1]; j++) {
698fa24674SBarry Smith           kk = aj[j];
708fa24674SBarry Smith           if (done[kk])       past += PetscAbsScalar(aa[j]);
718fa24674SBarry Smith           else if (kk != k) future += PetscAbsScalar(aa[j]);
728fa24674SBarry Smith         }
738fa24674SBarry Smith         if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */
748fa24674SBarry Smith         if (past/future > best) {
758fa24674SBarry Smith           best       = past/future;
768fa24674SBarry Smith           newcurrent = k;
778fa24674SBarry Smith         }
788fa24674SBarry Smith       }
798fa24674SBarry Smith     }
80e32f2f54SBarry Smith     if (current == newcurrent) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"newcurrent cannot be current");
818fa24674SBarry Smith     current    = newcurrent;
828fa24674SBarry Smith     order[i+1] = current;
838fa24674SBarry Smith   }
8470b3c8c7SBarry Smith   ierr  = ISCreateGeneral(PETSC_COMM_SELF,n,order,PETSC_COPY_VALUES,irow);CHKERRQ(ierr);
858fa24674SBarry Smith   *icol = *irow;
868fa24674SBarry Smith   ierr  = PetscObjectReference((PetscObject)*irow);CHKERRQ(ierr);
878fa24674SBarry Smith   ierr  = PetscFree(done);CHKERRQ(ierr);
888fa24674SBarry Smith   ierr  = PetscFree(order);CHKERRQ(ierr);
89d64ed03dSBarry Smith   PetscFunctionReturn(0);
903b2fbd54SBarry Smith }
913b2fbd54SBarry Smith 
92cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatGetFactor_seqaij_petsc(Mat A,MatFactorType ftype,Mat *B)
93b24902e0SBarry Smith {
94d0f46423SBarry Smith   PetscInt       n = A->rmap->n;
95b24902e0SBarry Smith   PetscErrorCode ierr;
96b24902e0SBarry Smith 
97b24902e0SBarry Smith   PetscFunctionBegin;
98891bceaeSHong Zhang #if defined(PETSC_USE_COMPLEX)
99eb1ec7c1SStefano Zampini   if (A->hermitian && !A->symmetric && (ftype == MAT_FACTOR_CHOLESKY||ftype == MAT_FACTOR_ICC)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hermitian CHOLESKY or ICC Factor is not supported");
100891bceaeSHong Zhang #endif
101ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr);
102b24902e0SBarry Smith   ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr);
103599ef60dSHong Zhang   if (ftype == MAT_FACTOR_LU || ftype == MAT_FACTOR_ILU || ftype == MAT_FACTOR_ILUDT) {
104b24902e0SBarry Smith     ierr = MatSetType(*B,MATSEQAIJ);CHKERRQ(ierr);
1052205254eSKarl Rupp 
10635233d99SShri Abhyankar     (*B)->ops->ilufactorsymbolic = MatILUFactorSymbolic_SeqAIJ;
10735233d99SShri Abhyankar     (*B)->ops->lufactorsymbolic  = MatLUFactorSymbolic_SeqAIJ;
1082205254eSKarl Rupp 
10933d57670SJed Brown     ierr = MatSetBlockSizesFromMats(*B,A,A);CHKERRQ(ierr);
110b24902e0SBarry Smith   } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
111b24902e0SBarry Smith     ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr);
1120298fd71SBarry Smith     ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
1132205254eSKarl Rupp 
11435233d99SShri Abhyankar     (*B)->ops->iccfactorsymbolic      = MatICCFactorSymbolic_SeqAIJ;
11535233d99SShri Abhyankar     (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ;
116e32f2f54SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported");
117d5f3da31SBarry Smith   (*B)->factortype = ftype;
11800c67f3bSHong Zhang 
11900c67f3bSHong Zhang   ierr = PetscFree((*B)->solvertype);CHKERRQ(ierr);
12000c67f3bSHong Zhang   ierr = PetscStrallocpy(MATSOLVERPETSC,&(*B)->solvertype);CHKERRQ(ierr);
121*f73b0415SBarry Smith   (*B)->canuseordering = PETSC_TRUE;
122b24902e0SBarry Smith   PetscFunctionReturn(0);
123b24902e0SBarry Smith }
124b24902e0SBarry Smith 
125ad04f41aSHong Zhang PetscErrorCode MatLUFactorSymbolic_SeqAIJ_inplace(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
126289bc588SBarry Smith {
127416022c9SBarry Smith   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
128289bc588SBarry Smith   IS                 isicol;
1296849ba73SBarry Smith   PetscErrorCode     ierr;
1305d0c19d7SBarry Smith   const PetscInt     *r,*ic;
1315d0c19d7SBarry Smith   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j;
1321393bc97SHong Zhang   PetscInt           *bi,*bj,*ajtmp;
1331393bc97SHong Zhang   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
1349e046878SBarry Smith   PetscReal          f;
1354875ba38SHong Zhang   PetscInt           nlnk,*lnk,k,**bi_ptr;
1360298fd71SBarry Smith   PetscFreeSpaceList free_space=NULL,current_space=NULL;
1371393bc97SHong Zhang   PetscBT            lnkbt;
138ece542f9SHong Zhang   PetscBool          missing;
139289bc588SBarry Smith 
1403a40ed3dSBarry Smith   PetscFunctionBegin;
141e32f2f54SBarry Smith   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square");
142ece542f9SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
143ece542f9SHong Zhang   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
144ece542f9SHong Zhang 
1454c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
146ac121b3dSBarry Smith   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
147ac121b3dSBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
148289bc588SBarry Smith 
149289bc588SBarry Smith   /* get new row pointers */
150854ce69bSBarry Smith   ierr  = PetscMalloc1(n+1,&bi);CHKERRQ(ierr);
1511393bc97SHong Zhang   bi[0] = 0;
1521393bc97SHong Zhang 
1531393bc97SHong Zhang   /* bdiag is location of diagonal in factor */
154854ce69bSBarry Smith   ierr     = PetscMalloc1(n+1,&bdiag);CHKERRQ(ierr);
1551393bc97SHong Zhang   bdiag[0] = 0;
1561393bc97SHong Zhang 
1571393bc97SHong Zhang   /* linked list for storing column indices of the active row */
1581393bc97SHong Zhang   nlnk = n + 1;
1591393bc97SHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
1601393bc97SHong Zhang 
161dcca6d9dSJed Brown   ierr = PetscMalloc2(n+1,&bi_ptr,n+1,&im);CHKERRQ(ierr);
1621393bc97SHong Zhang 
1631393bc97SHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
164d3d32019SBarry Smith   f             = info->fill;
165aa91b3e7SRichard Tran Mills   if (n==1)   f = 1; /* prevent failure in corner case of 1x1 matrix with fill < 0.5 */
1662f18eb33SBarry Smith   ierr          = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space);CHKERRQ(ierr);
1671393bc97SHong Zhang   current_space = free_space;
168289bc588SBarry Smith 
169289bc588SBarry Smith   for (i=0; i<n; i++) {
1701393bc97SHong Zhang     /* copy previous fill into linked list */
171289bc588SBarry Smith     nzi = 0;
1721393bc97SHong Zhang     nnz = ai[r[i]+1] - ai[r[i]];
1731393bc97SHong Zhang     ajtmp = aj + ai[r[i]];
174afcc9904SHong Zhang     ierr  = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
1751393bc97SHong Zhang     nzi  += nlnk;
1761393bc97SHong Zhang 
1771393bc97SHong Zhang     /* add pivot rows into linked list */
1781393bc97SHong Zhang     row = lnk[n];
1791393bc97SHong Zhang     while (row < i) {
180d1170560SHong Zhang       nzbd  = bdiag[row] - bi[row] + 1;   /* num of entries in the row with column index <= row */
181d1170560SHong Zhang       ajtmp = bi_ptr[row] + nzbd;   /* points to the entry next to the diagonal */
182d1170560SHong Zhang       ierr  = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
1831393bc97SHong Zhang       nzi  += nlnk;
1841393bc97SHong Zhang       row   = lnk[row];
185289bc588SBarry Smith     }
1861393bc97SHong Zhang     bi[i+1] = bi[i] + nzi;
1871393bc97SHong Zhang     im[i]   = nzi;
1881393bc97SHong Zhang 
1891393bc97SHong Zhang     /* mark bdiag */
1901393bc97SHong Zhang     nzbd = 0;
1911393bc97SHong Zhang     nnz  = nzi;
1921393bc97SHong Zhang     k    = lnk[n];
1931393bc97SHong Zhang     while (nnz-- && k < i) {
1941393bc97SHong Zhang       nzbd++;
1951393bc97SHong Zhang       k = lnk[k];
1961393bc97SHong Zhang     }
1971393bc97SHong Zhang     bdiag[i] = bi[i] + nzbd;
1981393bc97SHong Zhang 
1991393bc97SHong Zhang     /* if free space is not available, make more free space */
2001393bc97SHong Zhang     if (current_space->local_remaining<nzi) {
201f91af8c7SBarry Smith       nnz  = PetscIntMultTruncate(n - i,nzi); /* estimated and max additional space needed */
202a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
2031393bc97SHong Zhang       reallocs++;
2041393bc97SHong Zhang     }
2051393bc97SHong Zhang 
2061393bc97SHong Zhang     /* copy data into free space, then initialize lnk */
2071393bc97SHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
2082205254eSKarl Rupp 
2091393bc97SHong Zhang     bi_ptr[i]                       = current_space->array;
2101393bc97SHong Zhang     current_space->array           += nzi;
2111393bc97SHong Zhang     current_space->local_used      += nzi;
2121393bc97SHong Zhang     current_space->local_remaining -= nzi;
213289bc588SBarry Smith   }
2146cf91177SBarry Smith #if defined(PETSC_USE_INFO)
215464e8d44SSatish Balay   if (ai[n] != 0) {
2161393bc97SHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
21757622a8eSBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)f,(double)af);CHKERRQ(ierr);
21857622a8eSBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
21957622a8eSBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g);\n",(double)af);CHKERRQ(ierr);
220ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
22105bf559cSSatish Balay   } else {
222ae15b995SBarry Smith     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
22305bf559cSSatish Balay   }
22463ba0a88SBarry Smith #endif
2252fb3ed76SBarry Smith 
226898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
227898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
2281987afe7SBarry Smith 
2291393bc97SHong Zhang   /* destroy list of free space and other temporary array(s) */
230854ce69bSBarry Smith   ierr = PetscMalloc1(bi[n]+1,&bj);CHKERRQ(ierr);
231a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr);
2321393bc97SHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
23335baf27eSSatish Balay   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
234289bc588SBarry Smith 
235289bc588SBarry Smith   /* put together the new matrix */
2360298fd71SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
2373bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)B,(PetscObject)isicol);CHKERRQ(ierr);
238719d5645SBarry Smith   b    = (Mat_SeqAIJ*)(B)->data;
2392205254eSKarl Rupp 
240e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
241e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
2427c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
2432205254eSKarl Rupp 
244854ce69bSBarry Smith   ierr    = PetscMalloc1(bi[n]+1,&b->a);CHKERRQ(ierr);
2451393bc97SHong Zhang   b->j    = bj;
2461393bc97SHong Zhang   b->i    = bi;
2471393bc97SHong Zhang   b->diag = bdiag;
248f4259b30SLisandro Dalcin   b->ilen = NULL;
249f4259b30SLisandro Dalcin   b->imax = NULL;
250416022c9SBarry Smith   b->row  = isrow;
251416022c9SBarry Smith   b->col  = iscol;
252c38d4ed2SBarry Smith   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
253c38d4ed2SBarry Smith   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
25482bf6240SBarry Smith   b->icol = isicol;
255854ce69bSBarry Smith   ierr    = PetscMalloc1(n+1,&b->solve_work);CHKERRQ(ierr);
2561393bc97SHong Zhang 
2571393bc97SHong Zhang   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
2583bb1ff40SBarry Smith   ierr     = PetscLogObjectMemory((PetscObject)B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
2591393bc97SHong Zhang   b->maxnz = b->nz = bi[n];
2607fd98bd6SLois Curfman McInnes 
261d5f3da31SBarry Smith   (B)->factortype            = MAT_FACTOR_LU;
262719d5645SBarry Smith   (B)->info.factor_mallocs   = reallocs;
263719d5645SBarry Smith   (B)->info.fill_ratio_given = f;
264703deb49SSatish Balay 
2659f7d0b68SHong Zhang   if (ai[n]) {
266719d5645SBarry Smith     (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
267eea2913aSSatish Balay   } else {
268719d5645SBarry Smith     (B)->info.fill_ratio_needed = 0.0;
269eea2913aSSatish Balay   }
270ad04f41aSHong Zhang   (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace;
271f2344125SBarry Smith   if (a->inode.size) {
272f2344125SBarry Smith     (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
273f2344125SBarry Smith   }
2743a40ed3dSBarry Smith   PetscFunctionReturn(0);
275289bc588SBarry Smith }
2761393bc97SHong Zhang 
27735233d99SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
278ce3d78c0SShri Abhyankar {
279ce3d78c0SShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
280ce3d78c0SShri Abhyankar   IS                 isicol;
281ce3d78c0SShri Abhyankar   PetscErrorCode     ierr;
2828758e1faSBarry Smith   const PetscInt     *r,*ic,*ai=a->i,*aj=a->j,*ajtmp;
2838758e1faSBarry Smith   PetscInt           i,n=A->rmap->n;
2848758e1faSBarry Smith   PetscInt           *bi,*bj;
285ce3d78c0SShri Abhyankar   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
286ce3d78c0SShri Abhyankar   PetscReal          f;
287ce3d78c0SShri Abhyankar   PetscInt           nlnk,*lnk,k,**bi_ptr;
2880298fd71SBarry Smith   PetscFreeSpaceList free_space=NULL,current_space=NULL;
289ce3d78c0SShri Abhyankar   PetscBT            lnkbt;
290ece542f9SHong Zhang   PetscBool          missing;
291ce3d78c0SShri Abhyankar 
292ce3d78c0SShri Abhyankar   PetscFunctionBegin;
293e32f2f54SBarry Smith   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square");
294ece542f9SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
295ece542f9SHong Zhang   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
296ece542f9SHong Zhang 
297ce3d78c0SShri Abhyankar   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
298ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
299ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
300ce3d78c0SShri Abhyankar 
3011bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
302854ce69bSBarry Smith   ierr  = PetscMalloc1(n+1,&bi);CHKERRQ(ierr);
303854ce69bSBarry Smith   ierr  = PetscMalloc1(n+1,&bdiag);CHKERRQ(ierr);
304a6df321fSShri Abhyankar   bi[0] = bdiag[0] = 0;
3059b48462bSShri Abhyankar 
3069b48462bSShri Abhyankar   /* linked list for storing column indices of the active row */
3079b48462bSShri Abhyankar   nlnk = n + 1;
3089b48462bSShri Abhyankar   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3099b48462bSShri Abhyankar 
310dcca6d9dSJed Brown   ierr = PetscMalloc2(n+1,&bi_ptr,n+1,&im);CHKERRQ(ierr);
3119b48462bSShri Abhyankar 
3129b48462bSShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
3139b48462bSShri Abhyankar   f             = info->fill;
314aa91b3e7SRichard Tran Mills   if (n==1)   f = 1; /* prevent failure in corner case of 1x1 matrix with fill < 0.5 */
3152f18eb33SBarry Smith   ierr          = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space);CHKERRQ(ierr);
3169b48462bSShri Abhyankar   current_space = free_space;
3179b48462bSShri Abhyankar 
3189b48462bSShri Abhyankar   for (i=0; i<n; i++) {
3199b48462bSShri Abhyankar     /* copy previous fill into linked list */
3209b48462bSShri Abhyankar     nzi = 0;
3219b48462bSShri Abhyankar     nnz = ai[r[i]+1] - ai[r[i]];
3229b48462bSShri Abhyankar     ajtmp = aj + ai[r[i]];
3239b48462bSShri Abhyankar     ierr  = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3249b48462bSShri Abhyankar     nzi  += nlnk;
3259b48462bSShri Abhyankar 
3269b48462bSShri Abhyankar     /* add pivot rows into linked list */
3279b48462bSShri Abhyankar     row = lnk[n];
3289b48462bSShri Abhyankar     while (row < i) {
3299b48462bSShri Abhyankar       nzbd  = bdiag[row] + 1; /* num of entries in the row with column index <= row */
3309b48462bSShri Abhyankar       ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
3319b48462bSShri Abhyankar       ierr  = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
3329b48462bSShri Abhyankar       nzi  += nlnk;
3339b48462bSShri Abhyankar       row   = lnk[row];
3349b48462bSShri Abhyankar     }
3359b48462bSShri Abhyankar     bi[i+1] = bi[i] + nzi;
3369b48462bSShri Abhyankar     im[i]   = nzi;
3379b48462bSShri Abhyankar 
3389b48462bSShri Abhyankar     /* mark bdiag */
3399b48462bSShri Abhyankar     nzbd = 0;
3409b48462bSShri Abhyankar     nnz  = nzi;
3419b48462bSShri Abhyankar     k    = lnk[n];
3429b48462bSShri Abhyankar     while (nnz-- && k < i) {
3439b48462bSShri Abhyankar       nzbd++;
3449b48462bSShri Abhyankar       k = lnk[k];
3459b48462bSShri Abhyankar     }
3469b48462bSShri Abhyankar     bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */
3479b48462bSShri Abhyankar 
3489b48462bSShri Abhyankar     /* if free space is not available, make more free space */
3499b48462bSShri Abhyankar     if (current_space->local_remaining<nzi) {
3502f18eb33SBarry Smith       /* estimated additional space needed */
3511df4278eSBarry Smith       nnz  = PetscIntMultTruncate(2,PetscIntMultTruncate(n-1,nzi));
3529b48462bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
3539b48462bSShri Abhyankar       reallocs++;
3549b48462bSShri Abhyankar     }
3559b48462bSShri Abhyankar 
3569b48462bSShri Abhyankar     /* copy data into free space, then initialize lnk */
3579b48462bSShri Abhyankar     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
3582205254eSKarl Rupp 
3599b48462bSShri Abhyankar     bi_ptr[i]                       = current_space->array;
3609b48462bSShri Abhyankar     current_space->array           += nzi;
3619b48462bSShri Abhyankar     current_space->local_used      += nzi;
3629b48462bSShri Abhyankar     current_space->local_remaining -= nzi;
3639b48462bSShri Abhyankar   }
3649b48462bSShri Abhyankar 
3659b48462bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3669b48462bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3679b48462bSShri Abhyankar 
3689263d837SHong Zhang   /*   copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */
369854ce69bSBarry Smith   ierr = PetscMalloc1(bi[n]+1,&bj);CHKERRQ(ierr);
3702ce24eb6SHong Zhang   ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
3719b48462bSShri Abhyankar   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
3729b48462bSShri Abhyankar   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
3739b48462bSShri Abhyankar 
3749b48462bSShri Abhyankar   /* put together the new matrix */
3750298fd71SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
3763bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)B,(PetscObject)isicol);CHKERRQ(ierr);
3779b48462bSShri Abhyankar   b    = (Mat_SeqAIJ*)(B)->data;
3782205254eSKarl Rupp 
3799b48462bSShri Abhyankar   b->free_a       = PETSC_TRUE;
3809b48462bSShri Abhyankar   b->free_ij      = PETSC_TRUE;
3819b48462bSShri Abhyankar   b->singlemalloc = PETSC_FALSE;
3822205254eSKarl Rupp 
383854ce69bSBarry Smith   ierr = PetscMalloc1(bdiag[0]+1,&b->a);CHKERRQ(ierr);
3842205254eSKarl Rupp 
3859b48462bSShri Abhyankar   b->j    = bj;
3869b48462bSShri Abhyankar   b->i    = bi;
3879b48462bSShri Abhyankar   b->diag = bdiag;
388f4259b30SLisandro Dalcin   b->ilen = NULL;
389f4259b30SLisandro Dalcin   b->imax = NULL;
3909b48462bSShri Abhyankar   b->row  = isrow;
3919b48462bSShri Abhyankar   b->col  = iscol;
3929b48462bSShri Abhyankar   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
3939b48462bSShri Abhyankar   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
3949b48462bSShri Abhyankar   b->icol = isicol;
395854ce69bSBarry Smith   ierr    = PetscMalloc1(n+1,&b->solve_work);CHKERRQ(ierr);
3969b48462bSShri Abhyankar 
3979b48462bSShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
3983bb1ff40SBarry Smith   ierr     = PetscLogObjectMemory((PetscObject)B,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
3999b48462bSShri Abhyankar   b->maxnz = b->nz = bdiag[0]+1;
4002205254eSKarl Rupp 
401d5f3da31SBarry Smith   B->factortype            = MAT_FACTOR_LU;
402f284f12aSHong Zhang   B->info.factor_mallocs   = reallocs;
403f284f12aSHong Zhang   B->info.fill_ratio_given = f;
4049b48462bSShri Abhyankar 
4059b48462bSShri Abhyankar   if (ai[n]) {
406f284f12aSHong Zhang     B->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
4079b48462bSShri Abhyankar   } else {
408f284f12aSHong Zhang     B->info.fill_ratio_needed = 0.0;
4099b48462bSShri Abhyankar   }
4109263d837SHong Zhang #if defined(PETSC_USE_INFO)
4119263d837SHong Zhang   if (ai[n] != 0) {
4129263d837SHong Zhang     PetscReal af = B->info.fill_ratio_needed;
41357622a8eSBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)f,(double)af);CHKERRQ(ierr);
41457622a8eSBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
41557622a8eSBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g);\n",(double)af);CHKERRQ(ierr);
4169263d837SHong Zhang     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
4179263d837SHong Zhang   } else {
4189263d837SHong Zhang     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
4199263d837SHong Zhang   }
4209263d837SHong Zhang #endif
42135233d99SShri Abhyankar   B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ;
422f2344125SBarry Smith   if (a->inode.size) {
423d3ac4fa3SBarry Smith     B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode;
424d3ac4fa3SBarry Smith   }
425a02bda8eSBarry Smith   ierr = MatSeqAIJCheckInode_FactorLU(B);CHKERRQ(ierr);
4269b48462bSShri Abhyankar   PetscFunctionReturn(0);
4279b48462bSShri Abhyankar }
4289b48462bSShri Abhyankar 
4295b5bc046SBarry Smith /*
4305b5bc046SBarry Smith     Trouble in factorization, should we dump the original matrix?
4315b5bc046SBarry Smith */
4325b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A)
4335b5bc046SBarry Smith {
4345b5bc046SBarry Smith   PetscErrorCode ierr;
435ace3abfcSBarry Smith   PetscBool      flg = PETSC_FALSE;
4365b5bc046SBarry Smith 
4375b5bc046SBarry Smith   PetscFunctionBegin;
438c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)A)->options,NULL,"-mat_factor_dump_on_error",&flg,NULL);CHKERRQ(ierr);
4395b5bc046SBarry Smith   if (flg) {
4405b5bc046SBarry Smith     PetscViewer viewer;
4415b5bc046SBarry Smith     char        filename[PETSC_MAX_PATH_LEN];
4425b5bc046SBarry Smith 
4435b5bc046SBarry Smith     ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr);
444ce94432eSBarry Smith     ierr = PetscViewerBinaryOpen(PetscObjectComm((PetscObject)A),filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr);
4455b5bc046SBarry Smith     ierr = MatView(A,viewer);CHKERRQ(ierr);
4466bf464f9SBarry Smith     ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr);
4475b5bc046SBarry Smith   }
4485b5bc046SBarry Smith   PetscFunctionReturn(0);
4495b5bc046SBarry Smith }
4505b5bc046SBarry Smith 
45135233d99SShri Abhyankar PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
452c9c72f8fSHong Zhang {
453c9c72f8fSHong Zhang   Mat             C     =B;
454c9c72f8fSHong Zhang   Mat_SeqAIJ      *a    =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data;
455c9c72f8fSHong Zhang   IS              isrow = b->row,isicol = b->icol;
456c9c72f8fSHong Zhang   PetscErrorCode  ierr;
457c9c72f8fSHong Zhang   const PetscInt  *r,*ic,*ics;
458bbd65245SShri Abhyankar   const PetscInt  n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bdiag=b->diag;
459bbd65245SShri Abhyankar   PetscInt        i,j,k,nz,nzL,row,*pj;
460bbd65245SShri Abhyankar   const PetscInt  *ajtmp,*bjtmp;
461bbd65245SShri Abhyankar   MatScalar       *rtmp,*pc,multiplier,*pv;
462bbd65245SShri Abhyankar   const MatScalar *aa=a->a,*v;
463ace3abfcSBarry Smith   PetscBool       row_identity,col_identity;
464d90ac83dSHong Zhang   FactorShiftCtx  sctx;
4654f81c4b7SBarry Smith   const PetscInt  *ddiag;
466d4ad06f7SHong Zhang   PetscReal       rs;
467d4ad06f7SHong Zhang   MatScalar       d;
468d4ad06f7SHong Zhang 
469c9c72f8fSHong Zhang   PetscFunctionBegin;
470d6e5152cSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
471d90ac83dSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
472d4ad06f7SHong Zhang 
473f4db908eSBarry Smith   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
474d4ad06f7SHong Zhang     ddiag          = a->diag;
475d4ad06f7SHong Zhang     sctx.shift_top = info->zeropivot;
476d4ad06f7SHong Zhang     for (i=0; i<n; i++) {
477d4ad06f7SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
478d4ad06f7SHong Zhang       d  = (aa)[ddiag[i]];
479d4ad06f7SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
480d4ad06f7SHong Zhang       v  = aa+ai[i];
481d4ad06f7SHong Zhang       nz = ai[i+1] - ai[i];
4822205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
483d4ad06f7SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
484d4ad06f7SHong Zhang     }
485d4ad06f7SHong Zhang     sctx.shift_top *= 1.1;
486d4ad06f7SHong Zhang     sctx.nshift_max = 5;
487d4ad06f7SHong Zhang     sctx.shift_lo   = 0.;
488d4ad06f7SHong Zhang     sctx.shift_hi   = 1.;
489d4ad06f7SHong Zhang   }
490d4ad06f7SHong Zhang 
491c9c72f8fSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
492c9c72f8fSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
493854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&rtmp);CHKERRQ(ierr);
494c9c72f8fSHong Zhang   ics  = ic;
495c9c72f8fSHong Zhang 
496d4ad06f7SHong Zhang   do {
49707b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
498c9c72f8fSHong Zhang     for (i=0; i<n; i++) {
499c9c72f8fSHong Zhang       /* zero rtmp */
500c9c72f8fSHong Zhang       /* L part */
501c9c72f8fSHong Zhang       nz    = bi[i+1] - bi[i];
502c9c72f8fSHong Zhang       bjtmp = bj + bi[i];
503c9c72f8fSHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
504c9c72f8fSHong Zhang 
505c9c72f8fSHong Zhang       /* U part */
50662fbd6afSShri Abhyankar       nz    = bdiag[i]-bdiag[i+1];
50762fbd6afSShri Abhyankar       bjtmp = bj + bdiag[i+1]+1;
508813a1f2bSShri Abhyankar       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
509813a1f2bSShri Abhyankar 
510813a1f2bSShri Abhyankar       /* load in initial (unfactored row) */
511813a1f2bSShri Abhyankar       nz    = ai[r[i]+1] - ai[r[i]];
512813a1f2bSShri Abhyankar       ajtmp = aj + ai[r[i]];
513813a1f2bSShri Abhyankar       v     = aa + ai[r[i]];
514813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
515813a1f2bSShri Abhyankar         rtmp[ics[ajtmp[j]]] = v[j];
516813a1f2bSShri Abhyankar       }
517813a1f2bSShri Abhyankar       /* ZeropivotApply() */
51898a93161SHong Zhang       rtmp[i] += sctx.shift_amount;  /* shift the diagonal of the matrix */
519813a1f2bSShri Abhyankar 
520813a1f2bSShri Abhyankar       /* elimination */
521813a1f2bSShri Abhyankar       bjtmp = bj + bi[i];
522813a1f2bSShri Abhyankar       row   = *bjtmp++;
523813a1f2bSShri Abhyankar       nzL   = bi[i+1] - bi[i];
524f268cba8SShri Abhyankar       for (k=0; k < nzL; k++) {
525813a1f2bSShri Abhyankar         pc = rtmp + row;
526813a1f2bSShri Abhyankar         if (*pc != 0.0) {
527813a1f2bSShri Abhyankar           pv         = b->a + bdiag[row];
528813a1f2bSShri Abhyankar           multiplier = *pc * (*pv);
529813a1f2bSShri Abhyankar           *pc        = multiplier;
5302205254eSKarl Rupp 
53162fbd6afSShri Abhyankar           pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */
53262fbd6afSShri Abhyankar           pv = b->a + bdiag[row+1]+1;
53362fbd6afSShri Abhyankar           nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
5342205254eSKarl Rupp 
535813a1f2bSShri Abhyankar           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
536ca0c957dSBarry Smith           ierr = PetscLogFlops(1+2.0*nz);CHKERRQ(ierr);
537813a1f2bSShri Abhyankar         }
538f268cba8SShri Abhyankar         row = *bjtmp++;
539813a1f2bSShri Abhyankar       }
540813a1f2bSShri Abhyankar 
541813a1f2bSShri Abhyankar       /* finished row so stick it into b->a */
542813a1f2bSShri Abhyankar       rs = 0.0;
543813a1f2bSShri Abhyankar       /* L part */
544813a1f2bSShri Abhyankar       pv = b->a + bi[i];
545813a1f2bSShri Abhyankar       pj = b->j + bi[i];
546813a1f2bSShri Abhyankar       nz = bi[i+1] - bi[i];
547813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
548813a1f2bSShri Abhyankar         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
549813a1f2bSShri Abhyankar       }
550813a1f2bSShri Abhyankar 
551813a1f2bSShri Abhyankar       /* U part */
55262fbd6afSShri Abhyankar       pv = b->a + bdiag[i+1]+1;
55362fbd6afSShri Abhyankar       pj = b->j + bdiag[i+1]+1;
55462fbd6afSShri Abhyankar       nz = bdiag[i] - bdiag[i+1]-1;
555813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
556813a1f2bSShri Abhyankar         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
557813a1f2bSShri Abhyankar       }
558813a1f2bSShri Abhyankar 
559813a1f2bSShri Abhyankar       sctx.rs = rs;
560813a1f2bSShri Abhyankar       sctx.pv = rtmp[i];
5614cccfbddSHong Zhang       ierr    = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr);
56207b50cabSHong Zhang       if (sctx.newshift) break; /* break for-loop */
56307b50cabSHong Zhang       rtmp[i] = sctx.pv; /* sctx.pv might be updated in the case of MAT_SHIFT_INBLOCKS */
564813a1f2bSShri Abhyankar 
565813a1f2bSShri Abhyankar       /* Mark diagonal and invert diagonal for simplier triangular solves */
566813a1f2bSShri Abhyankar       pv  = b->a + bdiag[i];
567813a1f2bSShri Abhyankar       *pv = 1.0/rtmp[i];
568813a1f2bSShri Abhyankar 
569813a1f2bSShri Abhyankar     } /* endof for (i=0; i<n; i++) { */
570813a1f2bSShri Abhyankar 
5718ff23777SHong Zhang     /* MatPivotRefine() */
57207b50cabSHong Zhang     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
573813a1f2bSShri Abhyankar       /*
574813a1f2bSShri Abhyankar        * if no shift in this attempt & shifting & started shifting & can refine,
575813a1f2bSShri Abhyankar        * then try lower shift
576813a1f2bSShri Abhyankar        */
577813a1f2bSShri Abhyankar       sctx.shift_hi       = sctx.shift_fraction;
578813a1f2bSShri Abhyankar       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
579813a1f2bSShri Abhyankar       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
58007b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
581813a1f2bSShri Abhyankar       sctx.nshift++;
582813a1f2bSShri Abhyankar     }
58307b50cabSHong Zhang   } while (sctx.newshift);
584813a1f2bSShri Abhyankar 
585813a1f2bSShri Abhyankar   ierr = PetscFree(rtmp);CHKERRQ(ierr);
586813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
587813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
588d3ac4fa3SBarry Smith 
589813a1f2bSShri Abhyankar   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
590813a1f2bSShri Abhyankar   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
591abb87a52SBarry Smith   if (b->inode.size) {
592abb87a52SBarry Smith     C->ops->solve = MatSolve_SeqAIJ_Inode;
593abb87a52SBarry Smith   } else if (row_identity && col_identity) {
59435233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
595813a1f2bSShri Abhyankar   } else {
59635233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ;
597813a1f2bSShri Abhyankar   }
59835233d99SShri Abhyankar   C->ops->solveadd          = MatSolveAdd_SeqAIJ;
59935233d99SShri Abhyankar   C->ops->solvetranspose    = MatSolveTranspose_SeqAIJ;
60035233d99SShri Abhyankar   C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ;
60135233d99SShri Abhyankar   C->ops->matsolve          = MatMatSolve_SeqAIJ;
602813a1f2bSShri Abhyankar   C->assembled              = PETSC_TRUE;
603813a1f2bSShri Abhyankar   C->preallocated           = PETSC_TRUE;
6042205254eSKarl Rupp 
605813a1f2bSShri Abhyankar   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
606813a1f2bSShri Abhyankar 
60714d2772eSHong Zhang   /* MatShiftView(A,info,&sctx) */
608813a1f2bSShri Abhyankar   if (sctx.nshift) {
609f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
61057622a8eSBarry Smith       ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %g, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,(double)sctx.shift_amount,(double)sctx.shift_fraction,(double)sctx.shift_top);CHKERRQ(ierr);
611f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
61257622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
613f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS) {
61457622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %g\n",sctx.nshift,(double)info->shiftamount);CHKERRQ(ierr);
615813a1f2bSShri Abhyankar     }
616813a1f2bSShri Abhyankar   }
617813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
618813a1f2bSShri Abhyankar }
619813a1f2bSShri Abhyankar 
620ad04f41aSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info)
621289bc588SBarry Smith {
622719d5645SBarry Smith   Mat             C     =B;
623416022c9SBarry Smith   Mat_SeqAIJ      *a    =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data;
62482bf6240SBarry Smith   IS              isrow = b->row,isicol = b->icol;
6256849ba73SBarry Smith   PetscErrorCode  ierr;
6265d0c19d7SBarry Smith   const PetscInt  *r,*ic,*ics;
627d278edc7SBarry Smith   PetscInt        nz,row,i,j,n=A->rmap->n,diag;
628d278edc7SBarry Smith   const PetscInt  *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
629d278edc7SBarry Smith   const PetscInt  *ajtmp,*bjtmp,*diag_offset = b->diag,*pj;
630d278edc7SBarry Smith   MatScalar       *pv,*rtmp,*pc,multiplier,d;
631d278edc7SBarry Smith   const MatScalar *v,*aa=a->a;
63233f9893dSHong Zhang   PetscReal       rs=0.0;
633d90ac83dSHong Zhang   FactorShiftCtx  sctx;
6344f81c4b7SBarry Smith   const PetscInt  *ddiag;
635ace3abfcSBarry Smith   PetscBool       row_identity, col_identity;
636289bc588SBarry Smith 
6373a40ed3dSBarry Smith   PetscFunctionBegin;
638504a3fadSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
639504a3fadSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
640ada7143aSSatish Balay 
641f4db908eSBarry Smith   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
64242898933SHong Zhang     ddiag          = a->diag;
6439f7d0b68SHong Zhang     sctx.shift_top = info->zeropivot;
6446cc28720Svictorle     for (i=0; i<n; i++) {
6459f95998fSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
6469f7d0b68SHong Zhang       d  = (aa)[ddiag[i]];
6471a890ab1SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
6489f7d0b68SHong Zhang       v  = aa+ai[i];
6499f7d0b68SHong Zhang       nz = ai[i+1] - ai[i];
6502205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
6511a890ab1SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
6526cc28720Svictorle     }
653118f5789SBarry Smith     sctx.shift_top *= 1.1;
654d2276718SHong Zhang     sctx.nshift_max = 5;
655d2276718SHong Zhang     sctx.shift_lo   = 0.;
656d2276718SHong Zhang     sctx.shift_hi   = 1.;
657a0a356efSHong Zhang   }
658a0a356efSHong Zhang 
659504a3fadSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
660504a3fadSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
661854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&rtmp);CHKERRQ(ierr);
662504a3fadSHong Zhang   ics  = ic;
663504a3fadSHong Zhang 
664085256b3SBarry Smith   do {
66507b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
666289bc588SBarry Smith     for (i=0; i<n; i++) {
667b3bf805bSHong Zhang       nz    = bi[i+1] - bi[i];
668b3bf805bSHong Zhang       bjtmp = bj + bi[i];
6699f7d0b68SHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
670289bc588SBarry Smith 
671289bc588SBarry Smith       /* load in initial (unfactored row) */
6729f7d0b68SHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
6739f7d0b68SHong Zhang       ajtmp = aj + ai[r[i]];
6749f7d0b68SHong Zhang       v     = aa + ai[r[i]];
675085256b3SBarry Smith       for (j=0; j<nz; j++) {
676b3bf805bSHong Zhang         rtmp[ics[ajtmp[j]]] = v[j];
677085256b3SBarry Smith       }
678d2276718SHong Zhang       rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */
679289bc588SBarry Smith 
680b3bf805bSHong Zhang       row = *bjtmp++;
681f2582111SSatish Balay       while  (row < i) {
6828c37ef55SBarry Smith         pc = rtmp + row;
6838c37ef55SBarry Smith         if (*pc != 0.0) {
684010a20caSHong Zhang           pv         = b->a + diag_offset[row];
685010a20caSHong Zhang           pj         = b->j + diag_offset[row] + 1;
68635aab85fSBarry Smith           multiplier = *pc / *pv++;
6878c37ef55SBarry Smith           *pc        = multiplier;
688b3bf805bSHong Zhang           nz         = bi[row+1] - diag_offset[row] - 1;
6899f7d0b68SHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
690ca0c957dSBarry Smith           ierr = PetscLogFlops(1+2.0*nz);CHKERRQ(ierr);
691289bc588SBarry Smith         }
692b3bf805bSHong Zhang         row = *bjtmp++;
693289bc588SBarry Smith       }
694416022c9SBarry Smith       /* finished row so stick it into b->a */
695b3bf805bSHong Zhang       pv   = b->a + bi[i];
696b3bf805bSHong Zhang       pj   = b->j + bi[i];
697b3bf805bSHong Zhang       nz   = bi[i+1] - bi[i];
698b3bf805bSHong Zhang       diag = diag_offset[i] - bi[i];
699e57ff13aSBarry Smith       rs   = 0.0;
700d3d32019SBarry Smith       for (j=0; j<nz; j++) {
7019f7d0b68SHong Zhang         pv[j] = rtmp[pj[j]];
7029f7d0b68SHong Zhang         rs   += PetscAbsScalar(pv[j]);
703d3d32019SBarry Smith       }
704e57ff13aSBarry Smith       rs -= PetscAbsScalar(pv[diag]);
705d2276718SHong Zhang 
706d2276718SHong Zhang       sctx.rs = rs;
707d2276718SHong Zhang       sctx.pv = pv[diag];
7084cccfbddSHong Zhang       ierr    = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr);
70907b50cabSHong Zhang       if (sctx.newshift) break;
710504a3fadSHong Zhang       pv[diag] = sctx.pv;
71135aab85fSBarry Smith     }
712d2276718SHong Zhang 
71307b50cabSHong Zhang     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
7146cc28720Svictorle       /*
7156c037d1bSvictorle        * if no shift in this attempt & shifting & started shifting & can refine,
7166cc28720Svictorle        * then try lower shift
7176cc28720Svictorle        */
7180481f469SBarry Smith       sctx.shift_hi       = sctx.shift_fraction;
7190481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
7200481f469SBarry Smith       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
72107b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
722d2276718SHong Zhang       sctx.nshift++;
7236cc28720Svictorle     }
72407b50cabSHong Zhang   } while (sctx.newshift);
7250f11f9aeSSatish Balay 
72635aab85fSBarry Smith   /* invert diagonal entries for simplier triangular solves */
72735aab85fSBarry Smith   for (i=0; i<n; i++) {
728010a20caSHong Zhang     b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]];
72935aab85fSBarry Smith   }
730606d414cSSatish Balay   ierr = PetscFree(rtmp);CHKERRQ(ierr);
73178b31e54SBarry Smith   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
73278b31e54SBarry Smith   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
733d3ac4fa3SBarry Smith 
7348d8c7c43SBarry Smith   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
7358d8c7c43SBarry Smith   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
7368d8c7c43SBarry Smith   if (row_identity && col_identity) {
737ad04f41aSHong Zhang     C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_inplace;
7388d8c7c43SBarry Smith   } else {
739ad04f41aSHong Zhang     C->ops->solve = MatSolve_SeqAIJ_inplace;
740db4efbfdSBarry Smith   }
741ad04f41aSHong Zhang   C->ops->solveadd          = MatSolveAdd_SeqAIJ_inplace;
742ad04f41aSHong Zhang   C->ops->solvetranspose    = MatSolveTranspose_SeqAIJ_inplace;
743ad04f41aSHong Zhang   C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace;
744ad04f41aSHong Zhang   C->ops->matsolve          = MatMatSolve_SeqAIJ_inplace;
7452205254eSKarl Rupp 
746c456f294SBarry Smith   C->assembled    = PETSC_TRUE;
7475c9eb25fSBarry Smith   C->preallocated = PETSC_TRUE;
7482205254eSKarl Rupp 
749d0f46423SBarry Smith   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
750d2276718SHong Zhang   if (sctx.nshift) {
751f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
75257622a8eSBarry Smith       ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %g, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,(double)sctx.shift_amount,(double)sctx.shift_fraction,(double)sctx.shift_top);CHKERRQ(ierr);
753f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
75457622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
7556cc28720Svictorle     }
7560697b6caSHong Zhang   }
757d3ac4fa3SBarry Smith   (C)->ops->solve          = MatSolve_SeqAIJ_inplace;
758d3ac4fa3SBarry Smith   (C)->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace;
7592205254eSKarl Rupp 
760a02bda8eSBarry Smith   ierr = MatSeqAIJCheckInode(C);CHKERRQ(ierr);
7613a40ed3dSBarry Smith   PetscFunctionReturn(0);
762289bc588SBarry Smith }
7630889b5dcSvictorle 
764137fb511SHong Zhang /*
765137fb511SHong Zhang    This routine implements inplace ILU(0) with row or/and column permutations.
766137fb511SHong Zhang    Input:
767137fb511SHong Zhang      A - original matrix
768137fb511SHong Zhang    Output;
769137fb511SHong Zhang      A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i]
770137fb511SHong Zhang          a->j (col index) is permuted by the inverse of colperm, then sorted
771137fb511SHong Zhang          a->a reordered accordingly with a->j
772137fb511SHong Zhang          a->diag (ptr to diagonal elements) is updated.
773137fb511SHong Zhang */
7740481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info)
775137fb511SHong Zhang {
776b051339dSHong Zhang   Mat_SeqAIJ      *a    =(Mat_SeqAIJ*)A->data;
777b051339dSHong Zhang   IS              isrow = a->row,isicol = a->icol;
778137fb511SHong Zhang   PetscErrorCode  ierr;
7795d0c19d7SBarry Smith   const PetscInt  *r,*ic,*ics;
7805d0c19d7SBarry Smith   PetscInt        i,j,n=A->rmap->n,*ai=a->i,*aj=a->j;
7815d0c19d7SBarry Smith   PetscInt        *ajtmp,nz,row;
782b051339dSHong Zhang   PetscInt        *diag = a->diag,nbdiag,*pj;
783a77337e4SBarry Smith   PetscScalar     *rtmp,*pc,multiplier,d;
784504a3fadSHong Zhang   MatScalar       *pv,*v;
785137fb511SHong Zhang   PetscReal       rs;
786d90ac83dSHong Zhang   FactorShiftCtx  sctx;
787504a3fadSHong Zhang   const MatScalar *aa=a->a,*vtmp;
788137fb511SHong Zhang 
789137fb511SHong Zhang   PetscFunctionBegin;
790e32f2f54SBarry Smith   if (A != B) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address");
791504a3fadSHong Zhang 
792504a3fadSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
793504a3fadSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
794504a3fadSHong Zhang 
795504a3fadSHong Zhang   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
796504a3fadSHong Zhang     const PetscInt *ddiag = a->diag;
797504a3fadSHong Zhang     sctx.shift_top = info->zeropivot;
798504a3fadSHong Zhang     for (i=0; i<n; i++) {
799504a3fadSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
800504a3fadSHong Zhang       d    = (aa)[ddiag[i]];
801504a3fadSHong Zhang       rs   = -PetscAbsScalar(d) - PetscRealPart(d);
802504a3fadSHong Zhang       vtmp = aa+ai[i];
803504a3fadSHong Zhang       nz   = ai[i+1] - ai[i];
8042205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(vtmp[j]);
805504a3fadSHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
806504a3fadSHong Zhang     }
807504a3fadSHong Zhang     sctx.shift_top *= 1.1;
808504a3fadSHong Zhang     sctx.nshift_max = 5;
809504a3fadSHong Zhang     sctx.shift_lo   = 0.;
810504a3fadSHong Zhang     sctx.shift_hi   = 1.;
811504a3fadSHong Zhang   }
812504a3fadSHong Zhang 
813137fb511SHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
814137fb511SHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
815854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&rtmp);CHKERRQ(ierr);
816580bdb30SBarry Smith   ierr = PetscArrayzero(rtmp,n+1);CHKERRQ(ierr);
817b051339dSHong Zhang   ics  = ic;
818137fb511SHong Zhang 
819504a3fadSHong Zhang #if defined(MV)
82075567043SBarry Smith   sctx.shift_top      = 0.;
821e60cf9a0SBarry Smith   sctx.nshift_max     = 0;
82275567043SBarry Smith   sctx.shift_lo       = 0.;
82375567043SBarry Smith   sctx.shift_hi       = 0.;
82475567043SBarry Smith   sctx.shift_fraction = 0.;
825137fb511SHong Zhang 
826f4db908eSBarry Smith   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
82775567043SBarry Smith     sctx.shift_top = 0.;
828137fb511SHong Zhang     for (i=0; i<n; i++) {
829137fb511SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
830b051339dSHong Zhang       d  = (a->a)[diag[i]];
831137fb511SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
832b051339dSHong Zhang       v  = a->a+ai[i];
833b051339dSHong Zhang       nz = ai[i+1] - ai[i];
8342205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
835137fb511SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
836137fb511SHong Zhang     }
837137fb511SHong Zhang     if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot;
838137fb511SHong Zhang     sctx.shift_top *= 1.1;
839137fb511SHong Zhang     sctx.nshift_max = 5;
840137fb511SHong Zhang     sctx.shift_lo   = 0.;
841137fb511SHong Zhang     sctx.shift_hi   = 1.;
842137fb511SHong Zhang   }
843137fb511SHong Zhang 
84475567043SBarry Smith   sctx.shift_amount = 0.;
845137fb511SHong Zhang   sctx.nshift       = 0;
846504a3fadSHong Zhang #endif
847504a3fadSHong Zhang 
848137fb511SHong Zhang   do {
84907b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
850137fb511SHong Zhang     for (i=0; i<n; i++) {
851b051339dSHong Zhang       /* load in initial unfactored row */
852b051339dSHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
853b051339dSHong Zhang       ajtmp = aj + ai[r[i]];
854b051339dSHong Zhang       v     = a->a + ai[r[i]];
855137fb511SHong Zhang       /* sort permuted ajtmp and values v accordingly */
856b051339dSHong Zhang       for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]];
857137fb511SHong Zhang       ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr);
858137fb511SHong Zhang 
859b051339dSHong Zhang       diag[r[i]] = ai[r[i]];
860137fb511SHong Zhang       for (j=0; j<nz; j++) {
861137fb511SHong Zhang         rtmp[ajtmp[j]] = v[j];
862b051339dSHong Zhang         if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */
863137fb511SHong Zhang       }
864137fb511SHong Zhang       rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */
865137fb511SHong Zhang 
866b051339dSHong Zhang       row = *ajtmp++;
867137fb511SHong Zhang       while  (row < i) {
868137fb511SHong Zhang         pc = rtmp + row;
869137fb511SHong Zhang         if (*pc != 0.0) {
870b051339dSHong Zhang           pv = a->a + diag[r[row]];
871b051339dSHong Zhang           pj = aj + diag[r[row]] + 1;
872137fb511SHong Zhang 
873137fb511SHong Zhang           multiplier = *pc / *pv++;
874137fb511SHong Zhang           *pc        = multiplier;
875b051339dSHong Zhang           nz         = ai[r[row]+1] - diag[r[row]] - 1;
876b051339dSHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
877ca0c957dSBarry Smith           ierr = PetscLogFlops(1+2.0*nz);CHKERRQ(ierr);
878137fb511SHong Zhang         }
879b051339dSHong Zhang         row = *ajtmp++;
880137fb511SHong Zhang       }
881b051339dSHong Zhang       /* finished row so overwrite it onto a->a */
882b051339dSHong Zhang       pv     = a->a + ai[r[i]];
883b051339dSHong Zhang       pj     = aj + ai[r[i]];
884b051339dSHong Zhang       nz     = ai[r[i]+1] - ai[r[i]];
885b051339dSHong Zhang       nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */
886137fb511SHong Zhang 
887137fb511SHong Zhang       rs = 0.0;
888137fb511SHong Zhang       for (j=0; j<nz; j++) {
889b051339dSHong Zhang         pv[j] = rtmp[pj[j]];
890b051339dSHong Zhang         if (j != nbdiag) rs += PetscAbsScalar(pv[j]);
891137fb511SHong Zhang       }
892137fb511SHong Zhang 
893137fb511SHong Zhang       sctx.rs = rs;
894b051339dSHong Zhang       sctx.pv = pv[nbdiag];
8954cccfbddSHong Zhang       ierr    = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr);
89607b50cabSHong Zhang       if (sctx.newshift) break;
897504a3fadSHong Zhang       pv[nbdiag] = sctx.pv;
898137fb511SHong Zhang     }
899137fb511SHong Zhang 
90007b50cabSHong Zhang     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
901137fb511SHong Zhang       /*
902137fb511SHong Zhang        * if no shift in this attempt & shifting & started shifting & can refine,
903137fb511SHong Zhang        * then try lower shift
904137fb511SHong Zhang        */
9050481f469SBarry Smith       sctx.shift_hi       = sctx.shift_fraction;
9060481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
9070481f469SBarry Smith       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
90807b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
909137fb511SHong Zhang       sctx.nshift++;
910137fb511SHong Zhang     }
91107b50cabSHong Zhang   } while (sctx.newshift);
912137fb511SHong Zhang 
913137fb511SHong Zhang   /* invert diagonal entries for simplier triangular solves */
914137fb511SHong Zhang   for (i=0; i<n; i++) {
915b051339dSHong Zhang     a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]];
916137fb511SHong Zhang   }
917137fb511SHong Zhang 
918137fb511SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
919137fb511SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
920137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
9212205254eSKarl Rupp 
922b051339dSHong Zhang   A->ops->solve             = MatSolve_SeqAIJ_InplaceWithPerm;
923ad04f41aSHong Zhang   A->ops->solveadd          = MatSolveAdd_SeqAIJ_inplace;
924ad04f41aSHong Zhang   A->ops->solvetranspose    = MatSolveTranspose_SeqAIJ_inplace;
925ad04f41aSHong Zhang   A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace;
9262205254eSKarl Rupp 
927b051339dSHong Zhang   A->assembled    = PETSC_TRUE;
9285c9eb25fSBarry Smith   A->preallocated = PETSC_TRUE;
9292205254eSKarl Rupp 
930d0f46423SBarry Smith   ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr);
931137fb511SHong Zhang   if (sctx.nshift) {
932f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
93357622a8eSBarry Smith       ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %g, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,(double)sctx.shift_amount,(double)sctx.shift_fraction,(double)sctx.shift_top);CHKERRQ(ierr);
934f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
93557622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
936137fb511SHong Zhang     }
937137fb511SHong Zhang   }
938137fb511SHong Zhang   PetscFunctionReturn(0);
939137fb511SHong Zhang }
940137fb511SHong Zhang 
9410a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
9420481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info)
943da3a660dSBarry Smith {
944dfbe8321SBarry Smith   PetscErrorCode ierr;
945416022c9SBarry Smith   Mat            C;
946416022c9SBarry Smith 
9473a40ed3dSBarry Smith   PetscFunctionBegin;
9482692d6eeSBarry Smith   ierr = MatGetFactor(A,MATSOLVERPETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr);
949719d5645SBarry Smith   ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr);
950719d5645SBarry Smith   ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr);
9512205254eSKarl Rupp 
952db4efbfdSBarry Smith   A->ops->solve          = C->ops->solve;
953db4efbfdSBarry Smith   A->ops->solvetranspose = C->ops->solvetranspose;
9542205254eSKarl Rupp 
95528be2f97SBarry Smith   ierr = MatHeaderMerge(A,&C);CHKERRQ(ierr);
9563bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)A,(PetscObject)((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr);
9573a40ed3dSBarry Smith   PetscFunctionReturn(0);
958da3a660dSBarry Smith }
9590a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
9601d098868SHong Zhang 
961889a8dedSBarry Smith 
962ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_inplace(Mat A,Vec bb,Vec xx)
9638c37ef55SBarry Smith {
964416022c9SBarry Smith   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
965416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
9666849ba73SBarry Smith   PetscErrorCode    ierr;
9675d0c19d7SBarry Smith   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
9685d0c19d7SBarry Smith   PetscInt          nz;
9695d0c19d7SBarry Smith   const PetscInt    *rout,*cout,*r,*c;
970dd6ea824SBarry Smith   PetscScalar       *x,*tmp,*tmps,sum;
971d9fead3dSBarry Smith   const PetscScalar *b;
972dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
9738c37ef55SBarry Smith 
9743a40ed3dSBarry Smith   PetscFunctionBegin;
9753a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
97691bf9adeSBarry Smith 
9773649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
978fdc842d1SBarry Smith   ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr);
979416022c9SBarry Smith   tmp  = a->solve_work;
9808c37ef55SBarry Smith 
9813b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
9823b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
9838c37ef55SBarry Smith 
9848c37ef55SBarry Smith   /* forward solve the lower triangular */
9858c37ef55SBarry Smith   tmp[0] = b[*r++];
986010a20caSHong Zhang   tmps   = tmp;
9878c37ef55SBarry Smith   for (i=1; i<n; i++) {
988010a20caSHong Zhang     v   = aa + ai[i];
989010a20caSHong Zhang     vi  = aj + ai[i];
99053e7425aSBarry Smith     nz  = a->diag[i] - ai[i];
99153e7425aSBarry Smith     sum = b[*r++];
992003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
9938c37ef55SBarry Smith     tmp[i] = sum;
9948c37ef55SBarry Smith   }
9958c37ef55SBarry Smith 
9968c37ef55SBarry Smith   /* backward solve the upper triangular */
9978c37ef55SBarry Smith   for (i=n-1; i>=0; i--) {
998010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
999010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
1000416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
10018c37ef55SBarry Smith     sum = tmp[i];
1002003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1003010a20caSHong Zhang     x[*c--] = tmp[i] = sum*aa[a->diag[i]];
10048c37ef55SBarry Smith   }
10058c37ef55SBarry Smith 
10063b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
10073b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
10083649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1009fdc842d1SBarry Smith   ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr);
1010dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
10113a40ed3dSBarry Smith   PetscFunctionReturn(0);
10128c37ef55SBarry Smith }
1013026e39d0SSatish Balay 
1014ad04f41aSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ_inplace(Mat A,Mat B,Mat X)
10152bebee5dSHong Zhang {
10162bebee5dSHong Zhang   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
10172bebee5dSHong Zhang   IS                iscol = a->col,isrow = a->row;
10182bebee5dSHong Zhang   PetscErrorCode    ierr;
10195d0c19d7SBarry Smith   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
1020910cf402Sprj-   PetscInt          nz,neq,ldb,ldx;
10215d0c19d7SBarry Smith   const PetscInt    *rout,*cout,*r,*c;
1022910cf402Sprj-   PetscScalar       *x,*tmp = a->solve_work,*tmps,sum;
1023910cf402Sprj-   const PetscScalar *b,*aa = a->a,*v;
1024910cf402Sprj-   PetscBool         isdense;
10252bebee5dSHong Zhang 
10262bebee5dSHong Zhang   PetscFunctionBegin;
10272bebee5dSHong Zhang   if (!n) PetscFunctionReturn(0);
1028910cf402Sprj-   ierr = PetscObjectTypeCompare((PetscObject)B,MATSEQDENSE,&isdense);CHKERRQ(ierr);
1029910cf402Sprj-   if (!isdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix");
1030f9fb9879SHong Zhang   if (X != B) {
1031910cf402Sprj-     ierr = PetscObjectTypeCompare((PetscObject)X,MATSEQDENSE,&isdense);CHKERRQ(ierr);
1032910cf402Sprj-     if (!isdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix");
1033f9fb9879SHong Zhang   }
10341683a169SBarry Smith   ierr = MatDenseGetArrayRead(B,&b);CHKERRQ(ierr);
1035910cf402Sprj-   ierr = MatDenseGetLDA(B,&ldb);CHKERRQ(ierr);
10368c778c55SBarry Smith   ierr = MatDenseGetArray(X,&x);CHKERRQ(ierr);
1037910cf402Sprj-   ierr = MatDenseGetLDA(X,&ldx);CHKERRQ(ierr);
10382bebee5dSHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
10392bebee5dSHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1040a36b22ccSSatish Balay   for (neq=0; neq<B->cmap->n; neq++) {
10412bebee5dSHong Zhang     /* forward solve the lower triangular */
10422bebee5dSHong Zhang     tmp[0] = b[r[0]];
10432bebee5dSHong Zhang     tmps   = tmp;
10442bebee5dSHong Zhang     for (i=1; i<n; i++) {
10452bebee5dSHong Zhang       v   = aa + ai[i];
10462bebee5dSHong Zhang       vi  = aj + ai[i];
10472bebee5dSHong Zhang       nz  = a->diag[i] - ai[i];
10482bebee5dSHong Zhang       sum = b[r[i]];
1049003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10502bebee5dSHong Zhang       tmp[i] = sum;
10512bebee5dSHong Zhang     }
10522bebee5dSHong Zhang     /* backward solve the upper triangular */
10532bebee5dSHong Zhang     for (i=n-1; i>=0; i--) {
10542bebee5dSHong Zhang       v   = aa + a->diag[i] + 1;
10552bebee5dSHong Zhang       vi  = aj + a->diag[i] + 1;
10562bebee5dSHong Zhang       nz  = ai[i+1] - a->diag[i] - 1;
10572bebee5dSHong Zhang       sum = tmp[i];
1058003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10592bebee5dSHong Zhang       x[c[i]] = tmp[i] = sum*aa[a->diag[i]];
10602bebee5dSHong Zhang     }
1061910cf402Sprj-     b += ldb;
1062910cf402Sprj-     x += ldx;
10632bebee5dSHong Zhang   }
10642bebee5dSHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
10652bebee5dSHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
10661683a169SBarry Smith   ierr = MatDenseRestoreArrayRead(B,&b);CHKERRQ(ierr);
10678c778c55SBarry Smith   ierr = MatDenseRestoreArray(X,&x);CHKERRQ(ierr);
1068dc0b31edSSatish Balay   ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr);
10692bebee5dSHong Zhang   PetscFunctionReturn(0);
10702bebee5dSHong Zhang }
10712bebee5dSHong Zhang 
107235233d99SShri Abhyankar PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X)
10739bd0847aSShri Abhyankar {
10749bd0847aSShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
10759bd0847aSShri Abhyankar   IS                iscol = a->col,isrow = a->row;
10769bd0847aSShri Abhyankar   PetscErrorCode    ierr;
10779bd0847aSShri Abhyankar   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag;
1078910cf402Sprj-   PetscInt          nz,neq,ldb,ldx;
10799bd0847aSShri Abhyankar   const PetscInt    *rout,*cout,*r,*c;
1080910cf402Sprj-   PetscScalar       *x,*tmp = a->solve_work,sum;
1081910cf402Sprj-   const PetscScalar *b,*aa = a->a,*v;
1082910cf402Sprj-   PetscBool         isdense;
10839bd0847aSShri Abhyankar 
10849bd0847aSShri Abhyankar   PetscFunctionBegin;
10859bd0847aSShri Abhyankar   if (!n) PetscFunctionReturn(0);
1086910cf402Sprj-   ierr = PetscObjectTypeCompare((PetscObject)B,MATSEQDENSE,&isdense);CHKERRQ(ierr);
1087910cf402Sprj-   if (!isdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix");
1088f9fb9879SHong Zhang   if (X != B) {
1089910cf402Sprj-     ierr = PetscObjectTypeCompare((PetscObject)X,MATSEQDENSE,&isdense);CHKERRQ(ierr);
1090910cf402Sprj-     if (!isdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix");
1091f9fb9879SHong Zhang   }
10921683a169SBarry Smith   ierr = MatDenseGetArrayRead(B,&b);CHKERRQ(ierr);
1093910cf402Sprj-   ierr = MatDenseGetLDA(B,&ldb);CHKERRQ(ierr);
10948c778c55SBarry Smith   ierr = MatDenseGetArray(X,&x);CHKERRQ(ierr);
1095910cf402Sprj-   ierr = MatDenseGetLDA(X,&ldx);CHKERRQ(ierr);
10969bd0847aSShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
10979bd0847aSShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
10989bd0847aSShri Abhyankar   for (neq=0; neq<B->cmap->n; neq++) {
10999bd0847aSShri Abhyankar     /* forward solve the lower triangular */
11009bd0847aSShri Abhyankar     tmp[0] = b[r[0]];
11019bd0847aSShri Abhyankar     v      = aa;
11029bd0847aSShri Abhyankar     vi     = aj;
11039bd0847aSShri Abhyankar     for (i=1; i<n; i++) {
11049bd0847aSShri Abhyankar       nz  = ai[i+1] - ai[i];
11059bd0847aSShri Abhyankar       sum = b[r[i]];
11069bd0847aSShri Abhyankar       PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
11079bd0847aSShri Abhyankar       tmp[i] = sum;
11089bd0847aSShri Abhyankar       v     += nz; vi += nz;
11099bd0847aSShri Abhyankar     }
11109bd0847aSShri Abhyankar     /* backward solve the upper triangular */
11119bd0847aSShri Abhyankar     for (i=n-1; i>=0; i--) {
11129bd0847aSShri Abhyankar       v   = aa + adiag[i+1]+1;
11139bd0847aSShri Abhyankar       vi  = aj + adiag[i+1]+1;
11149bd0847aSShri Abhyankar       nz  = adiag[i]-adiag[i+1]-1;
11159bd0847aSShri Abhyankar       sum = tmp[i];
11169bd0847aSShri Abhyankar       PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
11179bd0847aSShri Abhyankar       x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */
11189bd0847aSShri Abhyankar     }
1119910cf402Sprj-     b += ldb;
1120910cf402Sprj-     x += ldx;
11219bd0847aSShri Abhyankar   }
11229bd0847aSShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
11239bd0847aSShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
11241683a169SBarry Smith   ierr = MatDenseRestoreArrayRead(B,&b);CHKERRQ(ierr);
11258c778c55SBarry Smith   ierr = MatDenseRestoreArray(X,&x);CHKERRQ(ierr);
11269bd0847aSShri Abhyankar   ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr);
11279bd0847aSShri Abhyankar   PetscFunctionReturn(0);
11289bd0847aSShri Abhyankar }
11299bd0847aSShri Abhyankar 
1130137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx)
1131137fb511SHong Zhang {
1132137fb511SHong Zhang   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
1133137fb511SHong Zhang   IS                iscol = a->col,isrow = a->row;
1134137fb511SHong Zhang   PetscErrorCode    ierr;
11355d0c19d7SBarry Smith   const PetscInt    *r,*c,*rout,*cout;
11365d0c19d7SBarry Smith   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
11375d0c19d7SBarry Smith   PetscInt          nz,row;
1138fdc842d1SBarry Smith   PetscScalar       *x,*tmp,*tmps,sum;
1139fdc842d1SBarry Smith   const PetscScalar *b;
1140dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
1141137fb511SHong Zhang 
1142137fb511SHong Zhang   PetscFunctionBegin;
1143137fb511SHong Zhang   if (!n) PetscFunctionReturn(0);
1144137fb511SHong Zhang 
1145fdc842d1SBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
1146fdc842d1SBarry Smith   ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr);
1147137fb511SHong Zhang   tmp  = a->solve_work;
1148137fb511SHong Zhang 
1149137fb511SHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1150137fb511SHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1151137fb511SHong Zhang 
1152137fb511SHong Zhang   /* forward solve the lower triangular */
1153137fb511SHong Zhang   tmp[0] = b[*r++];
1154137fb511SHong Zhang   tmps   = tmp;
1155137fb511SHong Zhang   for (row=1; row<n; row++) {
1156137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1157137fb511SHong Zhang     v   = aa + ai[i];
1158137fb511SHong Zhang     vi  = aj + ai[i];
1159137fb511SHong Zhang     nz  = a->diag[i] - ai[i];
1160137fb511SHong Zhang     sum = b[*r++];
1161003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1162137fb511SHong Zhang     tmp[row] = sum;
1163137fb511SHong Zhang   }
1164137fb511SHong Zhang 
1165137fb511SHong Zhang   /* backward solve the upper triangular */
1166137fb511SHong Zhang   for (row=n-1; row>=0; row--) {
1167137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1168137fb511SHong Zhang     v   = aa + a->diag[i] + 1;
1169137fb511SHong Zhang     vi  = aj + a->diag[i] + 1;
1170137fb511SHong Zhang     nz  = ai[i+1] - a->diag[i] - 1;
1171137fb511SHong Zhang     sum = tmp[row];
1172003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1173137fb511SHong Zhang     x[*c--] = tmp[row] = sum*aa[a->diag[i]];
1174137fb511SHong Zhang   }
1175137fb511SHong Zhang 
1176137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1177137fb511SHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1178fdc842d1SBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1179fdc842d1SBarry Smith   ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr);
1180dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
1181137fb511SHong Zhang   PetscFunctionReturn(0);
1182137fb511SHong Zhang }
1183137fb511SHong Zhang 
1184930ae53cSBarry Smith /* ----------------------------------------------------------- */
1185c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/ftn-kernels/fsolve.h>
1186ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_inplace(Mat A,Vec bb,Vec xx)
1187930ae53cSBarry Smith {
1188930ae53cSBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
11896849ba73SBarry Smith   PetscErrorCode    ierr;
1190d0f46423SBarry Smith   PetscInt          n   = A->rmap->n;
1191b377110cSBarry Smith   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag;
1192356650c2SBarry Smith   PetscScalar       *x;
1193356650c2SBarry Smith   const PetscScalar *b;
1194dd6ea824SBarry Smith   const MatScalar   *aa = a->a;
1195aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
1196356650c2SBarry Smith   PetscInt        adiag_i,i,nz,ai_i;
1197b377110cSBarry Smith   const PetscInt  *vi;
1198dd6ea824SBarry Smith   const MatScalar *v;
1199dd6ea824SBarry Smith   PetscScalar     sum;
1200d85afc42SSatish Balay #endif
1201930ae53cSBarry Smith 
12023a40ed3dSBarry Smith   PetscFunctionBegin;
12033a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
1204930ae53cSBarry Smith 
12053649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
1206fdc842d1SBarry Smith   ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr);
1207930ae53cSBarry Smith 
1208aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
12091c4feecaSSatish Balay   fortransolveaij_(&n,x,ai,aj,adiag,aa,b);
12101c4feecaSSatish Balay #else
1211930ae53cSBarry Smith   /* forward solve the lower triangular */
1212930ae53cSBarry Smith   x[0] = b[0];
1213930ae53cSBarry Smith   for (i=1; i<n; i++) {
1214930ae53cSBarry Smith     ai_i = ai[i];
1215930ae53cSBarry Smith     v    = aa + ai_i;
1216930ae53cSBarry Smith     vi   = aj + ai_i;
1217930ae53cSBarry Smith     nz   = adiag[i] - ai_i;
1218930ae53cSBarry Smith     sum  = b[i];
1219003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1220930ae53cSBarry Smith     x[i] = sum;
1221930ae53cSBarry Smith   }
1222930ae53cSBarry Smith 
1223930ae53cSBarry Smith   /* backward solve the upper triangular */
1224930ae53cSBarry Smith   for (i=n-1; i>=0; i--) {
1225930ae53cSBarry Smith     adiag_i = adiag[i];
1226d708749eSBarry Smith     v       = aa + adiag_i + 1;
1227d708749eSBarry Smith     vi      = aj + adiag_i + 1;
1228930ae53cSBarry Smith     nz      = ai[i+1] - adiag_i - 1;
1229930ae53cSBarry Smith     sum     = x[i];
1230003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1231930ae53cSBarry Smith     x[i] = sum*aa[adiag_i];
1232930ae53cSBarry Smith   }
12331c4feecaSSatish Balay #endif
1234dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
12353649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1236fdc842d1SBarry Smith   ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr);
12373a40ed3dSBarry Smith   PetscFunctionReturn(0);
1238930ae53cSBarry Smith }
1239930ae53cSBarry Smith 
1240ad04f41aSHong Zhang PetscErrorCode MatSolveAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec yy,Vec xx)
1241da3a660dSBarry Smith {
1242416022c9SBarry Smith   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
1243416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
12446849ba73SBarry Smith   PetscErrorCode    ierr;
124504fbf559SBarry Smith   PetscInt          i, n = A->rmap->n,j;
12465d0c19d7SBarry Smith   PetscInt          nz;
124704fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j;
124804fbf559SBarry Smith   PetscScalar       *x,*tmp,sum;
124904fbf559SBarry Smith   const PetscScalar *b;
1250dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
1251da3a660dSBarry Smith 
12523a40ed3dSBarry Smith   PetscFunctionBegin;
125378b31e54SBarry Smith   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
1254da3a660dSBarry Smith 
12553649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
12561ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1257416022c9SBarry Smith   tmp  = a->solve_work;
1258da3a660dSBarry Smith 
12593b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
12603b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1261da3a660dSBarry Smith 
1262da3a660dSBarry Smith   /* forward solve the lower triangular */
1263da3a660dSBarry Smith   tmp[0] = b[*r++];
1264da3a660dSBarry Smith   for (i=1; i<n; i++) {
1265010a20caSHong Zhang     v   = aa + ai[i];
1266010a20caSHong Zhang     vi  = aj + ai[i];
1267416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
1268da3a660dSBarry Smith     sum = b[*r++];
126904fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1270da3a660dSBarry Smith     tmp[i] = sum;
1271da3a660dSBarry Smith   }
1272da3a660dSBarry Smith 
1273da3a660dSBarry Smith   /* backward solve the upper triangular */
1274da3a660dSBarry Smith   for (i=n-1; i>=0; i--) {
1275010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
1276010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
1277416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
1278da3a660dSBarry Smith     sum = tmp[i];
127904fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1280010a20caSHong Zhang     tmp[i]   = sum*aa[a->diag[i]];
1281da3a660dSBarry Smith     x[*c--] += tmp[i];
1282da3a660dSBarry Smith   }
1283da3a660dSBarry Smith 
12843b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
12853b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
12863649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
12871ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1288dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
12893a40ed3dSBarry Smith   PetscFunctionReturn(0);
1290da3a660dSBarry Smith }
129104fbf559SBarry Smith 
129235233d99SShri Abhyankar PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx)
12933c0119dfSShri Abhyankar {
12943c0119dfSShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
12953c0119dfSShri Abhyankar   IS                iscol = a->col,isrow = a->row;
12963c0119dfSShri Abhyankar   PetscErrorCode    ierr;
12973c0119dfSShri Abhyankar   PetscInt          i, n = A->rmap->n,j;
12983c0119dfSShri Abhyankar   PetscInt          nz;
12993c0119dfSShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag;
13003c0119dfSShri Abhyankar   PetscScalar       *x,*tmp,sum;
13013c0119dfSShri Abhyankar   const PetscScalar *b;
13023c0119dfSShri Abhyankar   const MatScalar   *aa = a->a,*v;
13033c0119dfSShri Abhyankar 
13043c0119dfSShri Abhyankar   PetscFunctionBegin;
13053c0119dfSShri Abhyankar   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
13063c0119dfSShri Abhyankar 
13073649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
13083c0119dfSShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
13093c0119dfSShri Abhyankar   tmp  = a->solve_work;
13103c0119dfSShri Abhyankar 
13113c0119dfSShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
13123c0119dfSShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
13133c0119dfSShri Abhyankar 
13143c0119dfSShri Abhyankar   /* forward solve the lower triangular */
13153c0119dfSShri Abhyankar   tmp[0] = b[r[0]];
13163c0119dfSShri Abhyankar   v      = aa;
13173c0119dfSShri Abhyankar   vi     = aj;
13183c0119dfSShri Abhyankar   for (i=1; i<n; i++) {
13193c0119dfSShri Abhyankar     nz  = ai[i+1] - ai[i];
13203c0119dfSShri Abhyankar     sum = b[r[i]];
13213c0119dfSShri Abhyankar     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
13223c0119dfSShri Abhyankar     tmp[i] = sum;
13232205254eSKarl Rupp     v     += nz;
13242205254eSKarl Rupp     vi    += nz;
13253c0119dfSShri Abhyankar   }
13263c0119dfSShri Abhyankar 
13273c0119dfSShri Abhyankar   /* backward solve the upper triangular */
13283c0119dfSShri Abhyankar   v  = aa + adiag[n-1];
13293c0119dfSShri Abhyankar   vi = aj + adiag[n-1];
13303c0119dfSShri Abhyankar   for (i=n-1; i>=0; i--) {
13313c0119dfSShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
13323c0119dfSShri Abhyankar     sum = tmp[i];
13333c0119dfSShri Abhyankar     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
13343c0119dfSShri Abhyankar     tmp[i]   = sum*v[nz];
13353c0119dfSShri Abhyankar     x[c[i]] += tmp[i];
13363c0119dfSShri Abhyankar     v       += nz+1; vi += nz+1;
13373c0119dfSShri Abhyankar   }
13383c0119dfSShri Abhyankar 
13393c0119dfSShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
13403c0119dfSShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
13413649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
13423c0119dfSShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
13433c0119dfSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
13443c0119dfSShri Abhyankar   PetscFunctionReturn(0);
13453c0119dfSShri Abhyankar }
13463c0119dfSShri Abhyankar 
1347ad04f41aSHong Zhang PetscErrorCode MatSolveTranspose_SeqAIJ_inplace(Mat A,Vec bb,Vec xx)
1348da3a660dSBarry Smith {
1349416022c9SBarry Smith   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
13502235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
13516849ba73SBarry Smith   PetscErrorCode    ierr;
135204fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
135304fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
135404fbf559SBarry Smith   PetscInt          nz;
135504fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1356dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
135704fbf559SBarry Smith   const PetscScalar *b;
1358da3a660dSBarry Smith 
13593a40ed3dSBarry Smith   PetscFunctionBegin;
13603649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
1361fdc842d1SBarry Smith   ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr);
1362416022c9SBarry Smith   tmp  = a->solve_work;
1363da3a660dSBarry Smith 
13642235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
13652235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1366da3a660dSBarry Smith 
1367da3a660dSBarry Smith   /* copy the b into temp work space according to permutation */
13682235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1369da3a660dSBarry Smith 
1370da3a660dSBarry Smith   /* forward solve the U^T */
1371da3a660dSBarry Smith   for (i=0; i<n; i++) {
1372010a20caSHong Zhang     v   = aa + diag[i];
1373010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1374f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
1375c38d4ed2SBarry Smith     s1  = tmp[i];
1376ef66eb69SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
137704fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1378c38d4ed2SBarry Smith     tmp[i] = s1;
1379da3a660dSBarry Smith   }
1380da3a660dSBarry Smith 
1381da3a660dSBarry Smith   /* backward solve the L^T */
1382da3a660dSBarry Smith   for (i=n-1; i>=0; i--) {
1383010a20caSHong Zhang     v  = aa + diag[i] - 1;
1384010a20caSHong Zhang     vi = aj + diag[i] - 1;
1385f1af5d2fSBarry Smith     nz = diag[i] - ai[i];
1386f1af5d2fSBarry Smith     s1 = tmp[i];
138704fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
1388da3a660dSBarry Smith   }
1389da3a660dSBarry Smith 
1390da3a660dSBarry Smith   /* copy tmp into x according to permutation */
1391591294e4SBarry Smith   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1392da3a660dSBarry Smith 
13932235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
13942235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
13953649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1396fdc842d1SBarry Smith   ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr);
1397da3a660dSBarry Smith 
1398dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
13993a40ed3dSBarry Smith   PetscFunctionReturn(0);
1400da3a660dSBarry Smith }
1401da3a660dSBarry Smith 
140235233d99SShri Abhyankar PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx)
1403d1fa9404SShri Abhyankar {
1404d1fa9404SShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
1405d1fa9404SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
1406d1fa9404SShri Abhyankar   PetscErrorCode    ierr;
1407d1fa9404SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi;
1408d1fa9404SShri Abhyankar   PetscInt          i,n = A->rmap->n,j;
1409d1fa9404SShri Abhyankar   PetscInt          nz;
1410d1fa9404SShri Abhyankar   PetscScalar       *x,*tmp,s1;
1411d1fa9404SShri Abhyankar   const MatScalar   *aa = a->a,*v;
1412d1fa9404SShri Abhyankar   const PetscScalar *b;
1413d1fa9404SShri Abhyankar 
1414d1fa9404SShri Abhyankar   PetscFunctionBegin;
14153649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
1416fdc842d1SBarry Smith   ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr);
1417d1fa9404SShri Abhyankar   tmp  = a->solve_work;
1418d1fa9404SShri Abhyankar 
1419d1fa9404SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1420d1fa9404SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1421d1fa9404SShri Abhyankar 
1422d1fa9404SShri Abhyankar   /* copy the b into temp work space according to permutation */
1423d1fa9404SShri Abhyankar   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1424d1fa9404SShri Abhyankar 
1425d1fa9404SShri Abhyankar   /* forward solve the U^T */
1426d1fa9404SShri Abhyankar   for (i=0; i<n; i++) {
1427d1fa9404SShri Abhyankar     v   = aa + adiag[i+1] + 1;
1428d1fa9404SShri Abhyankar     vi  = aj + adiag[i+1] + 1;
1429d1fa9404SShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
1430d1fa9404SShri Abhyankar     s1  = tmp[i];
1431d1fa9404SShri Abhyankar     s1 *= v[nz];  /* multiply by inverse of diagonal entry */
1432d1fa9404SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1433d1fa9404SShri Abhyankar     tmp[i] = s1;
1434d1fa9404SShri Abhyankar   }
1435d1fa9404SShri Abhyankar 
1436d1fa9404SShri Abhyankar   /* backward solve the L^T */
1437d1fa9404SShri Abhyankar   for (i=n-1; i>=0; i--) {
1438d1fa9404SShri Abhyankar     v  = aa + ai[i];
1439d1fa9404SShri Abhyankar     vi = aj + ai[i];
1440d1fa9404SShri Abhyankar     nz = ai[i+1] - ai[i];
1441d1fa9404SShri Abhyankar     s1 = tmp[i];
1442d1fa9404SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1443d1fa9404SShri Abhyankar   }
1444d1fa9404SShri Abhyankar 
1445d1fa9404SShri Abhyankar   /* copy tmp into x according to permutation */
1446d1fa9404SShri Abhyankar   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1447d1fa9404SShri Abhyankar 
1448d1fa9404SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1449d1fa9404SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
14503649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1451fdc842d1SBarry Smith   ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr);
1452d1fa9404SShri Abhyankar 
1453d1fa9404SShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
1454d1fa9404SShri Abhyankar   PetscFunctionReturn(0);
1455d1fa9404SShri Abhyankar }
1456d1fa9404SShri Abhyankar 
1457ad04f41aSHong Zhang PetscErrorCode MatSolveTransposeAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec zz,Vec xx)
1458da3a660dSBarry Smith {
1459416022c9SBarry Smith   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
14602235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
14616849ba73SBarry Smith   PetscErrorCode    ierr;
146204fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
146304fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
146404fbf559SBarry Smith   PetscInt          nz;
146504fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1466dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
146704fbf559SBarry Smith   const PetscScalar *b;
14686abc6512SBarry Smith 
14693a40ed3dSBarry Smith   PetscFunctionBegin;
147023bc6035SMatthew Knepley   if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);}
14713649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
14721ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1473416022c9SBarry Smith   tmp  = a->solve_work;
14746abc6512SBarry Smith 
14752235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
14762235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
14776abc6512SBarry Smith 
14786abc6512SBarry Smith   /* copy the b into temp work space according to permutation */
14792235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
14806abc6512SBarry Smith 
14816abc6512SBarry Smith   /* forward solve the U^T */
14826abc6512SBarry Smith   for (i=0; i<n; i++) {
1483010a20caSHong Zhang     v   = aa + diag[i];
1484010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1485f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
148604fbf559SBarry Smith     s1  = tmp[i];
148704fbf559SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
148804fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
148904fbf559SBarry Smith     tmp[i] = s1;
14906abc6512SBarry Smith   }
14916abc6512SBarry Smith 
14926abc6512SBarry Smith   /* backward solve the L^T */
14936abc6512SBarry Smith   for (i=n-1; i>=0; i--) {
1494010a20caSHong Zhang     v  = aa + diag[i] - 1;
1495010a20caSHong Zhang     vi = aj + diag[i] - 1;
1496f1af5d2fSBarry Smith     nz = diag[i] - ai[i];
149704fbf559SBarry Smith     s1 = tmp[i];
149804fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
14996abc6512SBarry Smith   }
15006abc6512SBarry Smith 
15016abc6512SBarry Smith   /* copy tmp into x according to permutation */
15026abc6512SBarry Smith   for (i=0; i<n; i++) x[r[i]] += tmp[i];
15036abc6512SBarry Smith 
15042235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
15052235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
15063649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
15071ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
15086abc6512SBarry Smith 
150904fbf559SBarry Smith   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
15103a40ed3dSBarry Smith   PetscFunctionReturn(0);
1511da3a660dSBarry Smith }
151204fbf559SBarry Smith 
151335233d99SShri Abhyankar PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx)
1514533e6011SShri Abhyankar {
1515533e6011SShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
1516533e6011SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
1517533e6011SShri Abhyankar   PetscErrorCode    ierr;
1518533e6011SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi;
1519533e6011SShri Abhyankar   PetscInt          i,n = A->rmap->n,j;
1520533e6011SShri Abhyankar   PetscInt          nz;
1521533e6011SShri Abhyankar   PetscScalar       *x,*tmp,s1;
1522533e6011SShri Abhyankar   const MatScalar   *aa = a->a,*v;
1523533e6011SShri Abhyankar   const PetscScalar *b;
1524533e6011SShri Abhyankar 
1525533e6011SShri Abhyankar   PetscFunctionBegin;
1526533e6011SShri Abhyankar   if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);}
15273649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
1528533e6011SShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1529533e6011SShri Abhyankar   tmp  = a->solve_work;
1530533e6011SShri Abhyankar 
1531533e6011SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1532533e6011SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1533533e6011SShri Abhyankar 
1534533e6011SShri Abhyankar   /* copy the b into temp work space according to permutation */
1535533e6011SShri Abhyankar   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1536533e6011SShri Abhyankar 
1537533e6011SShri Abhyankar   /* forward solve the U^T */
1538533e6011SShri Abhyankar   for (i=0; i<n; i++) {
1539533e6011SShri Abhyankar     v   = aa + adiag[i+1] + 1;
1540533e6011SShri Abhyankar     vi  = aj + adiag[i+1] + 1;
1541533e6011SShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
1542533e6011SShri Abhyankar     s1  = tmp[i];
1543533e6011SShri Abhyankar     s1 *= v[nz];  /* multiply by inverse of diagonal entry */
1544533e6011SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1545533e6011SShri Abhyankar     tmp[i] = s1;
1546533e6011SShri Abhyankar   }
1547533e6011SShri Abhyankar 
1548533e6011SShri Abhyankar 
1549533e6011SShri Abhyankar   /* backward solve the L^T */
1550533e6011SShri Abhyankar   for (i=n-1; i>=0; i--) {
1551533e6011SShri Abhyankar     v  = aa + ai[i];
1552533e6011SShri Abhyankar     vi = aj + ai[i];
1553533e6011SShri Abhyankar     nz = ai[i+1] - ai[i];
1554533e6011SShri Abhyankar     s1 = tmp[i];
1555c5e3b2a3SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1556533e6011SShri Abhyankar   }
1557533e6011SShri Abhyankar 
1558533e6011SShri Abhyankar   /* copy tmp into x according to permutation */
1559533e6011SShri Abhyankar   for (i=0; i<n; i++) x[r[i]] += tmp[i];
1560533e6011SShri Abhyankar 
1561533e6011SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1562533e6011SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
15633649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1564533e6011SShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1565533e6011SShri Abhyankar 
1566533e6011SShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
1567533e6011SShri Abhyankar   PetscFunctionReturn(0);
1568533e6011SShri Abhyankar }
1569533e6011SShri Abhyankar 
15709e25ed09SBarry Smith /* ----------------------------------------------------------------*/
1571b47f5a65SHong Zhang 
15728fc3a347SHong Zhang /*
15738a76ed4fSHong Zhang    ilu() under revised new data structure.
1574813a1f2bSShri Abhyankar    Factored arrays bj and ba are stored as
1575813a1f2bSShri Abhyankar      L(0,:), L(1,:), ...,L(n-1,:),  U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:)
1576813a1f2bSShri Abhyankar 
1577813a1f2bSShri Abhyankar    bi=fact->i is an array of size n+1, in which
1578813a1f2bSShri Abhyankar    bi+
1579baabb860SHong Zhang      bi[i]:  points to 1st entry of L(i,:),i=0,...,n-1
1580baabb860SHong Zhang      bi[n]:  points to L(n-1,n-1)+1
1581baabb860SHong Zhang 
1582813a1f2bSShri Abhyankar   bdiag=fact->diag is an array of size n+1,in which
1583baabb860SHong Zhang      bdiag[i]: points to diagonal of U(i,:), i=0,...,n-1
1584a24f213cSHong Zhang      bdiag[n]: points to entry of U(n-1,0)-1
1585813a1f2bSShri Abhyankar 
1586813a1f2bSShri Abhyankar    U(i,:) contains bdiag[i] as its last entry, i.e.,
1587813a1f2bSShri Abhyankar     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
1588813a1f2bSShri Abhyankar */
158935233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
1590813a1f2bSShri Abhyankar {
1591813a1f2bSShri Abhyankar   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data,*b;
1592813a1f2bSShri Abhyankar   PetscErrorCode ierr;
1593bbd65245SShri Abhyankar   const PetscInt n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag;
1594bbd65245SShri Abhyankar   PetscInt       i,j,k=0,nz,*bi,*bj,*bdiag;
15951df811f5SHong Zhang   IS             isicol;
1596813a1f2bSShri Abhyankar 
1597813a1f2bSShri Abhyankar   PetscFunctionBegin;
15981df811f5SHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
1599813a1f2bSShri Abhyankar   ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr);
1600813a1f2bSShri Abhyankar   b    = (Mat_SeqAIJ*)(fact)->data;
1601813a1f2bSShri Abhyankar 
1602813a1f2bSShri Abhyankar   /* allocate matrix arrays for new data structure */
1603dcca6d9dSJed Brown   ierr = PetscMalloc3(ai[n]+1,&b->a,ai[n]+1,&b->j,n+1,&b->i);CHKERRQ(ierr);
16043bb1ff40SBarry Smith   ierr = PetscLogObjectMemory((PetscObject)fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(n+1)*sizeof(PetscInt));CHKERRQ(ierr);
16052205254eSKarl Rupp 
1606813a1f2bSShri Abhyankar   b->singlemalloc = PETSC_TRUE;
1607813a1f2bSShri Abhyankar   if (!b->diag) {
1608854ce69bSBarry Smith     ierr = PetscMalloc1(n+1,&b->diag);CHKERRQ(ierr);
16093bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)fact,(n+1)*sizeof(PetscInt));CHKERRQ(ierr);
1610813a1f2bSShri Abhyankar   }
1611813a1f2bSShri Abhyankar   bdiag = b->diag;
1612813a1f2bSShri Abhyankar 
1613813a1f2bSShri Abhyankar   if (n > 0) {
1614580bdb30SBarry Smith     ierr = PetscArrayzero(b->a,ai[n]);CHKERRQ(ierr);
1615813a1f2bSShri Abhyankar   }
1616813a1f2bSShri Abhyankar 
1617813a1f2bSShri Abhyankar   /* set bi and bj with new data structure */
1618813a1f2bSShri Abhyankar   bi = b->i;
1619813a1f2bSShri Abhyankar   bj = b->j;
1620813a1f2bSShri Abhyankar 
1621813a1f2bSShri Abhyankar   /* L part */
1622e60cf9a0SBarry Smith   bi[0] = 0;
1623813a1f2bSShri Abhyankar   for (i=0; i<n; i++) {
1624813a1f2bSShri Abhyankar     nz      = adiag[i] - ai[i];
1625813a1f2bSShri Abhyankar     bi[i+1] = bi[i] + nz;
1626813a1f2bSShri Abhyankar     aj      = a->j + ai[i];
1627813a1f2bSShri Abhyankar     for (j=0; j<nz; j++) {
1628bbd65245SShri Abhyankar       /*   *bj = aj[j]; bj++; */
1629bbd65245SShri Abhyankar       bj[k++] = aj[j];
1630813a1f2bSShri Abhyankar     }
1631813a1f2bSShri Abhyankar   }
1632813a1f2bSShri Abhyankar 
1633813a1f2bSShri Abhyankar   /* U part */
163462fbd6afSShri Abhyankar   bdiag[n] = bi[n]-1;
1635813a1f2bSShri Abhyankar   for (i=n-1; i>=0; i--) {
1636813a1f2bSShri Abhyankar     nz = ai[i+1] - adiag[i] - 1;
1637813a1f2bSShri Abhyankar     aj = a->j + adiag[i] + 1;
1638813a1f2bSShri Abhyankar     for (j=0; j<nz; j++) {
1639bbd65245SShri Abhyankar       /*      *bj = aj[j]; bj++; */
1640bbd65245SShri Abhyankar       bj[k++] = aj[j];
1641813a1f2bSShri Abhyankar     }
1642813a1f2bSShri Abhyankar     /* diag[i] */
1643bbd65245SShri Abhyankar     /*    *bj = i; bj++; */
1644bbd65245SShri Abhyankar     bj[k++]  = i;
1645a24f213cSHong Zhang     bdiag[i] = bdiag[i+1] + nz + 1;
1646813a1f2bSShri Abhyankar   }
16471df811f5SHong Zhang 
1648d5f3da31SBarry Smith   fact->factortype             = MAT_FACTOR_ILU;
16491df811f5SHong Zhang   fact->info.factor_mallocs    = 0;
16501df811f5SHong Zhang   fact->info.fill_ratio_given  = info->fill;
16511df811f5SHong Zhang   fact->info.fill_ratio_needed = 1.0;
165235233d99SShri Abhyankar   fact->ops->lufactornumeric   = MatLUFactorNumeric_SeqAIJ;
1653a02bda8eSBarry Smith   ierr = MatSeqAIJCheckInode_FactorLU(fact);CHKERRQ(ierr);
16541df811f5SHong Zhang 
16551df811f5SHong Zhang   b       = (Mat_SeqAIJ*)(fact)->data;
16561df811f5SHong Zhang   b->row  = isrow;
16571df811f5SHong Zhang   b->col  = iscol;
16581df811f5SHong Zhang   b->icol = isicol;
1659854ce69bSBarry Smith   ierr    = PetscMalloc1(fact->rmap->n+1,&b->solve_work);CHKERRQ(ierr);
16601df811f5SHong Zhang   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
16611df811f5SHong Zhang   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
1662813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
1663813a1f2bSShri Abhyankar }
1664813a1f2bSShri Abhyankar 
166535233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
1666813a1f2bSShri Abhyankar {
1667813a1f2bSShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
1668813a1f2bSShri Abhyankar   IS                 isicol;
1669813a1f2bSShri Abhyankar   PetscErrorCode     ierr;
1670813a1f2bSShri Abhyankar   const PetscInt     *r,*ic;
16711df811f5SHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j;
1672813a1f2bSShri Abhyankar   PetscInt           *bi,*cols,nnz,*cols_lvl;
1673813a1f2bSShri Abhyankar   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
1674813a1f2bSShri Abhyankar   PetscInt           i,levels,diagonal_fill;
1675035f4963SHong Zhang   PetscBool          col_identity,row_identity,missing;
1676813a1f2bSShri Abhyankar   PetscReal          f;
16770298fd71SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=NULL;
1678813a1f2bSShri Abhyankar   PetscBT            lnkbt;
1679813a1f2bSShri Abhyankar   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
16800298fd71SBarry Smith   PetscFreeSpaceList free_space    =NULL,current_space=NULL;
16810298fd71SBarry Smith   PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL;
1682813a1f2bSShri Abhyankar 
1683813a1f2bSShri Abhyankar   PetscFunctionBegin;
1684035f4963SHong Zhang   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
1685035f4963SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
1686035f4963SHong Zhang   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
1687ad04f41aSHong Zhang 
1688813a1f2bSShri Abhyankar   levels = (PetscInt)info->levels;
1689813a1f2bSShri Abhyankar   ierr   = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
1690813a1f2bSShri Abhyankar   ierr   = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
1691813a1f2bSShri Abhyankar   if (!levels && row_identity && col_identity) {
1692813a1f2bSShri Abhyankar     /* special case: ilu(0) with natural ordering */
169335233d99SShri Abhyankar     ierr = MatILUFactorSymbolic_SeqAIJ_ilu0(fact,A,isrow,iscol,info);CHKERRQ(ierr);
1694124a3351SHong Zhang     if (a->inode.size) {
1695124a3351SHong Zhang       fact->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode;
1696124a3351SHong Zhang     }
1697813a1f2bSShri Abhyankar     PetscFunctionReturn(0);
1698813a1f2bSShri Abhyankar   }
1699813a1f2bSShri Abhyankar 
17001df811f5SHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
1701813a1f2bSShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1702813a1f2bSShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
1703813a1f2bSShri Abhyankar 
17041bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
1705854ce69bSBarry Smith   ierr  = PetscMalloc1(n+1,&bi);CHKERRQ(ierr);
1706854ce69bSBarry Smith   ierr  = PetscMalloc1(n+1,&bdiag);CHKERRQ(ierr);
1707e60cf9a0SBarry Smith   bi[0] = bdiag[0] = 0;
1708dcca6d9dSJed Brown   ierr  = PetscMalloc2(n,&bj_ptr,n,&bjlvl_ptr);CHKERRQ(ierr);
1709813a1f2bSShri Abhyankar 
1710813a1f2bSShri Abhyankar   /* create a linked list for storing column indices of the active row */
1711813a1f2bSShri Abhyankar   nlnk = n + 1;
1712813a1f2bSShri Abhyankar   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
1713813a1f2bSShri Abhyankar 
1714813a1f2bSShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
17151df811f5SHong Zhang   f                 = info->fill;
17161df811f5SHong Zhang   diagonal_fill     = (PetscInt)info->diagonal_fill;
17172f18eb33SBarry Smith   ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space);CHKERRQ(ierr);
1718813a1f2bSShri Abhyankar   current_space     = free_space;
17192f18eb33SBarry Smith   ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space_lvl);CHKERRQ(ierr);
1720813a1f2bSShri Abhyankar   current_space_lvl = free_space_lvl;
1721813a1f2bSShri Abhyankar   for (i=0; i<n; i++) {
1722813a1f2bSShri Abhyankar     nzi = 0;
1723813a1f2bSShri Abhyankar     /* copy current row into linked list */
1724813a1f2bSShri Abhyankar     nnz = ai[r[i]+1] - ai[r[i]];
1725e32f2f54SBarry Smith     if (!nnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i);
1726813a1f2bSShri Abhyankar     cols   = aj + ai[r[i]];
1727813a1f2bSShri Abhyankar     lnk[i] = -1; /* marker to indicate if diagonal exists */
1728813a1f2bSShri Abhyankar     ierr   = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
1729813a1f2bSShri Abhyankar     nzi   += nlnk;
1730813a1f2bSShri Abhyankar 
1731813a1f2bSShri Abhyankar     /* make sure diagonal entry is included */
1732813a1f2bSShri Abhyankar     if (diagonal_fill && lnk[i] == -1) {
1733813a1f2bSShri Abhyankar       fm = n;
1734813a1f2bSShri Abhyankar       while (lnk[fm] < i) fm = lnk[fm];
1735813a1f2bSShri Abhyankar       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
1736813a1f2bSShri Abhyankar       lnk[fm]    = i;
1737813a1f2bSShri Abhyankar       lnk_lvl[i] = 0;
1738813a1f2bSShri Abhyankar       nzi++; dcount++;
1739813a1f2bSShri Abhyankar     }
1740813a1f2bSShri Abhyankar 
1741813a1f2bSShri Abhyankar     /* add pivot rows into the active row */
1742813a1f2bSShri Abhyankar     nzbd = 0;
1743813a1f2bSShri Abhyankar     prow = lnk[n];
1744813a1f2bSShri Abhyankar     while (prow < i) {
1745813a1f2bSShri Abhyankar       nnz      = bdiag[prow];
1746813a1f2bSShri Abhyankar       cols     = bj_ptr[prow] + nnz + 1;
1747813a1f2bSShri Abhyankar       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
1748813a1f2bSShri Abhyankar       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
1749813a1f2bSShri Abhyankar       ierr     = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
1750813a1f2bSShri Abhyankar       nzi     += nlnk;
1751813a1f2bSShri Abhyankar       prow     = lnk[prow];
1752813a1f2bSShri Abhyankar       nzbd++;
1753813a1f2bSShri Abhyankar     }
1754813a1f2bSShri Abhyankar     bdiag[i] = nzbd;
1755813a1f2bSShri Abhyankar     bi[i+1]  = bi[i] + nzi;
1756813a1f2bSShri Abhyankar     /* if free space is not available, make more free space */
1757813a1f2bSShri Abhyankar     if (current_space->local_remaining<nzi) {
1758f91af8c7SBarry Smith       nnz  = PetscIntMultTruncate(2,PetscIntMultTruncate(nzi,n - i)); /* estimated and max additional space needed */
1759813a1f2bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
1760813a1f2bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
1761813a1f2bSShri Abhyankar       reallocs++;
1762813a1f2bSShri Abhyankar     }
1763813a1f2bSShri Abhyankar 
1764813a1f2bSShri Abhyankar     /* copy data into free_space and free_space_lvl, then initialize lnk */
1765813a1f2bSShri Abhyankar     ierr         = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
1766813a1f2bSShri Abhyankar     bj_ptr[i]    = current_space->array;
1767813a1f2bSShri Abhyankar     bjlvl_ptr[i] = current_space_lvl->array;
1768813a1f2bSShri Abhyankar 
1769813a1f2bSShri Abhyankar     /* make sure the active row i has diagonal entry */
177065e19b50SBarry Smith     if (*(bj_ptr[i]+bdiag[i]) != i) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\ntry running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i);
1771813a1f2bSShri Abhyankar 
1772813a1f2bSShri Abhyankar     current_space->array               += nzi;
1773813a1f2bSShri Abhyankar     current_space->local_used          += nzi;
1774813a1f2bSShri Abhyankar     current_space->local_remaining     -= nzi;
1775813a1f2bSShri Abhyankar     current_space_lvl->array           += nzi;
1776813a1f2bSShri Abhyankar     current_space_lvl->local_used      += nzi;
1777813a1f2bSShri Abhyankar     current_space_lvl->local_remaining -= nzi;
1778813a1f2bSShri Abhyankar   }
1779813a1f2bSShri Abhyankar 
1780813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
1781813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
1782813a1f2bSShri Abhyankar   /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */
1783854ce69bSBarry Smith   ierr = PetscMalloc1(bi[n]+1,&bj);CHKERRQ(ierr);
17842ce24eb6SHong Zhang   ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
1785813a1f2bSShri Abhyankar 
1786813a1f2bSShri Abhyankar   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
1787813a1f2bSShri Abhyankar   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
17880e83c824SBarry Smith   ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr);
1789813a1f2bSShri Abhyankar 
1790813a1f2bSShri Abhyankar #if defined(PETSC_USE_INFO)
1791813a1f2bSShri Abhyankar   {
1792aef85c9fSShri Abhyankar     PetscReal af = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
179357622a8eSBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)f,(double)af);CHKERRQ(ierr);
179457622a8eSBarry Smith     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
179557622a8eSBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%g);\n",(double)af);CHKERRQ(ierr);
1796813a1f2bSShri Abhyankar     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
1797813a1f2bSShri Abhyankar     if (diagonal_fill) {
1798955c1f14SBarry Smith       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals\n",dcount);CHKERRQ(ierr);
1799813a1f2bSShri Abhyankar     }
1800813a1f2bSShri Abhyankar   }
1801813a1f2bSShri Abhyankar #endif
1802813a1f2bSShri Abhyankar   /* put together the new matrix */
18030298fd71SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
18043bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)fact,(PetscObject)isicol);CHKERRQ(ierr);
1805813a1f2bSShri Abhyankar   b    = (Mat_SeqAIJ*)(fact)->data;
18062205254eSKarl Rupp 
1807813a1f2bSShri Abhyankar   b->free_a       = PETSC_TRUE;
1808813a1f2bSShri Abhyankar   b->free_ij      = PETSC_TRUE;
1809813a1f2bSShri Abhyankar   b->singlemalloc = PETSC_FALSE;
18102205254eSKarl Rupp 
1811854ce69bSBarry Smith   ierr = PetscMalloc1(bdiag[0]+1,&b->a);CHKERRQ(ierr);
18122205254eSKarl Rupp 
1813813a1f2bSShri Abhyankar   b->j    = bj;
1814813a1f2bSShri Abhyankar   b->i    = bi;
1815813a1f2bSShri Abhyankar   b->diag = bdiag;
1816f4259b30SLisandro Dalcin   b->ilen = NULL;
1817f4259b30SLisandro Dalcin   b->imax = NULL;
1818813a1f2bSShri Abhyankar   b->row  = isrow;
1819813a1f2bSShri Abhyankar   b->col  = iscol;
1820813a1f2bSShri Abhyankar   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1821813a1f2bSShri Abhyankar   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
1822813a1f2bSShri Abhyankar   b->icol = isicol;
18232205254eSKarl Rupp 
1824854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&b->solve_work);CHKERRQ(ierr);
1825813a1f2bSShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.
1826813a1f2bSShri Abhyankar      Allocate bdiag, solve_work, new a, new j */
18273bb1ff40SBarry Smith   ierr     = PetscLogObjectMemory((PetscObject)fact,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
1828f268cba8SShri Abhyankar   b->maxnz = b->nz = bdiag[0]+1;
18292205254eSKarl Rupp 
1830813a1f2bSShri Abhyankar   (fact)->info.factor_mallocs    = reallocs;
1831813a1f2bSShri Abhyankar   (fact)->info.fill_ratio_given  = f;
1832f268cba8SShri Abhyankar   (fact)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
183335233d99SShri Abhyankar   (fact)->ops->lufactornumeric   = MatLUFactorNumeric_SeqAIJ;
1834f2344125SBarry Smith   if (a->inode.size) {
1835f2344125SBarry Smith     (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode;
1836f2344125SBarry Smith   }
1837a02bda8eSBarry Smith   ierr = MatSeqAIJCheckInode_FactorLU(fact);CHKERRQ(ierr);
1838813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
1839813a1f2bSShri Abhyankar }
1840813a1f2bSShri Abhyankar 
1841ad04f41aSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
18426516239fSHong Zhang {
18436516239fSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
18446516239fSHong Zhang   IS                 isicol;
18456516239fSHong Zhang   PetscErrorCode     ierr;
18466516239fSHong Zhang   const PetscInt     *r,*ic;
1847ece542f9SHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j;
18486516239fSHong Zhang   PetscInt           *bi,*cols,nnz,*cols_lvl;
18496516239fSHong Zhang   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
18506516239fSHong Zhang   PetscInt           i,levels,diagonal_fill;
1851ace3abfcSBarry Smith   PetscBool          col_identity,row_identity;
18526516239fSHong Zhang   PetscReal          f;
18530298fd71SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=NULL;
18546516239fSHong Zhang   PetscBT            lnkbt;
18556516239fSHong Zhang   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
18560298fd71SBarry Smith   PetscFreeSpaceList free_space    =NULL,current_space=NULL;
18570298fd71SBarry Smith   PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL;
1858ace3abfcSBarry Smith   PetscBool          missing;
18596516239fSHong Zhang 
18606516239fSHong Zhang   PetscFunctionBegin;
1861e32f2f54SBarry Smith   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
1862ece542f9SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
1863ece542f9SHong Zhang   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
1864ece542f9SHong Zhang 
18656516239fSHong Zhang   f             = info->fill;
18666516239fSHong Zhang   levels        = (PetscInt)info->levels;
18676516239fSHong Zhang   diagonal_fill = (PetscInt)info->diagonal_fill;
18682205254eSKarl Rupp 
18696516239fSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
18706516239fSHong Zhang 
18716516239fSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
18726516239fSHong Zhang   ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
18736516239fSHong Zhang   if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */
1874f77e22a1SHong Zhang     ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr);
18752205254eSKarl Rupp 
1876ad04f41aSHong Zhang     (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ_inplace;
1877f2344125SBarry Smith     if (a->inode.size) {
1878f2344125SBarry Smith       (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
1879f2344125SBarry Smith     }
1880d5f3da31SBarry Smith     fact->factortype               = MAT_FACTOR_ILU;
1881719d5645SBarry Smith     (fact)->info.factor_mallocs    = 0;
1882719d5645SBarry Smith     (fact)->info.fill_ratio_given  = info->fill;
1883719d5645SBarry Smith     (fact)->info.fill_ratio_needed = 1.0;
18842205254eSKarl Rupp 
1885719d5645SBarry Smith     b    = (Mat_SeqAIJ*)(fact)->data;
188608480c60SBarry Smith     b->row  = isrow;
188708480c60SBarry Smith     b->col  = iscol;
188882bf6240SBarry Smith     b->icol = isicol;
1889854ce69bSBarry Smith     ierr    = PetscMalloc1((fact)->rmap->n+1,&b->solve_work);CHKERRQ(ierr);
1890c38d4ed2SBarry Smith     ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1891c38d4ed2SBarry Smith     ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
18923a40ed3dSBarry Smith     PetscFunctionReturn(0);
189308480c60SBarry Smith   }
189408480c60SBarry Smith 
1895898183ecSLois Curfman McInnes   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1896898183ecSLois Curfman McInnes   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
18979e25ed09SBarry Smith 
18981bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
1899854ce69bSBarry Smith   ierr  = PetscMalloc1(n+1,&bi);CHKERRQ(ierr);
1900854ce69bSBarry Smith   ierr  = PetscMalloc1(n+1,&bdiag);CHKERRQ(ierr);
1901a6df321fSShri Abhyankar   bi[0] = bdiag[0] = 0;
19026cf997b0SBarry Smith 
1903dcca6d9dSJed Brown   ierr = PetscMalloc2(n,&bj_ptr,n,&bjlvl_ptr);CHKERRQ(ierr);
19045a8e39fbSHong Zhang 
19055a8e39fbSHong Zhang   /* create a linked list for storing column indices of the active row */
19065a8e39fbSHong Zhang   nlnk = n + 1;
19075a8e39fbSHong Zhang   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
19085a8e39fbSHong Zhang 
19095a8e39fbSHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
19102f18eb33SBarry Smith   ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space);CHKERRQ(ierr);
19115a8e39fbSHong Zhang   current_space     = free_space;
19122f18eb33SBarry Smith   ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space_lvl);CHKERRQ(ierr);
19135a8e39fbSHong Zhang   current_space_lvl = free_space_lvl;
19145a8e39fbSHong Zhang 
19155a8e39fbSHong Zhang   for (i=0; i<n; i++) {
19165a8e39fbSHong Zhang     nzi = 0;
19176cf997b0SBarry Smith     /* copy current row into linked list */
19185a8e39fbSHong Zhang     nnz = ai[r[i]+1] - ai[r[i]];
1919e32f2f54SBarry Smith     if (!nnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i);
19205a8e39fbSHong Zhang     cols   = aj + ai[r[i]];
19215a8e39fbSHong Zhang     lnk[i] = -1; /* marker to indicate if diagonal exists */
19225a8e39fbSHong Zhang     ierr   = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
19235a8e39fbSHong Zhang     nzi   += nlnk;
19246cf997b0SBarry Smith 
19256cf997b0SBarry Smith     /* make sure diagonal entry is included */
19265a8e39fbSHong Zhang     if (diagonal_fill && lnk[i] == -1) {
19276cf997b0SBarry Smith       fm = n;
19285a8e39fbSHong Zhang       while (lnk[fm] < i) fm = lnk[fm];
19295a8e39fbSHong Zhang       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
19305a8e39fbSHong Zhang       lnk[fm]    = i;
19315a8e39fbSHong Zhang       lnk_lvl[i] = 0;
19325a8e39fbSHong Zhang       nzi++; dcount++;
19336cf997b0SBarry Smith     }
19346cf997b0SBarry Smith 
19355a8e39fbSHong Zhang     /* add pivot rows into the active row */
19365a8e39fbSHong Zhang     nzbd = 0;
19375a8e39fbSHong Zhang     prow = lnk[n];
19385a8e39fbSHong Zhang     while (prow < i) {
19395a8e39fbSHong Zhang       nnz      = bdiag[prow];
19405a8e39fbSHong Zhang       cols     = bj_ptr[prow] + nnz + 1;
19415a8e39fbSHong Zhang       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
19425a8e39fbSHong Zhang       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
19435a8e39fbSHong Zhang       ierr     = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
19445a8e39fbSHong Zhang       nzi     += nlnk;
19455a8e39fbSHong Zhang       prow     = lnk[prow];
19465a8e39fbSHong Zhang       nzbd++;
194756beaf04SBarry Smith     }
19485a8e39fbSHong Zhang     bdiag[i] = nzbd;
19495a8e39fbSHong Zhang     bi[i+1]  = bi[i] + nzi;
1950ecf371e4SBarry Smith 
19515a8e39fbSHong Zhang     /* if free space is not available, make more free space */
19525a8e39fbSHong Zhang     if (current_space->local_remaining<nzi) {
1953f91af8c7SBarry Smith       nnz  = PetscIntMultTruncate(nzi,n - i); /* estimated and max additional space needed */
1954a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
1955a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
19565a8e39fbSHong Zhang       reallocs++;
19575a8e39fbSHong Zhang     }
1958ecf371e4SBarry Smith 
19595a8e39fbSHong Zhang     /* copy data into free_space and free_space_lvl, then initialize lnk */
19605a8e39fbSHong Zhang     ierr         = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
19615a8e39fbSHong Zhang     bj_ptr[i]    = current_space->array;
19625a8e39fbSHong Zhang     bjlvl_ptr[i] = current_space_lvl->array;
19635a8e39fbSHong Zhang 
19645a8e39fbSHong Zhang     /* make sure the active row i has diagonal entry */
196565e19b50SBarry Smith     if (*(bj_ptr[i]+bdiag[i]) != i) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\ntry running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i);
19665a8e39fbSHong Zhang 
19675a8e39fbSHong Zhang     current_space->array               += nzi;
19685a8e39fbSHong Zhang     current_space->local_used          += nzi;
19695a8e39fbSHong Zhang     current_space->local_remaining     -= nzi;
19705a8e39fbSHong Zhang     current_space_lvl->array           += nzi;
19715a8e39fbSHong Zhang     current_space_lvl->local_used      += nzi;
19725a8e39fbSHong Zhang     current_space_lvl->local_remaining -= nzi;
19739e25ed09SBarry Smith   }
19745a8e39fbSHong Zhang 
1975898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
1976898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
19775a8e39fbSHong Zhang 
19785a8e39fbSHong Zhang   /* destroy list of free space and other temporary arrays */
1979854ce69bSBarry Smith   ierr = PetscMalloc1(bi[n]+1,&bj);CHKERRQ(ierr);
19806516239fSHong Zhang   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */
19815a8e39fbSHong Zhang   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
1982a1a86e44SBarry Smith   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
19830e83c824SBarry Smith   ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr);
19849e25ed09SBarry Smith 
19856cf91177SBarry Smith #if defined(PETSC_USE_INFO)
1986f64065bbSBarry Smith   {
19875a8e39fbSHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
198857622a8eSBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)f,(double)af);CHKERRQ(ierr);
198957622a8eSBarry Smith     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
199057622a8eSBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%g);\n",(double)af);CHKERRQ(ierr);
1991ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
1992335d9088SBarry Smith     if (diagonal_fill) {
1993955c1f14SBarry Smith       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals\n",dcount);CHKERRQ(ierr);
1994335d9088SBarry Smith     }
1995f64065bbSBarry Smith   }
199663ba0a88SBarry Smith #endif
1997bbb6d6a8SBarry Smith 
19989e25ed09SBarry Smith   /* put together the new matrix */
19990298fd71SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
20003bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)fact,(PetscObject)isicol);CHKERRQ(ierr);
2001719d5645SBarry Smith   b    = (Mat_SeqAIJ*)(fact)->data;
20022205254eSKarl Rupp 
2003e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
2004e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
20057c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
20062205254eSKarl Rupp 
2007785e854fSJed Brown   ierr = PetscMalloc1(bi[n],&b->a);CHKERRQ(ierr);
20085a8e39fbSHong Zhang   b->j = bj;
20095a8e39fbSHong Zhang   b->i = bi;
20105a8e39fbSHong Zhang   for (i=0; i<n; i++) bdiag[i] += bi[i];
20115a8e39fbSHong Zhang   b->diag = bdiag;
2012f4259b30SLisandro Dalcin   b->ilen = NULL;
2013f4259b30SLisandro Dalcin   b->imax = NULL;
2014416022c9SBarry Smith   b->row  = isrow;
2015416022c9SBarry Smith   b->col  = iscol;
2016c38d4ed2SBarry Smith   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
2017c38d4ed2SBarry Smith   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
201882bf6240SBarry Smith   b->icol = isicol;
2019854ce69bSBarry Smith   ierr    = PetscMalloc1(n+1,&b->solve_work);CHKERRQ(ierr);
2020416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
20215a8e39fbSHong Zhang      Allocate bdiag, solve_work, new a, new j */
20223bb1ff40SBarry Smith   ierr     = PetscLogObjectMemory((PetscObject)fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
20235a8e39fbSHong Zhang   b->maxnz = b->nz = bi[n];
20242205254eSKarl Rupp 
2025719d5645SBarry Smith   (fact)->info.factor_mallocs    = reallocs;
2026719d5645SBarry Smith   (fact)->info.fill_ratio_given  = f;
2027719d5645SBarry Smith   (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
2028ad04f41aSHong Zhang   (fact)->ops->lufactornumeric   =  MatLUFactorNumeric_SeqAIJ_inplace;
2029f2344125SBarry Smith   if (a->inode.size) {
2030f2344125SBarry Smith     (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
2031d3ac4fa3SBarry Smith   }
20323a40ed3dSBarry Smith   PetscFunctionReturn(0);
20339e25ed09SBarry Smith }
2034d5d45c9bSBarry Smith 
203535233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
20365f44c854SHong Zhang {
20375f44c854SHong Zhang   Mat            C = B;
20385f44c854SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
20395f44c854SHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
20405f44c854SHong Zhang   IS             ip=b->row,iip = b->icol;
20415f44c854SHong Zhang   PetscErrorCode ierr;
20425f44c854SHong Zhang   const PetscInt *rip,*riip;
20435f44c854SHong Zhang   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp;
20445f44c854SHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
20455f44c854SHong Zhang   PetscInt       k,jmin,jmax,*c2r,*il,col,nexti,ili,nz;
20465f44c854SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
2047ace3abfcSBarry Smith   PetscBool      perm_identity;
2048d90ac83dSHong Zhang   FactorShiftCtx sctx;
204998a93161SHong Zhang   PetscReal      rs;
205098a93161SHong Zhang   MatScalar      d,*v;
205198a93161SHong Zhang 
20525f44c854SHong Zhang   PetscFunctionBegin;
205398a93161SHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
2054d90ac83dSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
205598a93161SHong Zhang 
2056f4db908eSBarry Smith   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
205798a93161SHong Zhang     sctx.shift_top = info->zeropivot;
205898a93161SHong Zhang     for (i=0; i<mbs; i++) {
205998a93161SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
206098a93161SHong Zhang       d  = (aa)[a->diag[i]];
206198a93161SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
206298a93161SHong Zhang       v  = aa+ai[i];
206398a93161SHong Zhang       nz = ai[i+1] - ai[i];
20642205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
206598a93161SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
206698a93161SHong Zhang     }
206798a93161SHong Zhang     sctx.shift_top *= 1.1;
206898a93161SHong Zhang     sctx.nshift_max = 5;
206998a93161SHong Zhang     sctx.shift_lo   = 0.;
207098a93161SHong Zhang     sctx.shift_hi   = 1.;
207198a93161SHong Zhang   }
20725f44c854SHong Zhang 
20735f44c854SHong Zhang   ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr);
20745f44c854SHong Zhang   ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr);
20755f44c854SHong Zhang 
20765f44c854SHong Zhang   /* allocate working arrays
20775f44c854SHong Zhang      c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col
20785f44c854SHong Zhang      il:  for active k row, il[i] gives the index of the 1st nonzero entry in U[i,k:n-1] in bj and ba arrays
20795f44c854SHong Zhang   */
2080dcca6d9dSJed Brown   ierr = PetscMalloc3(mbs,&rtmp,mbs,&il,mbs,&c2r);CHKERRQ(ierr);
20815f44c854SHong Zhang 
208298a93161SHong Zhang   do {
208307b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
208498a93161SHong Zhang 
2085c5546cabSHong Zhang     for (i=0; i<mbs; i++) c2r[i] = mbs;
20862e987568SSatish Balay     if (mbs) il[0] = 0;
20875f44c854SHong Zhang 
20885f44c854SHong Zhang     for (k = 0; k<mbs; k++) {
20895f44c854SHong Zhang       /* zero rtmp */
20905f44c854SHong Zhang       nz    = bi[k+1] - bi[k];
20915f44c854SHong Zhang       bjtmp = bj + bi[k];
20925f44c854SHong Zhang       for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
20935f44c854SHong Zhang 
20945f44c854SHong Zhang       /* load in initial unfactored row */
20955f44c854SHong Zhang       bval = ba + bi[k];
20965f44c854SHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
20975f44c854SHong Zhang       for (j = jmin; j < jmax; j++) {
20985f44c854SHong Zhang         col = riip[aj[j]];
20995f44c854SHong Zhang         if (col >= k) { /* only take upper triangular entry */
21005f44c854SHong Zhang           rtmp[col] = aa[j];
21015f44c854SHong Zhang           *bval++   = 0.0; /* for in-place factorization */
21025f44c854SHong Zhang         }
21035f44c854SHong Zhang       }
210498a93161SHong Zhang       /* shift the diagonal of the matrix: ZeropivotApply() */
210598a93161SHong Zhang       rtmp[k] += sctx.shift_amount;  /* shift the diagonal of the matrix */
21065f44c854SHong Zhang 
21075f44c854SHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
21085f44c854SHong Zhang       dk = rtmp[k];
21095f44c854SHong Zhang       i  = c2r[k]; /* first row to be added to k_th row  */
21105f44c854SHong Zhang 
21115f44c854SHong Zhang       while (i < k) {
21125f44c854SHong Zhang         nexti = c2r[i]; /* next row to be added to k_th row */
21135f44c854SHong Zhang 
21145f44c854SHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
21155f44c854SHong Zhang         ili     = il[i]; /* index of first nonzero element in U(i,k:bms-1) */
21165f44c854SHong Zhang         uikdi   = -ba[ili]*ba[bdiag[i]]; /* diagonal(k) */
21175f44c854SHong Zhang         dk     += uikdi*ba[ili]; /* update diag[k] */
21185f44c854SHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
21195f44c854SHong Zhang 
21205f44c854SHong Zhang         /* add multiple of row i to k-th row */
21215f44c854SHong Zhang         jmin = ili + 1; jmax = bi[i+1];
21225f44c854SHong Zhang         if (jmin < jmax) {
21235f44c854SHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
21245f44c854SHong Zhang           /* update il and c2r for row i */
21255f44c854SHong Zhang           il[i] = jmin;
21265f44c854SHong Zhang           j     = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i;
21275f44c854SHong Zhang         }
21285f44c854SHong Zhang         i = nexti;
21295f44c854SHong Zhang       }
21305f44c854SHong Zhang 
21315f44c854SHong Zhang       /* copy data into U(k,:) */
213298a93161SHong Zhang       rs   = 0.0;
21335f44c854SHong Zhang       jmin = bi[k]; jmax = bi[k+1]-1;
21345f44c854SHong Zhang       if (jmin < jmax) {
21355f44c854SHong Zhang         for (j=jmin; j<jmax; j++) {
213698a93161SHong Zhang           col = bj[j]; ba[j] = rtmp[col]; rs += PetscAbsScalar(ba[j]);
21375f44c854SHong Zhang         }
21385f44c854SHong Zhang         /* add the k-th row into il and c2r */
21395f44c854SHong Zhang         il[k] = jmin;
21405f44c854SHong Zhang         i     = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k;
21415f44c854SHong Zhang       }
214298a93161SHong Zhang 
214398a93161SHong Zhang       /* MatPivotCheck() */
214498a93161SHong Zhang       sctx.rs = rs;
214598a93161SHong Zhang       sctx.pv = dk;
21464cccfbddSHong Zhang       ierr    = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr);
214707b50cabSHong Zhang       if (sctx.newshift) break;
214898a93161SHong Zhang       dk = sctx.pv;
214998a93161SHong Zhang 
215098a93161SHong Zhang       ba[bdiag[k]] = 1.0/dk; /* U(k,k) */
215198a93161SHong Zhang     }
215207b50cabSHong Zhang   } while (sctx.newshift);
21535f44c854SHong Zhang 
21540e83c824SBarry Smith   ierr = PetscFree3(rtmp,il,c2r);CHKERRQ(ierr);
21555f44c854SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
21565f44c854SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
21575f44c854SHong Zhang 
21585f44c854SHong Zhang   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
21595f44c854SHong Zhang   if (perm_identity) {
216035233d99SShri Abhyankar     B->ops->solve          = MatSolve_SeqSBAIJ_1_NaturalOrdering;
216135233d99SShri Abhyankar     B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering;
21622169352eSHong Zhang     B->ops->forwardsolve   = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering;
21632169352eSHong Zhang     B->ops->backwardsolve  = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering;
21645f44c854SHong Zhang   } else {
216535233d99SShri Abhyankar     B->ops->solve          = MatSolve_SeqSBAIJ_1;
216635233d99SShri Abhyankar     B->ops->solvetranspose = MatSolve_SeqSBAIJ_1;
216735233d99SShri Abhyankar     B->ops->forwardsolve   = MatForwardSolve_SeqSBAIJ_1;
216835233d99SShri Abhyankar     B->ops->backwardsolve  = MatBackwardSolve_SeqSBAIJ_1;
21695f44c854SHong Zhang   }
21705f44c854SHong Zhang 
21715f44c854SHong Zhang   C->assembled    = PETSC_TRUE;
21725f44c854SHong Zhang   C->preallocated = PETSC_TRUE;
21732205254eSKarl Rupp 
21745f44c854SHong Zhang   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
217598a93161SHong Zhang 
217698a93161SHong Zhang   /* MatPivotView() */
217798a93161SHong Zhang   if (sctx.nshift) {
2178f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
217957622a8eSBarry Smith       ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %g, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,(double)sctx.shift_amount,(double)sctx.shift_fraction,(double)sctx.shift_top);CHKERRQ(ierr);
2180f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
218157622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
2182f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS) {
218357622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %g\n",sctx.nshift,(double)info->shiftamount);CHKERRQ(ierr);
218498a93161SHong Zhang     }
218598a93161SHong Zhang   }
21865f44c854SHong Zhang   PetscFunctionReturn(0);
21875f44c854SHong Zhang }
21885f44c854SHong Zhang 
21890a3c351aSHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info)
2190a6175056SHong Zhang {
2191719d5645SBarry Smith   Mat            C = B;
21920968510aSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
21932ed133dbSHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
21949bfd6278SHong Zhang   IS             ip=b->row,iip = b->icol;
21952ed133dbSHong Zhang   PetscErrorCode ierr;
21965d0c19d7SBarry Smith   const PetscInt *rip,*riip;
2197c5546cabSHong Zhang   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol,*bjtmp;
21982ed133dbSHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
21992ed133dbSHong Zhang   PetscInt       k,jmin,jmax,*jl,*il,col,nexti,ili,nz;
2200622e2028SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
2201ace3abfcSBarry Smith   PetscBool      perm_identity;
22020e95ead3SHong Zhang   FactorShiftCtx sctx;
22030e95ead3SHong Zhang   PetscReal      rs;
22040e95ead3SHong Zhang   MatScalar      d,*v;
2205d5d45c9bSBarry Smith 
2206a6175056SHong Zhang   PetscFunctionBegin;
22070e95ead3SHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
22080e95ead3SHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
22090e95ead3SHong Zhang 
22100e95ead3SHong Zhang   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
22110e95ead3SHong Zhang     sctx.shift_top = info->zeropivot;
22120e95ead3SHong Zhang     for (i=0; i<mbs; i++) {
22130e95ead3SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
22140e95ead3SHong Zhang       d  = (aa)[a->diag[i]];
22150e95ead3SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
22160e95ead3SHong Zhang       v  = aa+ai[i];
22170e95ead3SHong Zhang       nz = ai[i+1] - ai[i];
22182205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
22190e95ead3SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
22200e95ead3SHong Zhang     }
22210e95ead3SHong Zhang     sctx.shift_top *= 1.1;
22220e95ead3SHong Zhang     sctx.nshift_max = 5;
22230e95ead3SHong Zhang     sctx.shift_lo   = 0.;
22240e95ead3SHong Zhang     sctx.shift_hi   = 1.;
22250e95ead3SHong Zhang   }
2226ee45ca4aSHong Zhang 
22272ed133dbSHong Zhang   ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr);
22289bfd6278SHong Zhang   ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr);
22292ed133dbSHong Zhang 
22302ed133dbSHong Zhang   /* initialization */
2231dcca6d9dSJed Brown   ierr = PetscMalloc3(mbs,&rtmp,mbs,&il,mbs,&jl);CHKERRQ(ierr);
22320e95ead3SHong Zhang 
22332ed133dbSHong Zhang   do {
223407b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
22350e95ead3SHong Zhang 
2236c5546cabSHong Zhang     for (i=0; i<mbs; i++) jl[i] = mbs;
2237c5546cabSHong Zhang     il[0] = 0;
22382ed133dbSHong Zhang 
22392ed133dbSHong Zhang     for (k = 0; k<mbs; k++) {
2240c5546cabSHong Zhang       /* zero rtmp */
2241c5546cabSHong Zhang       nz    = bi[k+1] - bi[k];
2242c5546cabSHong Zhang       bjtmp = bj + bi[k];
2243c5546cabSHong Zhang       for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
2244c5546cabSHong Zhang 
2245c05c3958SHong Zhang       bval = ba + bi[k];
22462ed133dbSHong Zhang       /* initialize k-th row by the perm[k]-th row of A */
22472ed133dbSHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
22482ed133dbSHong Zhang       for (j = jmin; j < jmax; j++) {
22499bfd6278SHong Zhang         col = riip[aj[j]];
22502ed133dbSHong Zhang         if (col >= k) { /* only take upper triangular entry */
22512ed133dbSHong Zhang           rtmp[col] = aa[j];
2252c05c3958SHong Zhang           *bval++   = 0.0; /* for in-place factorization */
22532ed133dbSHong Zhang         }
22542ed133dbSHong Zhang       }
225539e3d630SHong Zhang       /* shift the diagonal of the matrix */
2256540703acSHong Zhang       if (sctx.nshift) rtmp[k] += sctx.shift_amount;
22572ed133dbSHong Zhang 
22582ed133dbSHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
22592ed133dbSHong Zhang       dk = rtmp[k];
22602ed133dbSHong Zhang       i  = jl[k]; /* first row to be added to k_th row  */
22612ed133dbSHong Zhang 
22622ed133dbSHong Zhang       while (i < k) {
22632ed133dbSHong Zhang         nexti = jl[i]; /* next row to be added to k_th row */
22642ed133dbSHong Zhang 
22652ed133dbSHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
22662ed133dbSHong Zhang         ili     = il[i]; /* index of first nonzero element in U(i,k:bms-1) */
22672ed133dbSHong Zhang         uikdi   = -ba[ili]*ba[bi[i]]; /* diagonal(k) */
22682ed133dbSHong Zhang         dk     += uikdi*ba[ili];
22692ed133dbSHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
22702ed133dbSHong Zhang 
22712ed133dbSHong Zhang         /* add multiple of row i to k-th row */
22722ed133dbSHong Zhang         jmin = ili + 1; jmax = bi[i+1];
22732ed133dbSHong Zhang         if (jmin < jmax) {
22742ed133dbSHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
22752ed133dbSHong Zhang           /* update il and jl for row i */
22762ed133dbSHong Zhang           il[i] = jmin;
22772ed133dbSHong Zhang           j     = bj[jmin]; jl[i] = jl[j]; jl[j] = i;
22782ed133dbSHong Zhang         }
22792ed133dbSHong Zhang         i = nexti;
22802ed133dbSHong Zhang       }
22812ed133dbSHong Zhang 
22820697b6caSHong Zhang       /* shift the diagonals when zero pivot is detected */
22830697b6caSHong Zhang       /* compute rs=sum of abs(off-diagonal) */
22840697b6caSHong Zhang       rs   = 0.0;
22853c9e8598SHong Zhang       jmin = bi[k]+1;
22863c9e8598SHong Zhang       nz   = bi[k+1] - jmin;
22873c9e8598SHong Zhang       bcol = bj + jmin;
228804fbf559SBarry Smith       for (j=0; j<nz; j++) {
228904fbf559SBarry Smith         rs += PetscAbsScalar(rtmp[bcol[j]]);
22903c9e8598SHong Zhang       }
2291540703acSHong Zhang 
2292540703acSHong Zhang       sctx.rs = rs;
2293540703acSHong Zhang       sctx.pv = dk;
22944cccfbddSHong Zhang       ierr    = MatPivotCheck(B,A,info,&sctx,k);CHKERRQ(ierr);
229507b50cabSHong Zhang       if (sctx.newshift) break;
22960e95ead3SHong Zhang       dk = sctx.pv;
22973c9e8598SHong Zhang 
22983c9e8598SHong Zhang       /* copy data into U(k,:) */
229939e3d630SHong Zhang       ba[bi[k]] = 1.0/dk; /* U(k,k) */
230039e3d630SHong Zhang       jmin      = bi[k]+1; jmax = bi[k+1];
230139e3d630SHong Zhang       if (jmin < jmax) {
230239e3d630SHong Zhang         for (j=jmin; j<jmax; j++) {
2303c6ae9bfcSHong Zhang           col = bj[j]; ba[j] = rtmp[col];
23043c9e8598SHong Zhang         }
230539e3d630SHong Zhang         /* add the k-th row into il and jl */
23063c9e8598SHong Zhang         il[k] = jmin;
23073c9e8598SHong Zhang         i     = bj[jmin]; jl[k] = jl[i]; jl[i] = k;
23083c9e8598SHong Zhang       }
23093c9e8598SHong Zhang     }
231007b50cabSHong Zhang   } while (sctx.newshift);
23110e95ead3SHong Zhang 
23120e83c824SBarry Smith   ierr = PetscFree3(rtmp,il,jl);CHKERRQ(ierr);
231339e3d630SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
23149bfd6278SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
2315db4efbfdSBarry Smith 
2316db4efbfdSBarry Smith   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
2317db4efbfdSBarry Smith   if (perm_identity) {
23180a3c351aSHong Zhang     B->ops->solve          = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23190a3c351aSHong Zhang     B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23200a3c351aSHong Zhang     B->ops->forwardsolve   = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23210a3c351aSHong Zhang     B->ops->backwardsolve  = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
2322db4efbfdSBarry Smith   } else {
23230a3c351aSHong Zhang     B->ops->solve          = MatSolve_SeqSBAIJ_1_inplace;
23240a3c351aSHong Zhang     B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_inplace;
23250a3c351aSHong Zhang     B->ops->forwardsolve   = MatForwardSolve_SeqSBAIJ_1_inplace;
23260a3c351aSHong Zhang     B->ops->backwardsolve  = MatBackwardSolve_SeqSBAIJ_1_inplace;
2327db4efbfdSBarry Smith   }
2328db4efbfdSBarry Smith 
23293c9e8598SHong Zhang   C->assembled    = PETSC_TRUE;
23303c9e8598SHong Zhang   C->preallocated = PETSC_TRUE;
23312205254eSKarl Rupp 
2332d0f46423SBarry Smith   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
2333540703acSHong Zhang   if (sctx.nshift) {
2334f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
233557622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
2336f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
233757622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
23380697b6caSHong Zhang     }
23393c9e8598SHong Zhang   }
23403c9e8598SHong Zhang   PetscFunctionReturn(0);
23413c9e8598SHong Zhang }
2342a6175056SHong Zhang 
23438a76ed4fSHong Zhang /*
23448a76ed4fSHong Zhang    icc() under revised new data structure.
23458a76ed4fSHong Zhang    Factored arrays bj and ba are stored as
23468a76ed4fSHong Zhang      U(0,:),...,U(i,:),U(n-1,:)
23478a76ed4fSHong Zhang 
23488a76ed4fSHong Zhang    ui=fact->i is an array of size n+1, in which
23498a76ed4fSHong Zhang    ui+
23508a76ed4fSHong Zhang      ui[i]:  points to 1st entry of U(i,:),i=0,...,n-1
23518a76ed4fSHong Zhang      ui[n]:  points to U(n-1,n-1)+1
23528a76ed4fSHong Zhang 
23538a76ed4fSHong Zhang   udiag=fact->diag is an array of size n,in which
23548a76ed4fSHong Zhang      udiag[i]: points to diagonal of U(i,:), i=0,...,n-1
23558a76ed4fSHong Zhang 
23568a76ed4fSHong Zhang    U(i,:) contains udiag[i] as its last entry, i.e.,
23578a76ed4fSHong Zhang     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
23588a76ed4fSHong Zhang */
23598a76ed4fSHong Zhang 
236035233d99SShri Abhyankar PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2361a6175056SHong Zhang {
23620968510aSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
2363ed59401aSHong Zhang   Mat_SeqSBAIJ       *b;
2364dfbe8321SBarry Smith   PetscErrorCode     ierr;
2365ace3abfcSBarry Smith   PetscBool          perm_identity,missing;
23665f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
23675d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
2368ed59401aSHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
23690298fd71SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=NULL,d;
23705a8e39fbSHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
2371ed59401aSHong Zhang   PetscReal          fill          =info->fill,levels=info->levels;
23720298fd71SBarry Smith   PetscFreeSpaceList free_space    =NULL,current_space=NULL;
23730298fd71SBarry Smith   PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL;
23747a48dd6fSHong Zhang   PetscBT            lnkbt;
2375b635d36bSHong Zhang   IS                 iperm;
2376a6175056SHong Zhang 
2377a6175056SHong Zhang   PetscFunctionBegin;
2378e32f2f54SBarry Smith   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
237958ebbce7SBarry Smith   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
2380e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
2381653a6975SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
2382b635d36bSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
23837a48dd6fSHong Zhang 
2384854ce69bSBarry Smith   ierr  = PetscMalloc1(am+1,&ui);CHKERRQ(ierr);
2385854ce69bSBarry Smith   ierr  = PetscMalloc1(am+1,&udiag);CHKERRQ(ierr);
238639e3d630SHong Zhang   ui[0] = 0;
238739e3d630SHong Zhang 
23888a76ed4fSHong Zhang   /* ICC(0) without matrix ordering: simply rearrange column indices */
2389622e2028SHong Zhang   if (!levels && perm_identity) {
23905f44c854SHong Zhang     for (i=0; i<am; i++) {
2391c5546cabSHong Zhang       ncols    = ai[i+1] - a->diag[i];
2392c5546cabSHong Zhang       ui[i+1]  = ui[i] + ncols;
2393c5546cabSHong Zhang       udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */
2394dc1e2a3fSHong Zhang     }
2395854ce69bSBarry Smith     ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr);
2396dc1e2a3fSHong Zhang     cols = uj;
2397dc1e2a3fSHong Zhang     for (i=0; i<am; i++) {
23985f44c854SHong Zhang       aj    = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */
2399dc1e2a3fSHong Zhang       ncols = ai[i+1] - a->diag[i] -1;
24005f44c854SHong Zhang       for (j=0; j<ncols; j++) *cols++ = aj[j];
2401910cf402Sprj-       *cols++ = i; /* diagonal is located as the last entry of U(i,:) */
24025f44c854SHong Zhang     }
24035f44c854SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
24045f44c854SHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
24055f44c854SHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
24065f44c854SHong Zhang 
24075f44c854SHong Zhang     /* initialization */
2408854ce69bSBarry Smith     ierr = PetscMalloc1(am+1,&ajtmp);CHKERRQ(ierr);
24095f44c854SHong Zhang 
24105f44c854SHong Zhang     /* jl: linked list for storing indices of the pivot rows
24115f44c854SHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
2412dcca6d9dSJed Brown     ierr = PetscMalloc4(am,&uj_ptr,am,&uj_lvl_ptr,am,&jl,am,&il);CHKERRQ(ierr);
24135f44c854SHong Zhang     for (i=0; i<am; i++) {
24145f44c854SHong Zhang       jl[i] = am; il[i] = 0;
24155f44c854SHong Zhang     }
24165f44c854SHong Zhang 
24175f44c854SHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
24185f44c854SHong Zhang     nlnk = am + 1;
24195f44c854SHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
24205f44c854SHong Zhang 
242195b5ac22SHong Zhang     /* initial FreeSpace size is fill*(ai[am]+am)/2 */
24222f18eb33SBarry Smith     ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,(ai[am]+am)/2),&free_space);CHKERRQ(ierr);
24235f44c854SHong Zhang     current_space     = free_space;
24242f18eb33SBarry Smith     ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,(ai[am]+am)/2),&free_space_lvl);CHKERRQ(ierr);
24255f44c854SHong Zhang     current_space_lvl = free_space_lvl;
24265f44c854SHong Zhang 
24275f44c854SHong Zhang     for (k=0; k<am; k++) {  /* for each active row k */
24285f44c854SHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
24295f44c854SHong Zhang       nzk   = 0;
24305f44c854SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
2431e32f2f54SBarry Smith       if (!ncols) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k);
24325f44c854SHong Zhang       ncols_upper = 0;
24335f44c854SHong Zhang       for (j=0; j<ncols; j++) {
24345f44c854SHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
24355f44c854SHong Zhang         if (riip[i] >= k) { /* only take upper triangular entry */
24365f44c854SHong Zhang           ajtmp[ncols_upper] = i;
24375f44c854SHong Zhang           ncols_upper++;
24385f44c854SHong Zhang         }
24395f44c854SHong Zhang       }
24405f44c854SHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
24415f44c854SHong Zhang       nzk += nlnk;
24425f44c854SHong Zhang 
24435f44c854SHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
24445f44c854SHong Zhang       prow = jl[k]; /* 1st pivot row */
24455f44c854SHong Zhang 
24465f44c854SHong Zhang       while (prow < k) {
24475f44c854SHong Zhang         nextprow = jl[prow];
24485f44c854SHong Zhang 
24495f44c854SHong Zhang         /* merge prow into k-th row */
24505f44c854SHong Zhang         jmin  = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */
24515f44c854SHong Zhang         jmax  = ui[prow+1];
24525f44c854SHong Zhang         ncols = jmax-jmin;
24535f44c854SHong Zhang         i     = jmin - ui[prow];
24545f44c854SHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
24555f44c854SHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
24565f44c854SHong Zhang         j     = *(uj - 1);
24575f44c854SHong Zhang         ierr  = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
24585f44c854SHong Zhang         nzk  += nlnk;
24595f44c854SHong Zhang 
24605f44c854SHong Zhang         /* update il and jl for prow */
24615f44c854SHong Zhang         if (jmin < jmax) {
24625f44c854SHong Zhang           il[prow] = jmin;
24635f44c854SHong Zhang           j        = *cols; jl[prow] = jl[j]; jl[j] = prow;
24645f44c854SHong Zhang         }
24655f44c854SHong Zhang         prow = nextprow;
24665f44c854SHong Zhang       }
24675f44c854SHong Zhang 
24685f44c854SHong Zhang       /* if free space is not available, make more free space */
24695f44c854SHong Zhang       if (current_space->local_remaining<nzk) {
24705f44c854SHong Zhang         i    = am - k + 1; /* num of unfactored rows */
2471f91af8c7SBarry Smith         i    = PetscIntMultTruncate(i,PetscMin(nzk, i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
24725f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
24735f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
24745f44c854SHong Zhang         reallocs++;
24755f44c854SHong Zhang       }
24765f44c854SHong Zhang 
24775f44c854SHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
2478e32f2f54SBarry Smith       if (nzk == 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
24795f44c854SHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
24805f44c854SHong Zhang 
24815f44c854SHong Zhang       /* add the k-th row into il and jl */
24825f44c854SHong Zhang       if (nzk > 1) {
24835f44c854SHong Zhang         i     = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
24845f44c854SHong Zhang         jl[k] = jl[i]; jl[i] = k;
24855f44c854SHong Zhang         il[k] = ui[k] + 1;
24865f44c854SHong Zhang       }
24875f44c854SHong Zhang       uj_ptr[k]     = current_space->array;
24885f44c854SHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
24895f44c854SHong Zhang 
24905f44c854SHong Zhang       current_space->array           += nzk;
24915f44c854SHong Zhang       current_space->local_used      += nzk;
24925f44c854SHong Zhang       current_space->local_remaining -= nzk;
24935f44c854SHong Zhang 
24945f44c854SHong Zhang       current_space_lvl->array           += nzk;
24955f44c854SHong Zhang       current_space_lvl->local_used      += nzk;
24965f44c854SHong Zhang       current_space_lvl->local_remaining -= nzk;
24975f44c854SHong Zhang 
24985f44c854SHong Zhang       ui[k+1] = ui[k] + nzk;
24995f44c854SHong Zhang     }
25005f44c854SHong Zhang 
25015f44c854SHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
25025f44c854SHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
25030e83c824SBarry Smith     ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr);
25045f44c854SHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
25055f44c854SHong Zhang 
25069263d837SHong Zhang     /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */
2507854ce69bSBarry Smith     ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr);
250835233d99SShri Abhyankar     ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor  */
25095f44c854SHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
25105f44c854SHong Zhang     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
25115f44c854SHong Zhang 
25125f44c854SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
25135f44c854SHong Zhang 
25145f44c854SHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
25155f44c854SHong Zhang   b               = (Mat_SeqSBAIJ*)(fact)->data;
25165f44c854SHong Zhang   b->singlemalloc = PETSC_FALSE;
25172205254eSKarl Rupp 
2518854ce69bSBarry Smith   ierr = PetscMalloc1(ui[am]+1,&b->a);CHKERRQ(ierr);
25192205254eSKarl Rupp 
25205f44c854SHong Zhang   b->j             = uj;
25215f44c854SHong Zhang   b->i             = ui;
25225f44c854SHong Zhang   b->diag          = udiag;
25237f53bb6cSHong Zhang   b->free_diag     = PETSC_TRUE;
2524f4259b30SLisandro Dalcin   b->ilen          = NULL;
2525f4259b30SLisandro Dalcin   b->imax          = NULL;
25265f44c854SHong Zhang   b->row           = perm;
25275f44c854SHong Zhang   b->col           = perm;
25285f44c854SHong Zhang   ierr             = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
25295f44c854SHong Zhang   ierr             = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
25305f44c854SHong Zhang   b->icol          = iperm;
25315f44c854SHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
25322205254eSKarl Rupp 
2533854ce69bSBarry Smith   ierr = PetscMalloc1(am+1,&b->solve_work);CHKERRQ(ierr);
25343bb1ff40SBarry Smith   ierr = PetscLogObjectMemory((PetscObject)fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
25352205254eSKarl Rupp 
25365f44c854SHong Zhang   b->maxnz   = b->nz = ui[am];
25375f44c854SHong Zhang   b->free_a  = PETSC_TRUE;
25385f44c854SHong Zhang   b->free_ij = PETSC_TRUE;
25395f44c854SHong Zhang 
2540f284f12aSHong Zhang   fact->info.factor_mallocs   = reallocs;
2541f284f12aSHong Zhang   fact->info.fill_ratio_given = fill;
25425f44c854SHong Zhang   if (ai[am] != 0) {
25436a69fef8SHong Zhang     /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */
254495b5ac22SHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am);
25455f44c854SHong Zhang   } else {
2546f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
25475f44c854SHong Zhang   }
25489263d837SHong Zhang #if defined(PETSC_USE_INFO)
25499263d837SHong Zhang   if (ai[am] != 0) {
25509263d837SHong Zhang     PetscReal af = fact->info.fill_ratio_needed;
255157622a8eSBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)fill,(double)af);CHKERRQ(ierr);
255257622a8eSBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
255357622a8eSBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g) for best performance.\n",(double)af);CHKERRQ(ierr);
25549263d837SHong Zhang   } else {
2555dbf6bb8dSprj-     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
25569263d837SHong Zhang   }
25579263d837SHong Zhang #endif
255835233d99SShri Abhyankar   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
25595f44c854SHong Zhang   PetscFunctionReturn(0);
25605f44c854SHong Zhang }
25615f44c854SHong Zhang 
2562ad04f41aSHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
25635f44c854SHong Zhang {
25645f44c854SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
25655f44c854SHong Zhang   Mat_SeqSBAIJ       *b;
25665f44c854SHong Zhang   PetscErrorCode     ierr;
2567ace3abfcSBarry Smith   PetscBool          perm_identity,missing;
25685f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
25695f44c854SHong Zhang   const PetscInt     *rip,*riip;
25705f44c854SHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
25710298fd71SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=NULL,d;
25725f44c854SHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
25735f44c854SHong Zhang   PetscReal          fill          =info->fill,levels=info->levels;
25740298fd71SBarry Smith   PetscFreeSpaceList free_space    =NULL,current_space=NULL;
25750298fd71SBarry Smith   PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL;
25765f44c854SHong Zhang   PetscBT            lnkbt;
25775f44c854SHong Zhang   IS                 iperm;
25785f44c854SHong Zhang 
25795f44c854SHong Zhang   PetscFunctionBegin;
2580e32f2f54SBarry Smith   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
25815f44c854SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
2582e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
25835f44c854SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
25845f44c854SHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
25855f44c854SHong Zhang 
2586854ce69bSBarry Smith   ierr  = PetscMalloc1(am+1,&ui);CHKERRQ(ierr);
2587854ce69bSBarry Smith   ierr  = PetscMalloc1(am+1,&udiag);CHKERRQ(ierr);
25885f44c854SHong Zhang   ui[0] = 0;
25895f44c854SHong Zhang 
25905f44c854SHong Zhang   /* ICC(0) without matrix ordering: simply copies fill pattern */
25915f44c854SHong Zhang   if (!levels && perm_identity) {
25925f44c854SHong Zhang 
25935f44c854SHong Zhang     for (i=0; i<am; i++) {
25945f44c854SHong Zhang       ui[i+1]  = ui[i] + ai[i+1] - a->diag[i];
25955f44c854SHong Zhang       udiag[i] = ui[i];
2596ed59401aSHong Zhang     }
2597854ce69bSBarry Smith     ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr);
259839e3d630SHong Zhang     cols = uj;
2599ed59401aSHong Zhang     for (i=0; i<am; i++) {
2600ed59401aSHong Zhang       aj    = a->j + a->diag[i];
260139e3d630SHong Zhang       ncols = ui[i+1] - ui[i];
260239e3d630SHong Zhang       for (j=0; j<ncols; j++) *cols++ = *aj++;
2603ed59401aSHong Zhang     }
260439e3d630SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
2605b635d36bSHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
2606b635d36bSHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
2607b635d36bSHong Zhang 
26087a48dd6fSHong Zhang     /* initialization */
2609854ce69bSBarry Smith     ierr = PetscMalloc1(am+1,&ajtmp);CHKERRQ(ierr);
26107a48dd6fSHong Zhang 
26117a48dd6fSHong Zhang     /* jl: linked list for storing indices of the pivot rows
26127a48dd6fSHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
2613dcca6d9dSJed Brown     ierr = PetscMalloc4(am,&uj_ptr,am,&uj_lvl_ptr,am,&jl,am,&il);CHKERRQ(ierr);
26147a48dd6fSHong Zhang     for (i=0; i<am; i++) {
26157a48dd6fSHong Zhang       jl[i] = am; il[i] = 0;
26167a48dd6fSHong Zhang     }
26177a48dd6fSHong Zhang 
26187a48dd6fSHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
26197a48dd6fSHong Zhang     nlnk = am + 1;
26202ed133dbSHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
26217a48dd6fSHong Zhang 
26227a48dd6fSHong Zhang     /* initial FreeSpace size is fill*(ai[am]+1) */
26232f18eb33SBarry Smith     ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,ai[am]+1),&free_space);CHKERRQ(ierr);
26247a48dd6fSHong Zhang     current_space     = free_space;
26252f18eb33SBarry Smith     ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,ai[am]+1),&free_space_lvl);CHKERRQ(ierr);
26267a48dd6fSHong Zhang     current_space_lvl = free_space_lvl;
26277a48dd6fSHong Zhang 
26287a48dd6fSHong Zhang     for (k=0; k<am; k++) {  /* for each active row k */
26297a48dd6fSHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
26307a48dd6fSHong Zhang       nzk   = 0;
2631622e2028SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
2632e32f2f54SBarry Smith       if (!ncols) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k);
2633622e2028SHong Zhang       ncols_upper = 0;
2634622e2028SHong Zhang       for (j=0; j<ncols; j++) {
2635b635d36bSHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
2636b635d36bSHong Zhang         if (riip[i] >= k) { /* only take upper triangular entry */
26375a8e39fbSHong Zhang           ajtmp[ncols_upper] = i;
2638622e2028SHong Zhang           ncols_upper++;
2639622e2028SHong Zhang         }
2640622e2028SHong Zhang       }
2641b635d36bSHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
26427a48dd6fSHong Zhang       nzk += nlnk;
26437a48dd6fSHong Zhang 
26447a48dd6fSHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
26457a48dd6fSHong Zhang       prow = jl[k]; /* 1st pivot row */
26467a48dd6fSHong Zhang 
26477a48dd6fSHong Zhang       while (prow < k) {
26487a48dd6fSHong Zhang         nextprow = jl[prow];
26497a48dd6fSHong Zhang 
26507a48dd6fSHong Zhang         /* merge prow into k-th row */
26517a48dd6fSHong Zhang         jmin  = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */
26527a48dd6fSHong Zhang         jmax  = ui[prow+1];
26537a48dd6fSHong Zhang         ncols = jmax-jmin;
2654ed59401aSHong Zhang         i     = jmin - ui[prow];
2655ed59401aSHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
26565a8e39fbSHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
26575a8e39fbSHong Zhang         j     = *(uj - 1);
26585a8e39fbSHong Zhang         ierr  = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
26597a48dd6fSHong Zhang         nzk  += nlnk;
26607a48dd6fSHong Zhang 
26617a48dd6fSHong Zhang         /* update il and jl for prow */
26627a48dd6fSHong Zhang         if (jmin < jmax) {
26637a48dd6fSHong Zhang           il[prow] = jmin;
26647a48dd6fSHong Zhang           j        = *cols; jl[prow] = jl[j]; jl[j] = prow;
26657a48dd6fSHong Zhang         }
26667a48dd6fSHong Zhang         prow = nextprow;
26677a48dd6fSHong Zhang       }
26687a48dd6fSHong Zhang 
26697a48dd6fSHong Zhang       /* if free space is not available, make more free space */
26707a48dd6fSHong Zhang       if (current_space->local_remaining<nzk) {
26717a48dd6fSHong Zhang         i    = am - k + 1; /* num of unfactored rows */
2672f91af8c7SBarry Smith         i    = PetscIntMultTruncate(i,PetscMin(nzk, i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
2673a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
2674a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
26757a48dd6fSHong Zhang         reallocs++;
26767a48dd6fSHong Zhang       }
26777a48dd6fSHong Zhang 
26787a48dd6fSHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
267965e19b50SBarry Smith       if (!nzk) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
26802ed133dbSHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
26817a48dd6fSHong Zhang 
26827a48dd6fSHong Zhang       /* add the k-th row into il and jl */
268339e3d630SHong Zhang       if (nzk > 1) {
26847a48dd6fSHong Zhang         i     = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
26857a48dd6fSHong Zhang         jl[k] = jl[i]; jl[i] = k;
26867a48dd6fSHong Zhang         il[k] = ui[k] + 1;
26877a48dd6fSHong Zhang       }
26887a48dd6fSHong Zhang       uj_ptr[k]     = current_space->array;
26897a48dd6fSHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
26907a48dd6fSHong Zhang 
26917a48dd6fSHong Zhang       current_space->array           += nzk;
26927a48dd6fSHong Zhang       current_space->local_used      += nzk;
26937a48dd6fSHong Zhang       current_space->local_remaining -= nzk;
26947a48dd6fSHong Zhang 
26957a48dd6fSHong Zhang       current_space_lvl->array           += nzk;
26967a48dd6fSHong Zhang       current_space_lvl->local_used      += nzk;
26977a48dd6fSHong Zhang       current_space_lvl->local_remaining -= nzk;
26987a48dd6fSHong Zhang 
26997a48dd6fSHong Zhang       ui[k+1] = ui[k] + nzk;
27007a48dd6fSHong Zhang     }
27017a48dd6fSHong Zhang 
27026cf91177SBarry Smith #if defined(PETSC_USE_INFO)
27037a48dd6fSHong Zhang     if (ai[am] != 0) {
270439e3d630SHong Zhang       PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]);
270557622a8eSBarry Smith       ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)fill,(double)af);CHKERRQ(ierr);
270657622a8eSBarry Smith       ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
270757622a8eSBarry Smith       ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g) for best performance.\n",(double)af);CHKERRQ(ierr);
27087a48dd6fSHong Zhang     } else {
2709dbf6bb8dSprj-       ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
27107a48dd6fSHong Zhang     }
271163ba0a88SBarry Smith #endif
27127a48dd6fSHong Zhang 
27137a48dd6fSHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
2714b635d36bSHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
27150e83c824SBarry Smith     ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr);
27165a8e39fbSHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
27177a48dd6fSHong Zhang 
27187a48dd6fSHong Zhang     /* destroy list of free space and other temporary array(s) */
2719854ce69bSBarry Smith     ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr);
2720a1a86e44SBarry Smith     ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
27212ed133dbSHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
2722a1a86e44SBarry Smith     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
27237a48dd6fSHong Zhang 
272439e3d630SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
272539e3d630SHong Zhang 
27267a48dd6fSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
27277a48dd6fSHong Zhang 
2728f284f12aSHong Zhang   b               = (Mat_SeqSBAIJ*)fact->data;
27297a48dd6fSHong Zhang   b->singlemalloc = PETSC_FALSE;
27302205254eSKarl Rupp 
2731854ce69bSBarry Smith   ierr = PetscMalloc1(ui[am]+1,&b->a);CHKERRQ(ierr);
27322205254eSKarl Rupp 
27337a48dd6fSHong Zhang   b->j         = uj;
27347a48dd6fSHong Zhang   b->i         = ui;
27355f44c854SHong Zhang   b->diag      = udiag;
27367f53bb6cSHong Zhang   b->free_diag = PETSC_TRUE;
2737f4259b30SLisandro Dalcin   b->ilen      = NULL;
2738f4259b30SLisandro Dalcin   b->imax      = NULL;
27397a48dd6fSHong Zhang   b->row       = perm;
2740b635d36bSHong Zhang   b->col       = perm;
27412205254eSKarl Rupp 
2742b635d36bSHong Zhang   ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
2743b635d36bSHong Zhang   ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
27442205254eSKarl Rupp 
2745b635d36bSHong Zhang   b->icol          = iperm;
27467a48dd6fSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
2747854ce69bSBarry Smith   ierr             = PetscMalloc1(am+1,&b->solve_work);CHKERRQ(ierr);
27483bb1ff40SBarry Smith   ierr             = PetscLogObjectMemory((PetscObject)fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
27497a48dd6fSHong Zhang   b->maxnz         = b->nz = ui[am];
2750e6b907acSBarry Smith   b->free_a        = PETSC_TRUE;
2751e6b907acSBarry Smith   b->free_ij       = PETSC_TRUE;
27527a48dd6fSHong Zhang 
2753f284f12aSHong Zhang   fact->info.factor_mallocs   = reallocs;
2754f284f12aSHong Zhang   fact->info.fill_ratio_given = fill;
27557a48dd6fSHong Zhang   if (ai[am] != 0) {
2756f284f12aSHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
27577a48dd6fSHong Zhang   } else {
2758f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
27597a48dd6fSHong Zhang   }
27600a3c351aSHong Zhang   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace;
2761a6175056SHong Zhang   PetscFunctionReturn(0);
2762a6175056SHong Zhang }
2763d5d45c9bSBarry Smith 
276435233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
27650be760fbSHong Zhang {
27660be760fbSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
27670be760fbSHong Zhang   Mat_SeqSBAIJ       *b;
27680be760fbSHong Zhang   PetscErrorCode     ierr;
27699186b105SHong Zhang   PetscBool          perm_identity,missing;
27700be760fbSHong Zhang   PetscReal          fill = info->fill;
27710be760fbSHong Zhang   const PetscInt     *rip,*riip;
27720be760fbSHong Zhang   PetscInt           i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
27730be760fbSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
27740be760fbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr,*udiag;
27750298fd71SBarry Smith   PetscFreeSpaceList free_space=NULL,current_space=NULL;
27760be760fbSHong Zhang   PetscBT            lnkbt;
27770be760fbSHong Zhang   IS                 iperm;
27780be760fbSHong Zhang 
27790be760fbSHong Zhang   PetscFunctionBegin;
2780e32f2f54SBarry Smith   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
27819186b105SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
27829186b105SHong Zhang   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
27839186b105SHong Zhang 
27840be760fbSHong Zhang   /* check whether perm is the identity mapping */
27850be760fbSHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
27860be760fbSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
27870be760fbSHong Zhang   ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
27880be760fbSHong Zhang   ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
27890be760fbSHong Zhang 
27900be760fbSHong Zhang   /* initialization */
2791854ce69bSBarry Smith   ierr  = PetscMalloc1(am+1,&ui);CHKERRQ(ierr);
2792854ce69bSBarry Smith   ierr  = PetscMalloc1(am+1,&udiag);CHKERRQ(ierr);
27930be760fbSHong Zhang   ui[0] = 0;
27940be760fbSHong Zhang 
27950be760fbSHong Zhang   /* jl: linked list for storing indices of the pivot rows
27960be760fbSHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
2797dcca6d9dSJed Brown   ierr = PetscMalloc4(am,&ui_ptr,am,&jl,am,&il,am,&cols);CHKERRQ(ierr);
27980be760fbSHong Zhang   for (i=0; i<am; i++) {
27990be760fbSHong Zhang     jl[i] = am; il[i] = 0;
28000be760fbSHong Zhang   }
28010be760fbSHong Zhang 
28020be760fbSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
28030be760fbSHong Zhang   nlnk = am + 1;
28040be760fbSHong Zhang   ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28050be760fbSHong Zhang 
280695b5ac22SHong Zhang   /* initial FreeSpace size is fill*(ai[am]+am)/2 */
28072f18eb33SBarry Smith   ierr          = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,(ai[am]+am)/2),&free_space);CHKERRQ(ierr);
28080be760fbSHong Zhang   current_space = free_space;
28090be760fbSHong Zhang 
28100be760fbSHong Zhang   for (k=0; k<am; k++) {  /* for each active row k */
28110be760fbSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
28120be760fbSHong Zhang     nzk   = 0;
28130be760fbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
2814e32f2f54SBarry Smith     if (!ncols) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k);
28150be760fbSHong Zhang     ncols_upper = 0;
28160be760fbSHong Zhang     for (j=0; j<ncols; j++) {
28170be760fbSHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
28180be760fbSHong Zhang       if (i >= k) { /* only take upper triangular entry */
28190be760fbSHong Zhang         cols[ncols_upper] = i;
28200be760fbSHong Zhang         ncols_upper++;
28210be760fbSHong Zhang       }
28220be760fbSHong Zhang     }
28230be760fbSHong Zhang     ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28240be760fbSHong Zhang     nzk += nlnk;
28250be760fbSHong Zhang 
28260be760fbSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
28270be760fbSHong Zhang     prow = jl[k]; /* 1st pivot row */
28280be760fbSHong Zhang 
28290be760fbSHong Zhang     while (prow < k) {
28300be760fbSHong Zhang       nextprow = jl[prow];
28310be760fbSHong Zhang       /* merge prow into k-th row */
28320be760fbSHong Zhang       jmin   = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */
28330be760fbSHong Zhang       jmax   = ui[prow+1];
28340be760fbSHong Zhang       ncols  = jmax-jmin;
28350be760fbSHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
28360be760fbSHong Zhang       ierr   = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28370be760fbSHong Zhang       nzk   += nlnk;
28380be760fbSHong Zhang 
28390be760fbSHong Zhang       /* update il and jl for prow */
28400be760fbSHong Zhang       if (jmin < jmax) {
28410be760fbSHong Zhang         il[prow] = jmin;
28422205254eSKarl Rupp         j        = *uj_ptr;
28432205254eSKarl Rupp         jl[prow] = jl[j];
28442205254eSKarl Rupp         jl[j]    = prow;
28450be760fbSHong Zhang       }
28460be760fbSHong Zhang       prow = nextprow;
28470be760fbSHong Zhang     }
28480be760fbSHong Zhang 
28490be760fbSHong Zhang     /* if free space is not available, make more free space */
28500be760fbSHong Zhang     if (current_space->local_remaining<nzk) {
28510be760fbSHong Zhang       i    = am - k + 1; /* num of unfactored rows */
2852f91af8c7SBarry Smith       i    = PetscIntMultTruncate(i,PetscMin(nzk,i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
28530be760fbSHong Zhang       ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
28540be760fbSHong Zhang       reallocs++;
28550be760fbSHong Zhang     }
28560be760fbSHong Zhang 
28570be760fbSHong Zhang     /* copy data into free space, then initialize lnk */
28580be760fbSHong Zhang     ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
28590be760fbSHong Zhang 
28600be760fbSHong Zhang     /* add the k-th row into il and jl */
28617b056e98SHong Zhang     if (nzk > 1) {
28620be760fbSHong Zhang       i     = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
28630be760fbSHong Zhang       jl[k] = jl[i]; jl[i] = k;
28640be760fbSHong Zhang       il[k] = ui[k] + 1;
28650be760fbSHong Zhang     }
28660be760fbSHong Zhang     ui_ptr[k] = current_space->array;
28672205254eSKarl Rupp 
28680be760fbSHong Zhang     current_space->array           += nzk;
28690be760fbSHong Zhang     current_space->local_used      += nzk;
28700be760fbSHong Zhang     current_space->local_remaining -= nzk;
28710be760fbSHong Zhang 
28720be760fbSHong Zhang     ui[k+1] = ui[k] + nzk;
28730be760fbSHong Zhang   }
28740be760fbSHong Zhang 
28750be760fbSHong Zhang   ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
28760be760fbSHong Zhang   ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
28770be760fbSHong Zhang   ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr);
28780be760fbSHong Zhang 
28799263d837SHong Zhang   /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */
2880854ce69bSBarry Smith   ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr);
288135233d99SShri Abhyankar   ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor */
28820be760fbSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
28830be760fbSHong Zhang 
28840be760fbSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
28850be760fbSHong Zhang 
2886f284f12aSHong Zhang   b               = (Mat_SeqSBAIJ*)fact->data;
28870be760fbSHong Zhang   b->singlemalloc = PETSC_FALSE;
28880be760fbSHong Zhang   b->free_a       = PETSC_TRUE;
28890be760fbSHong Zhang   b->free_ij      = PETSC_TRUE;
28902205254eSKarl Rupp 
2891854ce69bSBarry Smith   ierr = PetscMalloc1(ui[am]+1,&b->a);CHKERRQ(ierr);
28922205254eSKarl Rupp 
28930be760fbSHong Zhang   b->j         = uj;
28940be760fbSHong Zhang   b->i         = ui;
28950be760fbSHong Zhang   b->diag      = udiag;
28960be760fbSHong Zhang   b->free_diag = PETSC_TRUE;
2897f4259b30SLisandro Dalcin   b->ilen      = NULL;
2898f4259b30SLisandro Dalcin   b->imax      = NULL;
28990be760fbSHong Zhang   b->row       = perm;
29000be760fbSHong Zhang   b->col       = perm;
290126fbe8dcSKarl Rupp 
29020be760fbSHong Zhang   ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
29030be760fbSHong Zhang   ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
29042205254eSKarl Rupp 
29050be760fbSHong Zhang   b->icol          = iperm;
29060be760fbSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
290726fbe8dcSKarl Rupp 
2908854ce69bSBarry Smith   ierr = PetscMalloc1(am+1,&b->solve_work);CHKERRQ(ierr);
29093bb1ff40SBarry Smith   ierr = PetscLogObjectMemory((PetscObject)fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
29102205254eSKarl Rupp 
29110be760fbSHong Zhang   b->maxnz = b->nz = ui[am];
29120be760fbSHong Zhang 
2913f284f12aSHong Zhang   fact->info.factor_mallocs   = reallocs;
2914f284f12aSHong Zhang   fact->info.fill_ratio_given = fill;
29150be760fbSHong Zhang   if (ai[am] != 0) {
29166a69fef8SHong Zhang     /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */
291795b5ac22SHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am);
29180be760fbSHong Zhang   } else {
2919f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
29200be760fbSHong Zhang   }
29219263d837SHong Zhang #if defined(PETSC_USE_INFO)
29229263d837SHong Zhang   if (ai[am] != 0) {
29239263d837SHong Zhang     PetscReal af = fact->info.fill_ratio_needed;
292457622a8eSBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)fill,(double)af);CHKERRQ(ierr);
292557622a8eSBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
292657622a8eSBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g) for best performance.\n",(double)af);CHKERRQ(ierr);
29279263d837SHong Zhang   } else {
2928dbf6bb8dSprj-     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
29299263d837SHong Zhang   }
29309263d837SHong Zhang #endif
293135233d99SShri Abhyankar   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
29320be760fbSHong Zhang   PetscFunctionReturn(0);
29330be760fbSHong Zhang }
29340be760fbSHong Zhang 
2935ad04f41aSHong Zhang PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2936f76d2b81SHong Zhang {
2937f76d2b81SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
293810c27e3dSHong Zhang   Mat_SeqSBAIJ       *b;
2939dfbe8321SBarry Smith   PetscErrorCode     ierr;
29409186b105SHong Zhang   PetscBool          perm_identity,missing;
294110c27e3dSHong Zhang   PetscReal          fill = info->fill;
29425d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
29435d0c19d7SBarry Smith   PetscInt           i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
294410c27e3dSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
29452ed133dbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr;
29460298fd71SBarry Smith   PetscFreeSpaceList free_space=NULL,current_space=NULL;
294710c27e3dSHong Zhang   PetscBT            lnkbt;
29482ed133dbSHong Zhang   IS                 iperm;
2949f76d2b81SHong Zhang 
2950f76d2b81SHong Zhang   PetscFunctionBegin;
2951e32f2f54SBarry Smith   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
29529186b105SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
29539186b105SHong Zhang   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
29549186b105SHong Zhang 
29552ed133dbSHong Zhang   /* check whether perm is the identity mapping */
2956f76d2b81SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
29572ed133dbSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
29582ed133dbSHong Zhang   ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
29599bfd6278SHong Zhang   ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
296010c27e3dSHong Zhang 
296110c27e3dSHong Zhang   /* initialization */
2962854ce69bSBarry Smith   ierr  = PetscMalloc1(am+1,&ui);CHKERRQ(ierr);
296310c27e3dSHong Zhang   ui[0] = 0;
296410c27e3dSHong Zhang 
296510c27e3dSHong Zhang   /* jl: linked list for storing indices of the pivot rows
29661393bc97SHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
2967dcca6d9dSJed Brown   ierr = PetscMalloc4(am,&ui_ptr,am,&jl,am,&il,am,&cols);CHKERRQ(ierr);
29681393bc97SHong Zhang   for (i=0; i<am; i++) {
29691393bc97SHong Zhang     jl[i] = am; il[i] = 0;
2970f76d2b81SHong Zhang   }
297110c27e3dSHong Zhang 
297210c27e3dSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
29731393bc97SHong Zhang   nlnk = am + 1;
29741393bc97SHong Zhang   ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
297510c27e3dSHong Zhang 
29761393bc97SHong Zhang   /* initial FreeSpace size is fill*(ai[am]+1) */
29772f18eb33SBarry Smith   ierr          = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,ai[am]+1),&free_space);CHKERRQ(ierr);
297810c27e3dSHong Zhang   current_space = free_space;
297910c27e3dSHong Zhang 
29801393bc97SHong Zhang   for (k=0; k<am; k++) {  /* for each active row k */
298110c27e3dSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
298210c27e3dSHong Zhang     nzk   = 0;
29832ed133dbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
2984e32f2f54SBarry Smith     if (!ncols) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k);
29852ed133dbSHong Zhang     ncols_upper = 0;
29862ed133dbSHong Zhang     for (j=0; j<ncols; j++) {
29879bfd6278SHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
29882ed133dbSHong Zhang       if (i >= k) { /* only take upper triangular entry */
29892ed133dbSHong Zhang         cols[ncols_upper] = i;
29902ed133dbSHong Zhang         ncols_upper++;
29912ed133dbSHong Zhang       }
29922ed133dbSHong Zhang     }
29931393bc97SHong Zhang     ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
299410c27e3dSHong Zhang     nzk += nlnk;
299510c27e3dSHong Zhang 
299610c27e3dSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
299710c27e3dSHong Zhang     prow = jl[k]; /* 1st pivot row */
299810c27e3dSHong Zhang 
299910c27e3dSHong Zhang     while (prow < k) {
300010c27e3dSHong Zhang       nextprow = jl[prow];
300110c27e3dSHong Zhang       /* merge prow into k-th row */
30021393bc97SHong Zhang       jmin   = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */
300310c27e3dSHong Zhang       jmax   = ui[prow+1];
300410c27e3dSHong Zhang       ncols  = jmax-jmin;
30051393bc97SHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
30065a8e39fbSHong Zhang       ierr   = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
300710c27e3dSHong Zhang       nzk   += nlnk;
300810c27e3dSHong Zhang 
300910c27e3dSHong Zhang       /* update il and jl for prow */
301010c27e3dSHong Zhang       if (jmin < jmax) {
301110c27e3dSHong Zhang         il[prow] = jmin;
30122ed133dbSHong Zhang         j        = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow;
301310c27e3dSHong Zhang       }
301410c27e3dSHong Zhang       prow = nextprow;
301510c27e3dSHong Zhang     }
301610c27e3dSHong Zhang 
301710c27e3dSHong Zhang     /* if free space is not available, make more free space */
301810c27e3dSHong Zhang     if (current_space->local_remaining<nzk) {
30191393bc97SHong Zhang       i    = am - k + 1; /* num of unfactored rows */
302010c27e3dSHong Zhang       i    = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
3021a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
302210c27e3dSHong Zhang       reallocs++;
302310c27e3dSHong Zhang     }
302410c27e3dSHong Zhang 
302510c27e3dSHong Zhang     /* copy data into free space, then initialize lnk */
30261393bc97SHong Zhang     ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
302710c27e3dSHong Zhang 
302810c27e3dSHong Zhang     /* add the k-th row into il and jl */
302910c27e3dSHong Zhang     if (nzk-1 > 0) {
30301393bc97SHong Zhang       i     = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
303110c27e3dSHong Zhang       jl[k] = jl[i]; jl[i] = k;
303210c27e3dSHong Zhang       il[k] = ui[k] + 1;
303310c27e3dSHong Zhang     }
30342ed133dbSHong Zhang     ui_ptr[k] = current_space->array;
30352205254eSKarl Rupp 
303610c27e3dSHong Zhang     current_space->array           += nzk;
303710c27e3dSHong Zhang     current_space->local_used      += nzk;
303810c27e3dSHong Zhang     current_space->local_remaining -= nzk;
303910c27e3dSHong Zhang 
304010c27e3dSHong Zhang     ui[k+1] = ui[k] + nzk;
304110c27e3dSHong Zhang   }
304210c27e3dSHong Zhang 
30436cf91177SBarry Smith #if defined(PETSC_USE_INFO)
30441393bc97SHong Zhang   if (ai[am] != 0) {
30451393bc97SHong Zhang     PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]);
304657622a8eSBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)fill,(double)af);CHKERRQ(ierr);
304757622a8eSBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
304857622a8eSBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g) for best performance.\n",(double)af);CHKERRQ(ierr);
304910c27e3dSHong Zhang   } else {
3050dbf6bb8dSprj-     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
305110c27e3dSHong Zhang   }
305263ba0a88SBarry Smith #endif
305310c27e3dSHong Zhang 
305410c27e3dSHong Zhang   ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
30559bfd6278SHong Zhang   ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
30560e83c824SBarry Smith   ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr);
305710c27e3dSHong Zhang 
305810c27e3dSHong Zhang   /* destroy list of free space and other temporary array(s) */
3059854ce69bSBarry Smith   ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr);
3060a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
306110c27e3dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
306210c27e3dSHong Zhang 
306310c27e3dSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
306410c27e3dSHong Zhang 
3065f284f12aSHong Zhang   b               = (Mat_SeqSBAIJ*)fact->data;
306610c27e3dSHong Zhang   b->singlemalloc = PETSC_FALSE;
3067e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
3068e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
30692205254eSKarl Rupp 
3070854ce69bSBarry Smith   ierr = PetscMalloc1(ui[am]+1,&b->a);CHKERRQ(ierr);
30712205254eSKarl Rupp 
307210c27e3dSHong Zhang   b->j    = uj;
307310c27e3dSHong Zhang   b->i    = ui;
3074f4259b30SLisandro Dalcin   b->diag = NULL;
3075f4259b30SLisandro Dalcin   b->ilen = NULL;
3076f4259b30SLisandro Dalcin   b->imax = NULL;
307710c27e3dSHong Zhang   b->row  = perm;
30789bfd6278SHong Zhang   b->col  = perm;
30792205254eSKarl Rupp 
30809bfd6278SHong Zhang   ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
30819bfd6278SHong Zhang   ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
30822205254eSKarl Rupp 
30839bfd6278SHong Zhang   b->icol          = iperm;
308410c27e3dSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
30852205254eSKarl Rupp 
3086854ce69bSBarry Smith   ierr     = PetscMalloc1(am+1,&b->solve_work);CHKERRQ(ierr);
30873bb1ff40SBarry Smith   ierr     = PetscLogObjectMemory((PetscObject)fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
30881393bc97SHong Zhang   b->maxnz = b->nz = ui[am];
308910c27e3dSHong Zhang 
3090f284f12aSHong Zhang   fact->info.factor_mallocs   = reallocs;
3091f284f12aSHong Zhang   fact->info.fill_ratio_given = fill;
30921393bc97SHong Zhang   if (ai[am] != 0) {
3093f284f12aSHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
309410c27e3dSHong Zhang   } else {
3095f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
309610c27e3dSHong Zhang   }
30970a3c351aSHong Zhang   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace;
3098f76d2b81SHong Zhang   PetscFunctionReturn(0);
3099f76d2b81SHong Zhang }
3100599ef60dSHong Zhang 
310135233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx)
3102813a1f2bSShri Abhyankar {
3103813a1f2bSShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
3104813a1f2bSShri Abhyankar   PetscErrorCode    ierr;
3105813a1f2bSShri Abhyankar   PetscInt          n   = A->rmap->n;
3106813a1f2bSShri Abhyankar   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag,*vi;
3107813a1f2bSShri Abhyankar   PetscScalar       *x,sum;
3108813a1f2bSShri Abhyankar   const PetscScalar *b;
3109813a1f2bSShri Abhyankar   const MatScalar   *aa = a->a,*v;
3110813a1f2bSShri Abhyankar   PetscInt          i,nz;
3111813a1f2bSShri Abhyankar 
3112813a1f2bSShri Abhyankar   PetscFunctionBegin;
3113813a1f2bSShri Abhyankar   if (!n) PetscFunctionReturn(0);
3114813a1f2bSShri Abhyankar 
31153649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
3116fdc842d1SBarry Smith   ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr);
3117813a1f2bSShri Abhyankar 
3118813a1f2bSShri Abhyankar   /* forward solve the lower triangular */
3119813a1f2bSShri Abhyankar   x[0] = b[0];
3120813a1f2bSShri Abhyankar   v    = aa;
3121813a1f2bSShri Abhyankar   vi   = aj;
3122813a1f2bSShri Abhyankar   for (i=1; i<n; i++) {
3123813a1f2bSShri Abhyankar     nz  = ai[i+1] - ai[i];
3124813a1f2bSShri Abhyankar     sum = b[i];
3125813a1f2bSShri Abhyankar     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
3126813a1f2bSShri Abhyankar     v   += nz;
3127813a1f2bSShri Abhyankar     vi  += nz;
3128813a1f2bSShri Abhyankar     x[i] = sum;
3129813a1f2bSShri Abhyankar   }
3130813a1f2bSShri Abhyankar 
3131813a1f2bSShri Abhyankar   /* backward solve the upper triangular */
313262fbd6afSShri Abhyankar   for (i=n-1; i>=0; i--) {
31333c0119dfSShri Abhyankar     v   = aa + adiag[i+1] + 1;
31343c0119dfSShri Abhyankar     vi  = aj + adiag[i+1] + 1;
3135813a1f2bSShri Abhyankar     nz  = adiag[i] - adiag[i+1]-1;
3136813a1f2bSShri Abhyankar     sum = x[i];
3137813a1f2bSShri Abhyankar     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
31383c0119dfSShri Abhyankar     x[i] = sum*v[nz]; /* x[i]=aa[adiag[i]]*sum; v++; */
3139813a1f2bSShri Abhyankar   }
3140813a1f2bSShri Abhyankar 
3141813a1f2bSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
31423649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
3143fdc842d1SBarry Smith   ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr);
3144813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
3145813a1f2bSShri Abhyankar }
3146813a1f2bSShri Abhyankar 
314735233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx)
3148f268cba8SShri Abhyankar {
3149f268cba8SShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
3150f268cba8SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
3151f268cba8SShri Abhyankar   PetscErrorCode    ierr;
3152f268cba8SShri Abhyankar   PetscInt          i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,*adiag = a->diag,nz;
3153f268cba8SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c;
3154f268cba8SShri Abhyankar   PetscScalar       *x,*tmp,sum;
3155f268cba8SShri Abhyankar   const PetscScalar *b;
3156f268cba8SShri Abhyankar   const MatScalar   *aa = a->a,*v;
3157f268cba8SShri Abhyankar 
3158f268cba8SShri Abhyankar   PetscFunctionBegin;
3159f268cba8SShri Abhyankar   if (!n) PetscFunctionReturn(0);
3160f268cba8SShri Abhyankar 
31613649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
3162fdc842d1SBarry Smith   ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr);
3163f268cba8SShri Abhyankar   tmp  = a->solve_work;
3164f268cba8SShri Abhyankar 
3165f268cba8SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
3166f268cba8SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
3167f268cba8SShri Abhyankar 
3168f268cba8SShri Abhyankar   /* forward solve the lower triangular */
3169f268cba8SShri Abhyankar   tmp[0] = b[r[0]];
3170f268cba8SShri Abhyankar   v      = aa;
3171f268cba8SShri Abhyankar   vi     = aj;
3172f268cba8SShri Abhyankar   for (i=1; i<n; i++) {
3173f268cba8SShri Abhyankar     nz  = ai[i+1] - ai[i];
3174f268cba8SShri Abhyankar     sum = b[r[i]];
3175f268cba8SShri Abhyankar     PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
3176f268cba8SShri Abhyankar     tmp[i] = sum;
3177f268cba8SShri Abhyankar     v     += nz; vi += nz;
3178f268cba8SShri Abhyankar   }
3179f268cba8SShri Abhyankar 
3180f268cba8SShri Abhyankar   /* backward solve the upper triangular */
3181f268cba8SShri Abhyankar   for (i=n-1; i>=0; i--) {
31823c0119dfSShri Abhyankar     v   = aa + adiag[i+1]+1;
31833c0119dfSShri Abhyankar     vi  = aj + adiag[i+1]+1;
3184f268cba8SShri Abhyankar     nz  = adiag[i]-adiag[i+1]-1;
3185f268cba8SShri Abhyankar     sum = tmp[i];
3186f268cba8SShri Abhyankar     PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
3187f268cba8SShri Abhyankar     x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */
3188f268cba8SShri Abhyankar   }
3189f268cba8SShri Abhyankar 
3190f268cba8SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
3191f268cba8SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
31923649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
3193fdc842d1SBarry Smith   ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr);
3194ca0c957dSBarry Smith   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
3195f268cba8SShri Abhyankar   PetscFunctionReturn(0);
3196f268cba8SShri Abhyankar }
3197f268cba8SShri Abhyankar 
3198fe97e370SBarry Smith /*
3199bebe2cf6SSatish Balay     This will get a new name and become a varient of MatILUFactor_SeqAIJ() there is no longer separate functions in the matrix function table for dt factors
3200fe97e370SBarry Smith */
32011d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact)
32021d098868SHong Zhang {
32031d098868SHong Zhang   Mat            B = *fact;
3204599ef60dSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*b;
3205599ef60dSHong Zhang   IS             isicol;
3206599ef60dSHong Zhang   PetscErrorCode ierr;
32071d098868SHong Zhang   const PetscInt *r,*ic;
32081d098868SHong Zhang   PetscInt       i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag;
3209dc3a2fd3SHong Zhang   PetscInt       *bi,*bj,*bdiag,*bdiag_rev;
3210f61a948aSLisandro Dalcin   PetscInt       row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au;
32111d098868SHong Zhang   PetscInt       nlnk,*lnk;
32121d098868SHong Zhang   PetscBT        lnkbt;
3213ace3abfcSBarry Smith   PetscBool      row_identity,icol_identity;
32148aee2decSHong Zhang   MatScalar      *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp;
32151d098868SHong Zhang   const PetscInt *ics;
32161d098868SHong Zhang   PetscInt       j,nz,*pj,*bjtmp,k,ncut,*jtmp;
3217a2ea699eSBarry Smith   PetscReal      dt     =info->dt,shift=info->shiftamount;
3218f61a948aSLisandro Dalcin   PetscInt       dtcount=(PetscInt)info->dtcount,nnz_max;
3219ace3abfcSBarry Smith   PetscBool      missing;
3220599ef60dSHong Zhang 
3221599ef60dSHong Zhang   PetscFunctionBegin;
3222f61a948aSLisandro Dalcin   if (dt      == PETSC_DEFAULT) dt = 0.005;
3223f61a948aSLisandro Dalcin   if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax);
3224f61a948aSLisandro Dalcin 
32251d098868SHong Zhang   /* ------- symbolic factorization, can be reused ---------*/
32261d098868SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
3227e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
32281d098868SHong Zhang   adiag=a->diag;
3229599ef60dSHong Zhang 
3230599ef60dSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
3231599ef60dSHong Zhang 
3232599ef60dSHong Zhang   /* bdiag is location of diagonal in factor */
3233854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&bdiag);CHKERRQ(ierr);     /* becomes b->diag */
3234854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&bdiag_rev);CHKERRQ(ierr); /* temporary */
3235599ef60dSHong Zhang 
32361d098868SHong Zhang   /* allocate row pointers bi */
3237854ce69bSBarry Smith   ierr = PetscMalloc1(2*n+2,&bi);CHKERRQ(ierr);
32381d098868SHong Zhang 
32391d098868SHong Zhang   /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */
3240393d3a68SHong Zhang   if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */
32411d098868SHong Zhang   nnz_max = ai[n]+2*n*dtcount+2;
32428fc3a347SHong Zhang 
3243854ce69bSBarry Smith   ierr = PetscMalloc1(nnz_max+1,&bj);CHKERRQ(ierr);
3244854ce69bSBarry Smith   ierr = PetscMalloc1(nnz_max+1,&ba);CHKERRQ(ierr);
3245599ef60dSHong Zhang 
3246599ef60dSHong Zhang   /* put together the new matrix */
32470298fd71SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
32483bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)B,(PetscObject)isicol);CHKERRQ(ierr);
3249f284f12aSHong Zhang   b    = (Mat_SeqAIJ*)B->data;
32502205254eSKarl Rupp 
3251599ef60dSHong Zhang   b->free_a       = PETSC_TRUE;
3252599ef60dSHong Zhang   b->free_ij      = PETSC_TRUE;
3253599ef60dSHong Zhang   b->singlemalloc = PETSC_FALSE;
32542205254eSKarl Rupp 
3255599ef60dSHong Zhang   b->a    = ba;
3256599ef60dSHong Zhang   b->j    = bj;
3257599ef60dSHong Zhang   b->i    = bi;
3258599ef60dSHong Zhang   b->diag = bdiag;
3259f4259b30SLisandro Dalcin   b->ilen = NULL;
3260f4259b30SLisandro Dalcin   b->imax = NULL;
3261599ef60dSHong Zhang   b->row  = isrow;
3262599ef60dSHong Zhang   b->col  = iscol;
3263599ef60dSHong Zhang   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
3264599ef60dSHong Zhang   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
3265599ef60dSHong Zhang   b->icol = isicol;
3266599ef60dSHong Zhang 
3267854ce69bSBarry Smith   ierr     = PetscMalloc1(n+1,&b->solve_work);CHKERRQ(ierr);
32683bb1ff40SBarry Smith   ierr     = PetscLogObjectMemory((PetscObject)B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
32691d098868SHong Zhang   b->maxnz = nnz_max;
3270599ef60dSHong Zhang 
3271d5f3da31SBarry Smith   B->factortype            = MAT_FACTOR_ILUDT;
3272f284f12aSHong Zhang   B->info.factor_mallocs   = 0;
3273f284f12aSHong Zhang   B->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]);
32741d098868SHong Zhang   /* ------- end of symbolic factorization ---------*/
3275599ef60dSHong Zhang 
3276599ef60dSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
3277599ef60dSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
3278599ef60dSHong Zhang   ics  = ic;
3279599ef60dSHong Zhang 
3280599ef60dSHong Zhang   /* linked list for storing column indices of the active row */
3281599ef60dSHong Zhang   nlnk = n + 1;
3282599ef60dSHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
32831d098868SHong Zhang 
32841d098868SHong Zhang   /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */
3285dcca6d9dSJed Brown   ierr = PetscMalloc2(n,&im,n,&jtmp);CHKERRQ(ierr);
32861d098868SHong Zhang   /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */
3287dcca6d9dSJed Brown   ierr = PetscMalloc2(n,&rtmp,n,&vtmp);CHKERRQ(ierr);
3288580bdb30SBarry Smith   ierr = PetscArrayzero(rtmp,n);CHKERRQ(ierr);
3289599ef60dSHong Zhang 
3290599ef60dSHong Zhang   bi[0]        = 0;
32918fc3a347SHong Zhang   bdiag[0]     = nnz_max-1; /* location of diag[0] in factor B */
3292dc3a2fd3SHong Zhang   bdiag_rev[n] = bdiag[0];
32938fc3a347SHong Zhang   bi[2*n+1]    = bdiag[0]+1; /* endof bj and ba array */
3294599ef60dSHong Zhang   for (i=0; i<n; i++) {
3295599ef60dSHong Zhang     /* copy initial fill into linked list */
32968369b28dSHong Zhang     nzi = ai[r[i]+1] - ai[r[i]];
3297e32f2f54SBarry Smith     if (!nzi) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i);
32988369b28dSHong Zhang     nzi_al = adiag[r[i]] - ai[r[i]];
32998369b28dSHong Zhang     nzi_au = ai[r[i]+1] - adiag[r[i]] -1;
3300599ef60dSHong Zhang     ajtmp  = aj + ai[r[i]];
33018369b28dSHong Zhang     ierr   = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3302599ef60dSHong Zhang 
3303599ef60dSHong Zhang     /* load in initial (unfactored row) */
33041d098868SHong Zhang     aatmp = a->a + ai[r[i]];
33058369b28dSHong Zhang     for (j=0; j<nzi; j++) {
33061d098868SHong Zhang       rtmp[ics[*ajtmp++]] = *aatmp++;
3307599ef60dSHong Zhang     }
3308599ef60dSHong Zhang 
3309599ef60dSHong Zhang     /* add pivot rows into linked list */
3310599ef60dSHong Zhang     row = lnk[n];
3311599ef60dSHong Zhang     while (row < i) {
33121d098868SHong Zhang       nzi_bl = bi[row+1] - bi[row] + 1;
33131d098868SHong Zhang       bjtmp  = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */
33141d098868SHong Zhang       ierr   = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr);
3315599ef60dSHong Zhang       nzi   += nlnk;
3316599ef60dSHong Zhang       row    = lnk[row];
3317599ef60dSHong Zhang     }
3318599ef60dSHong Zhang 
33198369b28dSHong Zhang     /* copy data from lnk into jtmp, then initialize lnk */
33208369b28dSHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr);
3321599ef60dSHong Zhang 
3322599ef60dSHong Zhang     /* numerical factorization */
33238369b28dSHong Zhang     bjtmp = jtmp;
3324599ef60dSHong Zhang     row   = *bjtmp++; /* 1st pivot row */
3325599ef60dSHong Zhang     while (row < i) {
3326599ef60dSHong Zhang       pc         = rtmp + row;
33273c2a7987SHong Zhang       pv         = ba + bdiag[row]; /* 1./(diag of the pivot row) */
33283c2a7987SHong Zhang       multiplier = (*pc) * (*pv);
3329599ef60dSHong Zhang       *pc        = multiplier;
33303c2a7987SHong Zhang       if (PetscAbsScalar(*pc) > dt) { /* apply tolerance dropping rule */
33311d098868SHong Zhang         pj = bj + bdiag[row+1] + 1;         /* point to 1st entry of U(row,:) */
33321d098868SHong Zhang         pv = ba + bdiag[row+1] + 1;
33331d098868SHong Zhang         nz = bdiag[row] - bdiag[row+1] - 1;         /* num of entries in U(row,:), excluding diagonal */
33341d098868SHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
3335ca0c957dSBarry Smith         ierr = PetscLogFlops(1+2.0*nz);CHKERRQ(ierr);
3336599ef60dSHong Zhang       }
3337599ef60dSHong Zhang       row = *bjtmp++;
3338599ef60dSHong Zhang     }
3339599ef60dSHong Zhang 
33408369b28dSHong Zhang     /* copy sparse rtmp into contiguous vtmp; separate L and U part */
33418aee2decSHong Zhang     diag_tmp = rtmp[i];  /* save diagonal value - may not needed?? */
33428369b28dSHong Zhang     nzi_bl   = 0; j = 0;
33438369b28dSHong Zhang     while (jtmp[j] < i) { /* Note: jtmp is sorted */
33448aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
33458369b28dSHong Zhang       nzi_bl++; j++;
33468369b28dSHong Zhang     }
33478369b28dSHong Zhang     nzi_bu = nzi - nzi_bl -1;
33488369b28dSHong Zhang     while (j < nzi) {
33498aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
33508369b28dSHong Zhang       j++;
33518369b28dSHong Zhang     }
33528369b28dSHong Zhang 
3353599ef60dSHong Zhang     bjtmp = bj + bi[i];
3354599ef60dSHong Zhang     batmp = ba + bi[i];
33558369b28dSHong Zhang     /* apply level dropping rule to L part */
33568369b28dSHong Zhang     ncut = nzi_al + dtcount;
33578369b28dSHong Zhang     if (ncut < nzi_bl) {
33588369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr);
33598369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr);
3360599ef60dSHong Zhang     } else {
33618369b28dSHong Zhang       ncut = nzi_bl;
33628369b28dSHong Zhang     }
33638369b28dSHong Zhang     for (j=0; j<ncut; j++) {
33648369b28dSHong Zhang       bjtmp[j] = jtmp[j];
33658369b28dSHong Zhang       batmp[j] = vtmp[j];
33668369b28dSHong Zhang     }
33671d098868SHong Zhang     bi[i+1] = bi[i] + ncut;
33688369b28dSHong Zhang     nzi     = ncut + 1;
33698369b28dSHong Zhang 
33708369b28dSHong Zhang     /* apply level dropping rule to U part */
33718369b28dSHong Zhang     ncut = nzi_au + dtcount;
33728369b28dSHong Zhang     if (ncut < nzi_bu) {
33738369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr);
33748369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr);
33758369b28dSHong Zhang     } else {
33768369b28dSHong Zhang       ncut = nzi_bu;
33778369b28dSHong Zhang     }
33788369b28dSHong Zhang     nzi += ncut;
33791d098868SHong Zhang 
33801d098868SHong Zhang     /* mark bdiagonal */
33811d098868SHong Zhang     bdiag[i+1]       = bdiag[i] - (ncut + 1);
3382dc3a2fd3SHong Zhang     bdiag_rev[n-i-1] = bdiag[i+1];
33838fc3a347SHong Zhang     bi[2*n - i]      = bi[2*n - i +1] - (ncut + 1);
33841d098868SHong Zhang     bjtmp            = bj + bdiag[i];
33851d098868SHong Zhang     batmp            = ba + bdiag[i];
33861d098868SHong Zhang     *bjtmp           = i;
33878aee2decSHong Zhang     *batmp           = diag_tmp; /* rtmp[i]; */
3388c9c72f8fSHong Zhang     if (*batmp == 0.0) {
3389c9c72f8fSHong Zhang       *batmp = dt+shift;
3390c9c72f8fSHong Zhang     }
3391b78a477dSHong Zhang     *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */
33921d098868SHong Zhang 
33931d098868SHong Zhang     bjtmp = bj + bdiag[i+1]+1;
33941d098868SHong Zhang     batmp = ba + bdiag[i+1]+1;
33958369b28dSHong Zhang     for (k=0; k<ncut; k++) {
33961d098868SHong Zhang       bjtmp[k] = jtmp[nzi_bl+1+k];
33971d098868SHong Zhang       batmp[k] = vtmp[nzi_bl+1+k];
3398599ef60dSHong Zhang     }
3399599ef60dSHong Zhang 
3400599ef60dSHong Zhang     im[i] = nzi;   /* used by PetscLLAddSortedLU() */
34018369b28dSHong Zhang   } /* for (i=0; i<n; i++) */
3402e32f2f54SBarry Smith   if (bi[n] >= bdiag[n]) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"end of L array %d cannot >= the beginning of U array %d",bi[n],bdiag[n]);
3403599ef60dSHong Zhang 
3404599ef60dSHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3405599ef60dSHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3406599ef60dSHong Zhang 
3407599ef60dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
34080e83c824SBarry Smith   ierr = PetscFree2(im,jtmp);CHKERRQ(ierr);
34090e83c824SBarry Smith   ierr = PetscFree2(rtmp,vtmp);CHKERRQ(ierr);
34101bfa06eaSJed Brown   ierr = PetscFree(bdiag_rev);CHKERRQ(ierr);
3411599ef60dSHong Zhang 
3412599ef60dSHong Zhang   ierr     = PetscLogFlops(B->cmap->n);CHKERRQ(ierr);
34131d098868SHong Zhang   b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n];
3414599ef60dSHong Zhang 
3415599ef60dSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
3416599ef60dSHong Zhang   ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr);
3417599ef60dSHong Zhang   if (row_identity && icol_identity) {
341835233d99SShri Abhyankar     B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
3419599ef60dSHong Zhang   } else {
342035233d99SShri Abhyankar     B->ops->solve = MatSolve_SeqAIJ;
3421599ef60dSHong Zhang   }
3422599ef60dSHong Zhang 
3423f4259b30SLisandro Dalcin   B->ops->solveadd          = NULL;
3424f4259b30SLisandro Dalcin   B->ops->solvetranspose    = NULL;
3425f4259b30SLisandro Dalcin   B->ops->solvetransposeadd = NULL;
3426f4259b30SLisandro Dalcin   B->ops->matsolve          = NULL;
3427599ef60dSHong Zhang   B->assembled              = PETSC_TRUE;
3428599ef60dSHong Zhang   B->preallocated           = PETSC_TRUE;
3429599ef60dSHong Zhang   PetscFunctionReturn(0);
3430599ef60dSHong Zhang }
3431599ef60dSHong Zhang 
34323c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */
3433fe97e370SBarry Smith /*
3434bebe2cf6SSatish Balay     This will get a new name and become a varient of MatILUFactor_SeqAIJ() there is no longer separate functions in the matrix function table for dt factors
3435fe97e370SBarry Smith */
3436fe97e370SBarry Smith 
34377087cfbeSBarry Smith PetscErrorCode  MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info)
34383c2a7987SHong Zhang {
34393c2a7987SHong Zhang   PetscErrorCode ierr;
34403c2a7987SHong Zhang 
34413c2a7987SHong Zhang   PetscFunctionBegin;
34423c2a7987SHong Zhang   ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr);
34433c2a7987SHong Zhang   PetscFunctionReturn(0);
34443c2a7987SHong Zhang }
34453c2a7987SHong Zhang 
34463c2a7987SHong Zhang /*
34473c2a7987SHong Zhang    same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors
34483c2a7987SHong Zhang    - intend to replace existing MatLUFactorNumeric_SeqAIJ()
34493c2a7987SHong Zhang */
3450fe97e370SBarry Smith /*
3451bebe2cf6SSatish Balay     This will get a new name and become a varient of MatILUFactor_SeqAIJ() there is no longer separate functions in the matrix function table for dt factors
3452fe97e370SBarry Smith */
3453fe97e370SBarry Smith 
34547087cfbeSBarry Smith PetscErrorCode  MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info)
34553c2a7987SHong Zhang {
34563c2a7987SHong Zhang   Mat            C     =fact;
34573c2a7987SHong Zhang   Mat_SeqAIJ     *a    =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data;
34583c2a7987SHong Zhang   IS             isrow = b->row,isicol = b->icol;
34593c2a7987SHong Zhang   PetscErrorCode ierr;
34603c2a7987SHong Zhang   const PetscInt *r,*ic,*ics;
3461b78a477dSHong Zhang   PetscInt       i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
3462b78a477dSHong Zhang   PetscInt       *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj;
34633c2a7987SHong Zhang   MatScalar      *rtmp,*pc,multiplier,*v,*pv,*aa=a->a;
3464182b8fbaSHong Zhang   PetscReal      dt=info->dt,shift=info->shiftamount;
3465ace3abfcSBarry Smith   PetscBool      row_identity, col_identity;
34663c2a7987SHong Zhang 
34673c2a7987SHong Zhang   PetscFunctionBegin;
34683c2a7987SHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
34693c2a7987SHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
3470854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&rtmp);CHKERRQ(ierr);
34713c2a7987SHong Zhang   ics  = ic;
34723c2a7987SHong Zhang 
34733c2a7987SHong Zhang   for (i=0; i<n; i++) {
3474b78a477dSHong Zhang     /* initialize rtmp array */
3475b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i];       /* num of nozeros in L(i,:) */
34763c2a7987SHong Zhang     bjtmp = bj + bi[i];
3477b78a477dSHong Zhang     for  (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0;
3478b78a477dSHong Zhang     rtmp[i] = 0.0;
3479b78a477dSHong Zhang     nzu     = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */
3480b78a477dSHong Zhang     bjtmp   = bj + bdiag[i+1] + 1;
3481b78a477dSHong Zhang     for  (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0;
34823c2a7987SHong Zhang 
3483b78a477dSHong Zhang     /* load in initial unfactored row of A */
34843c2a7987SHong Zhang     nz    = ai[r[i]+1] - ai[r[i]];
34853c2a7987SHong Zhang     ajtmp = aj + ai[r[i]];
34863c2a7987SHong Zhang     v     = aa + ai[r[i]];
34873c2a7987SHong Zhang     for (j=0; j<nz; j++) {
3488b78a477dSHong Zhang       rtmp[ics[*ajtmp++]] = v[j];
34893c2a7987SHong Zhang     }
34903c2a7987SHong Zhang 
3491b78a477dSHong Zhang     /* numerical factorization */
3492b78a477dSHong Zhang     bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */
3493b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i]; /* num of entries in L(i,:) */
3494b78a477dSHong Zhang     k     = 0;
3495b78a477dSHong Zhang     while (k < nzl) {
34963c2a7987SHong Zhang       row        = *bjtmp++;
34973c2a7987SHong Zhang       pc         = rtmp + row;
3498b78a477dSHong Zhang       pv         = b->a + bdiag[row]; /* 1./(diag of the pivot row) */
3499b78a477dSHong Zhang       multiplier = (*pc) * (*pv);
35003c2a7987SHong Zhang       *pc        = multiplier;
3501b78a477dSHong Zhang       if (PetscAbsScalar(multiplier) > dt) {
3502b78a477dSHong Zhang         pj = bj + bdiag[row+1] + 1;         /* point to 1st entry of U(row,:) */
3503b78a477dSHong Zhang         pv = b->a + bdiag[row+1] + 1;
3504b78a477dSHong Zhang         nz = bdiag[row] - bdiag[row+1] - 1;         /* num of entries in U(row,:), excluding diagonal */
3505b78a477dSHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
3506ca0c957dSBarry Smith         ierr = PetscLogFlops(1+2.0*nz);CHKERRQ(ierr);
35073c2a7987SHong Zhang       }
3508b78a477dSHong Zhang       k++;
35093c2a7987SHong Zhang     }
35103c2a7987SHong Zhang 
3511b78a477dSHong Zhang     /* finished row so stick it into b->a */
3512b78a477dSHong Zhang     /* L-part */
3513b78a477dSHong Zhang     pv  = b->a + bi[i];
3514b78a477dSHong Zhang     pj  = bj + bi[i];
3515b78a477dSHong Zhang     nzl = bi[i+1] - bi[i];
3516b78a477dSHong Zhang     for (j=0; j<nzl; j++) {
3517b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
35183c2a7987SHong Zhang     }
3519b78a477dSHong Zhang 
3520b78a477dSHong Zhang     /* diagonal: invert diagonal entries for simplier triangular solves */
3521b78a477dSHong Zhang     if (rtmp[i] == 0.0) rtmp[i] = dt+shift;
3522b78a477dSHong Zhang     b->a[bdiag[i]] = 1.0/rtmp[i];
3523b78a477dSHong Zhang 
3524b78a477dSHong Zhang     /* U-part */
3525b78a477dSHong Zhang     pv  = b->a + bdiag[i+1] + 1;
3526b78a477dSHong Zhang     pj  = bj + bdiag[i+1] + 1;
3527b78a477dSHong Zhang     nzu = bdiag[i] - bdiag[i+1] - 1;
3528b78a477dSHong Zhang     for (j=0; j<nzu; j++) {
3529b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
3530b78a477dSHong Zhang     }
3531b78a477dSHong Zhang   }
3532b78a477dSHong Zhang 
35333c2a7987SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
35343c2a7987SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
35353c2a7987SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3536b78a477dSHong Zhang 
35373c2a7987SHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
35383c2a7987SHong Zhang   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
35393c2a7987SHong Zhang   if (row_identity && col_identity) {
354035233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
35413c2a7987SHong Zhang   } else {
354235233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ;
35433c2a7987SHong Zhang   }
3544f4259b30SLisandro Dalcin   C->ops->solveadd          = NULL;
3545f4259b30SLisandro Dalcin   C->ops->solvetranspose    = NULL;
3546f4259b30SLisandro Dalcin   C->ops->solvetransposeadd = NULL;
3547f4259b30SLisandro Dalcin   C->ops->matsolve          = NULL;
35483c2a7987SHong Zhang   C->assembled              = PETSC_TRUE;
35493c2a7987SHong Zhang   C->preallocated           = PETSC_TRUE;
35502205254eSKarl Rupp 
35513c2a7987SHong Zhang   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
35523c2a7987SHong Zhang   PetscFunctionReturn(0);
35533c2a7987SHong Zhang }
3554