xref: /petsc/src/mat/impls/aij/seq/aijfact.c (revision aa91b3e714cd18505e9e9fc669c9f6d1eda5214a)
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)
99891bceaeSHong Zhang   if (A->hermitian && (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hermitian 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);
121b24902e0SBarry Smith   PetscFunctionReturn(0);
122b24902e0SBarry Smith }
123b24902e0SBarry Smith 
124ad04f41aSHong Zhang PetscErrorCode MatLUFactorSymbolic_SeqAIJ_inplace(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
125289bc588SBarry Smith {
126416022c9SBarry Smith   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
127289bc588SBarry Smith   IS                 isicol;
1286849ba73SBarry Smith   PetscErrorCode     ierr;
1295d0c19d7SBarry Smith   const PetscInt     *r,*ic;
1305d0c19d7SBarry Smith   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j;
1311393bc97SHong Zhang   PetscInt           *bi,*bj,*ajtmp;
1321393bc97SHong Zhang   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
1339e046878SBarry Smith   PetscReal          f;
1344875ba38SHong Zhang   PetscInt           nlnk,*lnk,k,**bi_ptr;
1350298fd71SBarry Smith   PetscFreeSpaceList free_space=NULL,current_space=NULL;
1361393bc97SHong Zhang   PetscBT            lnkbt;
137ece542f9SHong Zhang   PetscBool          missing;
138289bc588SBarry Smith 
1393a40ed3dSBarry Smith   PetscFunctionBegin;
140e32f2f54SBarry Smith   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square");
141ece542f9SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
142ece542f9SHong Zhang   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
143ece542f9SHong Zhang 
1444c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
145ac121b3dSBarry Smith   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
146ac121b3dSBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
147289bc588SBarry Smith 
148289bc588SBarry Smith   /* get new row pointers */
149854ce69bSBarry Smith   ierr  = PetscMalloc1(n+1,&bi);CHKERRQ(ierr);
1501393bc97SHong Zhang   bi[0] = 0;
1511393bc97SHong Zhang 
1521393bc97SHong Zhang   /* bdiag is location of diagonal in factor */
153854ce69bSBarry Smith   ierr     = PetscMalloc1(n+1,&bdiag);CHKERRQ(ierr);
1541393bc97SHong Zhang   bdiag[0] = 0;
1551393bc97SHong Zhang 
1561393bc97SHong Zhang   /* linked list for storing column indices of the active row */
1571393bc97SHong Zhang   nlnk = n + 1;
1581393bc97SHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
1591393bc97SHong Zhang 
160dcca6d9dSJed Brown   ierr = PetscMalloc2(n+1,&bi_ptr,n+1,&im);CHKERRQ(ierr);
1611393bc97SHong Zhang 
1621393bc97SHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
163d3d32019SBarry Smith   f             = info->fill;
164*aa91b3e7SRichard Tran Mills   if (n==1)   f = 1; /* prevent failure in corner case of 1x1 matrix with fill < 0.5 */
1652f18eb33SBarry Smith   ierr          = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space);CHKERRQ(ierr);
1661393bc97SHong Zhang   current_space = free_space;
167289bc588SBarry Smith 
168289bc588SBarry Smith   for (i=0; i<n; i++) {
1691393bc97SHong Zhang     /* copy previous fill into linked list */
170289bc588SBarry Smith     nzi = 0;
1711393bc97SHong Zhang     nnz = ai[r[i]+1] - ai[r[i]];
1721393bc97SHong Zhang     ajtmp = aj + ai[r[i]];
173afcc9904SHong Zhang     ierr  = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
1741393bc97SHong Zhang     nzi  += nlnk;
1751393bc97SHong Zhang 
1761393bc97SHong Zhang     /* add pivot rows into linked list */
1771393bc97SHong Zhang     row = lnk[n];
1781393bc97SHong Zhang     while (row < i) {
179d1170560SHong Zhang       nzbd  = bdiag[row] - bi[row] + 1;   /* num of entries in the row with column index <= row */
180d1170560SHong Zhang       ajtmp = bi_ptr[row] + nzbd;   /* points to the entry next to the diagonal */
181d1170560SHong Zhang       ierr  = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
1821393bc97SHong Zhang       nzi  += nlnk;
1831393bc97SHong Zhang       row   = lnk[row];
184289bc588SBarry Smith     }
1851393bc97SHong Zhang     bi[i+1] = bi[i] + nzi;
1861393bc97SHong Zhang     im[i]   = nzi;
1871393bc97SHong Zhang 
1881393bc97SHong Zhang     /* mark bdiag */
1891393bc97SHong Zhang     nzbd = 0;
1901393bc97SHong Zhang     nnz  = nzi;
1911393bc97SHong Zhang     k    = lnk[n];
1921393bc97SHong Zhang     while (nnz-- && k < i) {
1931393bc97SHong Zhang       nzbd++;
1941393bc97SHong Zhang       k = lnk[k];
1951393bc97SHong Zhang     }
1961393bc97SHong Zhang     bdiag[i] = bi[i] + nzbd;
1971393bc97SHong Zhang 
1981393bc97SHong Zhang     /* if free space is not available, make more free space */
1991393bc97SHong Zhang     if (current_space->local_remaining<nzi) {
200f91af8c7SBarry Smith       nnz  = PetscIntMultTruncate(n - i,nzi); /* estimated and max additional space needed */
201a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
2021393bc97SHong Zhang       reallocs++;
2031393bc97SHong Zhang     }
2041393bc97SHong Zhang 
2051393bc97SHong Zhang     /* copy data into free space, then initialize lnk */
2061393bc97SHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
2072205254eSKarl Rupp 
2081393bc97SHong Zhang     bi_ptr[i]                       = current_space->array;
2091393bc97SHong Zhang     current_space->array           += nzi;
2101393bc97SHong Zhang     current_space->local_used      += nzi;
2111393bc97SHong Zhang     current_space->local_remaining -= nzi;
212289bc588SBarry Smith   }
2136cf91177SBarry Smith #if defined(PETSC_USE_INFO)
214464e8d44SSatish Balay   if (ai[n] != 0) {
2151393bc97SHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
21657622a8eSBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)f,(double)af);CHKERRQ(ierr);
21757622a8eSBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
21857622a8eSBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g);\n",(double)af);CHKERRQ(ierr);
219ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
22005bf559cSSatish Balay   } else {
221ae15b995SBarry Smith     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
22205bf559cSSatish Balay   }
22363ba0a88SBarry Smith #endif
2242fb3ed76SBarry Smith 
225898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
226898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
2271987afe7SBarry Smith 
2281393bc97SHong Zhang   /* destroy list of free space and other temporary array(s) */
229854ce69bSBarry Smith   ierr = PetscMalloc1(bi[n]+1,&bj);CHKERRQ(ierr);
230a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr);
2311393bc97SHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
23235baf27eSSatish Balay   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
233289bc588SBarry Smith 
234289bc588SBarry Smith   /* put together the new matrix */
2350298fd71SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
2363bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)B,(PetscObject)isicol);CHKERRQ(ierr);
237719d5645SBarry Smith   b    = (Mat_SeqAIJ*)(B)->data;
2382205254eSKarl Rupp 
239e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
240e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
2417c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
2422205254eSKarl Rupp 
243854ce69bSBarry Smith   ierr    = PetscMalloc1(bi[n]+1,&b->a);CHKERRQ(ierr);
2441393bc97SHong Zhang   b->j    = bj;
2451393bc97SHong Zhang   b->i    = bi;
2461393bc97SHong Zhang   b->diag = bdiag;
247416022c9SBarry Smith   b->ilen = 0;
248416022c9SBarry Smith   b->imax = 0;
249416022c9SBarry Smith   b->row  = isrow;
250416022c9SBarry Smith   b->col  = iscol;
251c38d4ed2SBarry Smith   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
252c38d4ed2SBarry Smith   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
25382bf6240SBarry Smith   b->icol = isicol;
254854ce69bSBarry Smith   ierr    = PetscMalloc1(n+1,&b->solve_work);CHKERRQ(ierr);
2551393bc97SHong Zhang 
2561393bc97SHong Zhang   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
2573bb1ff40SBarry Smith   ierr     = PetscLogObjectMemory((PetscObject)B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
2581393bc97SHong Zhang   b->maxnz = b->nz = bi[n];
2597fd98bd6SLois Curfman McInnes 
260d5f3da31SBarry Smith   (B)->factortype            = MAT_FACTOR_LU;
261719d5645SBarry Smith   (B)->info.factor_mallocs   = reallocs;
262719d5645SBarry Smith   (B)->info.fill_ratio_given = f;
263703deb49SSatish Balay 
2649f7d0b68SHong Zhang   if (ai[n]) {
265719d5645SBarry Smith     (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
266eea2913aSSatish Balay   } else {
267719d5645SBarry Smith     (B)->info.fill_ratio_needed = 0.0;
268eea2913aSSatish Balay   }
269ad04f41aSHong Zhang   (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace;
270f2344125SBarry Smith   if (a->inode.size) {
271f2344125SBarry Smith     (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
272f2344125SBarry Smith   }
2733a40ed3dSBarry Smith   PetscFunctionReturn(0);
274289bc588SBarry Smith }
2751393bc97SHong Zhang 
27635233d99SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
277ce3d78c0SShri Abhyankar {
278ce3d78c0SShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
279ce3d78c0SShri Abhyankar   IS                 isicol;
280ce3d78c0SShri Abhyankar   PetscErrorCode     ierr;
2818758e1faSBarry Smith   const PetscInt     *r,*ic,*ai=a->i,*aj=a->j,*ajtmp;
2828758e1faSBarry Smith   PetscInt           i,n=A->rmap->n;
2838758e1faSBarry Smith   PetscInt           *bi,*bj;
284ce3d78c0SShri Abhyankar   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
285ce3d78c0SShri Abhyankar   PetscReal          f;
286ce3d78c0SShri Abhyankar   PetscInt           nlnk,*lnk,k,**bi_ptr;
2870298fd71SBarry Smith   PetscFreeSpaceList free_space=NULL,current_space=NULL;
288ce3d78c0SShri Abhyankar   PetscBT            lnkbt;
289ece542f9SHong Zhang   PetscBool          missing;
290ce3d78c0SShri Abhyankar 
291ce3d78c0SShri Abhyankar   PetscFunctionBegin;
292e32f2f54SBarry Smith   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square");
293ece542f9SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
294ece542f9SHong Zhang   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
295ece542f9SHong Zhang 
296ce3d78c0SShri Abhyankar   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
297ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
298ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
299ce3d78c0SShri Abhyankar 
3001bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
301854ce69bSBarry Smith   ierr  = PetscMalloc1(n+1,&bi);CHKERRQ(ierr);
302854ce69bSBarry Smith   ierr  = PetscMalloc1(n+1,&bdiag);CHKERRQ(ierr);
303a6df321fSShri Abhyankar   bi[0] = bdiag[0] = 0;
3049b48462bSShri Abhyankar 
3059b48462bSShri Abhyankar   /* linked list for storing column indices of the active row */
3069b48462bSShri Abhyankar   nlnk = n + 1;
3079b48462bSShri Abhyankar   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3089b48462bSShri Abhyankar 
309dcca6d9dSJed Brown   ierr = PetscMalloc2(n+1,&bi_ptr,n+1,&im);CHKERRQ(ierr);
3109b48462bSShri Abhyankar 
3119b48462bSShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
3129b48462bSShri Abhyankar   f             = info->fill;
313*aa91b3e7SRichard Tran Mills   if (n==1)   f = 1; /* prevent failure in corner case of 1x1 matrix with fill < 0.5 */
3142f18eb33SBarry Smith   ierr          = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space);CHKERRQ(ierr);
3159b48462bSShri Abhyankar   current_space = free_space;
3169b48462bSShri Abhyankar 
3179b48462bSShri Abhyankar   for (i=0; i<n; i++) {
3189b48462bSShri Abhyankar     /* copy previous fill into linked list */
3199b48462bSShri Abhyankar     nzi = 0;
3209b48462bSShri Abhyankar     nnz = ai[r[i]+1] - ai[r[i]];
3219b48462bSShri Abhyankar     ajtmp = aj + ai[r[i]];
3229b48462bSShri Abhyankar     ierr  = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3239b48462bSShri Abhyankar     nzi  += nlnk;
3249b48462bSShri Abhyankar 
3259b48462bSShri Abhyankar     /* add pivot rows into linked list */
3269b48462bSShri Abhyankar     row = lnk[n];
3279b48462bSShri Abhyankar     while (row < i) {
3289b48462bSShri Abhyankar       nzbd  = bdiag[row] + 1; /* num of entries in the row with column index <= row */
3299b48462bSShri Abhyankar       ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
3309b48462bSShri Abhyankar       ierr  = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
3319b48462bSShri Abhyankar       nzi  += nlnk;
3329b48462bSShri Abhyankar       row   = lnk[row];
3339b48462bSShri Abhyankar     }
3349b48462bSShri Abhyankar     bi[i+1] = bi[i] + nzi;
3359b48462bSShri Abhyankar     im[i]   = nzi;
3369b48462bSShri Abhyankar 
3379b48462bSShri Abhyankar     /* mark bdiag */
3389b48462bSShri Abhyankar     nzbd = 0;
3399b48462bSShri Abhyankar     nnz  = nzi;
3409b48462bSShri Abhyankar     k    = lnk[n];
3419b48462bSShri Abhyankar     while (nnz-- && k < i) {
3429b48462bSShri Abhyankar       nzbd++;
3439b48462bSShri Abhyankar       k = lnk[k];
3449b48462bSShri Abhyankar     }
3459b48462bSShri Abhyankar     bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */
3469b48462bSShri Abhyankar 
3479b48462bSShri Abhyankar     /* if free space is not available, make more free space */
3489b48462bSShri Abhyankar     if (current_space->local_remaining<nzi) {
3492f18eb33SBarry Smith       /* estimated additional space needed */
3501df4278eSBarry Smith       nnz  = PetscIntMultTruncate(2,PetscIntMultTruncate(n-1,nzi));
3519b48462bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
3529b48462bSShri Abhyankar       reallocs++;
3539b48462bSShri Abhyankar     }
3549b48462bSShri Abhyankar 
3559b48462bSShri Abhyankar     /* copy data into free space, then initialize lnk */
3569b48462bSShri Abhyankar     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
3572205254eSKarl Rupp 
3589b48462bSShri Abhyankar     bi_ptr[i]                       = current_space->array;
3599b48462bSShri Abhyankar     current_space->array           += nzi;
3609b48462bSShri Abhyankar     current_space->local_used      += nzi;
3619b48462bSShri Abhyankar     current_space->local_remaining -= nzi;
3629b48462bSShri Abhyankar   }
3639b48462bSShri Abhyankar 
3649b48462bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3659b48462bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3669b48462bSShri Abhyankar 
3679263d837SHong Zhang   /*   copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */
368854ce69bSBarry Smith   ierr = PetscMalloc1(bi[n]+1,&bj);CHKERRQ(ierr);
3692ce24eb6SHong Zhang   ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
3709b48462bSShri Abhyankar   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
3719b48462bSShri Abhyankar   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
3729b48462bSShri Abhyankar 
3739b48462bSShri Abhyankar   /* put together the new matrix */
3740298fd71SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
3753bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)B,(PetscObject)isicol);CHKERRQ(ierr);
3769b48462bSShri Abhyankar   b    = (Mat_SeqAIJ*)(B)->data;
3772205254eSKarl Rupp 
3789b48462bSShri Abhyankar   b->free_a       = PETSC_TRUE;
3799b48462bSShri Abhyankar   b->free_ij      = PETSC_TRUE;
3809b48462bSShri Abhyankar   b->singlemalloc = PETSC_FALSE;
3812205254eSKarl Rupp 
382854ce69bSBarry Smith   ierr = PetscMalloc1(bdiag[0]+1,&b->a);CHKERRQ(ierr);
3832205254eSKarl Rupp 
3849b48462bSShri Abhyankar   b->j    = bj;
3859b48462bSShri Abhyankar   b->i    = bi;
3869b48462bSShri Abhyankar   b->diag = bdiag;
3879b48462bSShri Abhyankar   b->ilen = 0;
3889b48462bSShri Abhyankar   b->imax = 0;
3899b48462bSShri Abhyankar   b->row  = isrow;
3909b48462bSShri Abhyankar   b->col  = iscol;
3919b48462bSShri Abhyankar   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
3929b48462bSShri Abhyankar   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
3939b48462bSShri Abhyankar   b->icol = isicol;
394854ce69bSBarry Smith   ierr    = PetscMalloc1(n+1,&b->solve_work);CHKERRQ(ierr);
3959b48462bSShri Abhyankar 
3969b48462bSShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
3973bb1ff40SBarry Smith   ierr     = PetscLogObjectMemory((PetscObject)B,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
3989b48462bSShri Abhyankar   b->maxnz = b->nz = bdiag[0]+1;
3992205254eSKarl Rupp 
400d5f3da31SBarry Smith   B->factortype            = MAT_FACTOR_LU;
401f284f12aSHong Zhang   B->info.factor_mallocs   = reallocs;
402f284f12aSHong Zhang   B->info.fill_ratio_given = f;
4039b48462bSShri Abhyankar 
4049b48462bSShri Abhyankar   if (ai[n]) {
405f284f12aSHong Zhang     B->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
4069b48462bSShri Abhyankar   } else {
407f284f12aSHong Zhang     B->info.fill_ratio_needed = 0.0;
4089b48462bSShri Abhyankar   }
4099263d837SHong Zhang #if defined(PETSC_USE_INFO)
4109263d837SHong Zhang   if (ai[n] != 0) {
4119263d837SHong Zhang     PetscReal af = B->info.fill_ratio_needed;
41257622a8eSBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)f,(double)af);CHKERRQ(ierr);
41357622a8eSBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
41457622a8eSBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g);\n",(double)af);CHKERRQ(ierr);
4159263d837SHong Zhang     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
4169263d837SHong Zhang   } else {
4179263d837SHong Zhang     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
4189263d837SHong Zhang   }
4199263d837SHong Zhang #endif
42035233d99SShri Abhyankar   B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ;
421f2344125SBarry Smith   if (a->inode.size) {
422d3ac4fa3SBarry Smith     B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode;
423d3ac4fa3SBarry Smith   }
424a02bda8eSBarry Smith   ierr = MatSeqAIJCheckInode_FactorLU(B);CHKERRQ(ierr);
4259b48462bSShri Abhyankar   PetscFunctionReturn(0);
4269b48462bSShri Abhyankar }
4279b48462bSShri Abhyankar 
4285b5bc046SBarry Smith /*
4295b5bc046SBarry Smith     Trouble in factorization, should we dump the original matrix?
4305b5bc046SBarry Smith */
4315b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A)
4325b5bc046SBarry Smith {
4335b5bc046SBarry Smith   PetscErrorCode ierr;
434ace3abfcSBarry Smith   PetscBool      flg = PETSC_FALSE;
4355b5bc046SBarry Smith 
4365b5bc046SBarry Smith   PetscFunctionBegin;
437c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)A)->options,NULL,"-mat_factor_dump_on_error",&flg,NULL);CHKERRQ(ierr);
4385b5bc046SBarry Smith   if (flg) {
4395b5bc046SBarry Smith     PetscViewer viewer;
4405b5bc046SBarry Smith     char        filename[PETSC_MAX_PATH_LEN];
4415b5bc046SBarry Smith 
4425b5bc046SBarry Smith     ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr);
443ce94432eSBarry Smith     ierr = PetscViewerBinaryOpen(PetscObjectComm((PetscObject)A),filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr);
4445b5bc046SBarry Smith     ierr = MatView(A,viewer);CHKERRQ(ierr);
4456bf464f9SBarry Smith     ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr);
4465b5bc046SBarry Smith   }
4475b5bc046SBarry Smith   PetscFunctionReturn(0);
4485b5bc046SBarry Smith }
4495b5bc046SBarry Smith 
45035233d99SShri Abhyankar PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
451c9c72f8fSHong Zhang {
452c9c72f8fSHong Zhang   Mat             C     =B;
453c9c72f8fSHong Zhang   Mat_SeqAIJ      *a    =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data;
454c9c72f8fSHong Zhang   IS              isrow = b->row,isicol = b->icol;
455c9c72f8fSHong Zhang   PetscErrorCode  ierr;
456c9c72f8fSHong Zhang   const PetscInt  *r,*ic,*ics;
457bbd65245SShri Abhyankar   const PetscInt  n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bdiag=b->diag;
458bbd65245SShri Abhyankar   PetscInt        i,j,k,nz,nzL,row,*pj;
459bbd65245SShri Abhyankar   const PetscInt  *ajtmp,*bjtmp;
460bbd65245SShri Abhyankar   MatScalar       *rtmp,*pc,multiplier,*pv;
461bbd65245SShri Abhyankar   const MatScalar *aa=a->a,*v;
462ace3abfcSBarry Smith   PetscBool       row_identity,col_identity;
463d90ac83dSHong Zhang   FactorShiftCtx  sctx;
4644f81c4b7SBarry Smith   const PetscInt  *ddiag;
465d4ad06f7SHong Zhang   PetscReal       rs;
466d4ad06f7SHong Zhang   MatScalar       d;
467d4ad06f7SHong Zhang 
468c9c72f8fSHong Zhang   PetscFunctionBegin;
469d6e5152cSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
470d90ac83dSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
471d4ad06f7SHong Zhang 
472f4db908eSBarry Smith   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
473d4ad06f7SHong Zhang     ddiag          = a->diag;
474d4ad06f7SHong Zhang     sctx.shift_top = info->zeropivot;
475d4ad06f7SHong Zhang     for (i=0; i<n; i++) {
476d4ad06f7SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
477d4ad06f7SHong Zhang       d  = (aa)[ddiag[i]];
478d4ad06f7SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
479d4ad06f7SHong Zhang       v  = aa+ai[i];
480d4ad06f7SHong Zhang       nz = ai[i+1] - ai[i];
4812205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
482d4ad06f7SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
483d4ad06f7SHong Zhang     }
484d4ad06f7SHong Zhang     sctx.shift_top *= 1.1;
485d4ad06f7SHong Zhang     sctx.nshift_max = 5;
486d4ad06f7SHong Zhang     sctx.shift_lo   = 0.;
487d4ad06f7SHong Zhang     sctx.shift_hi   = 1.;
488d4ad06f7SHong Zhang   }
489d4ad06f7SHong Zhang 
490c9c72f8fSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
491c9c72f8fSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
492854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&rtmp);CHKERRQ(ierr);
493c9c72f8fSHong Zhang   ics  = ic;
494c9c72f8fSHong Zhang 
495d4ad06f7SHong Zhang   do {
49607b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
497c9c72f8fSHong Zhang     for (i=0; i<n; i++) {
498c9c72f8fSHong Zhang       /* zero rtmp */
499c9c72f8fSHong Zhang       /* L part */
500c9c72f8fSHong Zhang       nz    = bi[i+1] - bi[i];
501c9c72f8fSHong Zhang       bjtmp = bj + bi[i];
502c9c72f8fSHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
503c9c72f8fSHong Zhang 
504c9c72f8fSHong Zhang       /* U part */
50562fbd6afSShri Abhyankar       nz    = bdiag[i]-bdiag[i+1];
50662fbd6afSShri Abhyankar       bjtmp = bj + bdiag[i+1]+1;
507813a1f2bSShri Abhyankar       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
508813a1f2bSShri Abhyankar 
509813a1f2bSShri Abhyankar       /* load in initial (unfactored row) */
510813a1f2bSShri Abhyankar       nz    = ai[r[i]+1] - ai[r[i]];
511813a1f2bSShri Abhyankar       ajtmp = aj + ai[r[i]];
512813a1f2bSShri Abhyankar       v     = aa + ai[r[i]];
513813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
514813a1f2bSShri Abhyankar         rtmp[ics[ajtmp[j]]] = v[j];
515813a1f2bSShri Abhyankar       }
516813a1f2bSShri Abhyankar       /* ZeropivotApply() */
51798a93161SHong Zhang       rtmp[i] += sctx.shift_amount;  /* shift the diagonal of the matrix */
518813a1f2bSShri Abhyankar 
519813a1f2bSShri Abhyankar       /* elimination */
520813a1f2bSShri Abhyankar       bjtmp = bj + bi[i];
521813a1f2bSShri Abhyankar       row   = *bjtmp++;
522813a1f2bSShri Abhyankar       nzL   = bi[i+1] - bi[i];
523f268cba8SShri Abhyankar       for (k=0; k < nzL; k++) {
524813a1f2bSShri Abhyankar         pc = rtmp + row;
525813a1f2bSShri Abhyankar         if (*pc != 0.0) {
526813a1f2bSShri Abhyankar           pv         = b->a + bdiag[row];
527813a1f2bSShri Abhyankar           multiplier = *pc * (*pv);
528813a1f2bSShri Abhyankar           *pc        = multiplier;
5292205254eSKarl Rupp 
53062fbd6afSShri Abhyankar           pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */
53162fbd6afSShri Abhyankar           pv = b->a + bdiag[row+1]+1;
53262fbd6afSShri Abhyankar           nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
5332205254eSKarl Rupp 
534813a1f2bSShri Abhyankar           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
53528b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
536813a1f2bSShri Abhyankar         }
537f268cba8SShri Abhyankar         row = *bjtmp++;
538813a1f2bSShri Abhyankar       }
539813a1f2bSShri Abhyankar 
540813a1f2bSShri Abhyankar       /* finished row so stick it into b->a */
541813a1f2bSShri Abhyankar       rs = 0.0;
542813a1f2bSShri Abhyankar       /* L part */
543813a1f2bSShri Abhyankar       pv = b->a + bi[i];
544813a1f2bSShri Abhyankar       pj = b->j + bi[i];
545813a1f2bSShri Abhyankar       nz = bi[i+1] - bi[i];
546813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
547813a1f2bSShri Abhyankar         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
548813a1f2bSShri Abhyankar       }
549813a1f2bSShri Abhyankar 
550813a1f2bSShri Abhyankar       /* U part */
55162fbd6afSShri Abhyankar       pv = b->a + bdiag[i+1]+1;
55262fbd6afSShri Abhyankar       pj = b->j + bdiag[i+1]+1;
55362fbd6afSShri Abhyankar       nz = bdiag[i] - bdiag[i+1]-1;
554813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
555813a1f2bSShri Abhyankar         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
556813a1f2bSShri Abhyankar       }
557813a1f2bSShri Abhyankar 
558813a1f2bSShri Abhyankar       sctx.rs = rs;
559813a1f2bSShri Abhyankar       sctx.pv = rtmp[i];
5604cccfbddSHong Zhang       ierr    = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr);
56107b50cabSHong Zhang       if (sctx.newshift) break; /* break for-loop */
56207b50cabSHong Zhang       rtmp[i] = sctx.pv; /* sctx.pv might be updated in the case of MAT_SHIFT_INBLOCKS */
563813a1f2bSShri Abhyankar 
564813a1f2bSShri Abhyankar       /* Mark diagonal and invert diagonal for simplier triangular solves */
565813a1f2bSShri Abhyankar       pv  = b->a + bdiag[i];
566813a1f2bSShri Abhyankar       *pv = 1.0/rtmp[i];
567813a1f2bSShri Abhyankar 
568813a1f2bSShri Abhyankar     } /* endof for (i=0; i<n; i++) { */
569813a1f2bSShri Abhyankar 
5708ff23777SHong Zhang     /* MatPivotRefine() */
57107b50cabSHong Zhang     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
572813a1f2bSShri Abhyankar       /*
573813a1f2bSShri Abhyankar        * if no shift in this attempt & shifting & started shifting & can refine,
574813a1f2bSShri Abhyankar        * then try lower shift
575813a1f2bSShri Abhyankar        */
576813a1f2bSShri Abhyankar       sctx.shift_hi       = sctx.shift_fraction;
577813a1f2bSShri Abhyankar       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
578813a1f2bSShri Abhyankar       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
57907b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
580813a1f2bSShri Abhyankar       sctx.nshift++;
581813a1f2bSShri Abhyankar     }
58207b50cabSHong Zhang   } while (sctx.newshift);
583813a1f2bSShri Abhyankar 
584813a1f2bSShri Abhyankar   ierr = PetscFree(rtmp);CHKERRQ(ierr);
585813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
586813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
587d3ac4fa3SBarry Smith 
588813a1f2bSShri Abhyankar   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
589813a1f2bSShri Abhyankar   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
590abb87a52SBarry Smith   if (b->inode.size) {
591abb87a52SBarry Smith     C->ops->solve = MatSolve_SeqAIJ_Inode;
592abb87a52SBarry Smith   } else if (row_identity && col_identity) {
59335233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
594813a1f2bSShri Abhyankar   } else {
59535233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ;
596813a1f2bSShri Abhyankar   }
59735233d99SShri Abhyankar   C->ops->solveadd          = MatSolveAdd_SeqAIJ;
59835233d99SShri Abhyankar   C->ops->solvetranspose    = MatSolveTranspose_SeqAIJ;
59935233d99SShri Abhyankar   C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ;
60035233d99SShri Abhyankar   C->ops->matsolve          = MatMatSolve_SeqAIJ;
601813a1f2bSShri Abhyankar   C->assembled              = PETSC_TRUE;
602813a1f2bSShri Abhyankar   C->preallocated           = PETSC_TRUE;
6032205254eSKarl Rupp 
604813a1f2bSShri Abhyankar   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
605813a1f2bSShri Abhyankar 
60614d2772eSHong Zhang   /* MatShiftView(A,info,&sctx) */
607813a1f2bSShri Abhyankar   if (sctx.nshift) {
608f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
60957622a8eSBarry 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);
610f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
61157622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
612f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS) {
61357622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %g\n",sctx.nshift,(double)info->shiftamount);CHKERRQ(ierr);
614813a1f2bSShri Abhyankar     }
615813a1f2bSShri Abhyankar   }
616813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
617813a1f2bSShri Abhyankar }
618813a1f2bSShri Abhyankar 
619ad04f41aSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info)
620289bc588SBarry Smith {
621719d5645SBarry Smith   Mat             C     =B;
622416022c9SBarry Smith   Mat_SeqAIJ      *a    =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data;
62382bf6240SBarry Smith   IS              isrow = b->row,isicol = b->icol;
6246849ba73SBarry Smith   PetscErrorCode  ierr;
6255d0c19d7SBarry Smith   const PetscInt  *r,*ic,*ics;
626d278edc7SBarry Smith   PetscInt        nz,row,i,j,n=A->rmap->n,diag;
627d278edc7SBarry Smith   const PetscInt  *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
628d278edc7SBarry Smith   const PetscInt  *ajtmp,*bjtmp,*diag_offset = b->diag,*pj;
629d278edc7SBarry Smith   MatScalar       *pv,*rtmp,*pc,multiplier,d;
630d278edc7SBarry Smith   const MatScalar *v,*aa=a->a;
63133f9893dSHong Zhang   PetscReal       rs=0.0;
632d90ac83dSHong Zhang   FactorShiftCtx  sctx;
6334f81c4b7SBarry Smith   const PetscInt  *ddiag;
634ace3abfcSBarry Smith   PetscBool       row_identity, col_identity;
635289bc588SBarry Smith 
6363a40ed3dSBarry Smith   PetscFunctionBegin;
637504a3fadSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
638504a3fadSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
639ada7143aSSatish Balay 
640f4db908eSBarry Smith   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
64142898933SHong Zhang     ddiag          = a->diag;
6429f7d0b68SHong Zhang     sctx.shift_top = info->zeropivot;
6436cc28720Svictorle     for (i=0; i<n; i++) {
6449f95998fSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
6459f7d0b68SHong Zhang       d  = (aa)[ddiag[i]];
6461a890ab1SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
6479f7d0b68SHong Zhang       v  = aa+ai[i];
6489f7d0b68SHong Zhang       nz = ai[i+1] - ai[i];
6492205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
6501a890ab1SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
6516cc28720Svictorle     }
652118f5789SBarry Smith     sctx.shift_top *= 1.1;
653d2276718SHong Zhang     sctx.nshift_max = 5;
654d2276718SHong Zhang     sctx.shift_lo   = 0.;
655d2276718SHong Zhang     sctx.shift_hi   = 1.;
656a0a356efSHong Zhang   }
657a0a356efSHong Zhang 
658504a3fadSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
659504a3fadSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
660854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&rtmp);CHKERRQ(ierr);
661504a3fadSHong Zhang   ics  = ic;
662504a3fadSHong Zhang 
663085256b3SBarry Smith   do {
66407b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
665289bc588SBarry Smith     for (i=0; i<n; i++) {
666b3bf805bSHong Zhang       nz    = bi[i+1] - bi[i];
667b3bf805bSHong Zhang       bjtmp = bj + bi[i];
6689f7d0b68SHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
669289bc588SBarry Smith 
670289bc588SBarry Smith       /* load in initial (unfactored row) */
6719f7d0b68SHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
6729f7d0b68SHong Zhang       ajtmp = aj + ai[r[i]];
6739f7d0b68SHong Zhang       v     = aa + ai[r[i]];
674085256b3SBarry Smith       for (j=0; j<nz; j++) {
675b3bf805bSHong Zhang         rtmp[ics[ajtmp[j]]] = v[j];
676085256b3SBarry Smith       }
677d2276718SHong Zhang       rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */
678289bc588SBarry Smith 
679b3bf805bSHong Zhang       row = *bjtmp++;
680f2582111SSatish Balay       while  (row < i) {
6818c37ef55SBarry Smith         pc = rtmp + row;
6828c37ef55SBarry Smith         if (*pc != 0.0) {
683010a20caSHong Zhang           pv         = b->a + diag_offset[row];
684010a20caSHong Zhang           pj         = b->j + diag_offset[row] + 1;
68535aab85fSBarry Smith           multiplier = *pc / *pv++;
6868c37ef55SBarry Smith           *pc        = multiplier;
687b3bf805bSHong Zhang           nz         = bi[row+1] - diag_offset[row] - 1;
6889f7d0b68SHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
68928b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
690289bc588SBarry Smith         }
691b3bf805bSHong Zhang         row = *bjtmp++;
692289bc588SBarry Smith       }
693416022c9SBarry Smith       /* finished row so stick it into b->a */
694b3bf805bSHong Zhang       pv   = b->a + bi[i];
695b3bf805bSHong Zhang       pj   = b->j + bi[i];
696b3bf805bSHong Zhang       nz   = bi[i+1] - bi[i];
697b3bf805bSHong Zhang       diag = diag_offset[i] - bi[i];
698e57ff13aSBarry Smith       rs   = 0.0;
699d3d32019SBarry Smith       for (j=0; j<nz; j++) {
7009f7d0b68SHong Zhang         pv[j] = rtmp[pj[j]];
7019f7d0b68SHong Zhang         rs   += PetscAbsScalar(pv[j]);
702d3d32019SBarry Smith       }
703e57ff13aSBarry Smith       rs -= PetscAbsScalar(pv[diag]);
704d2276718SHong Zhang 
705d2276718SHong Zhang       sctx.rs = rs;
706d2276718SHong Zhang       sctx.pv = pv[diag];
7074cccfbddSHong Zhang       ierr    = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr);
70807b50cabSHong Zhang       if (sctx.newshift) break;
709504a3fadSHong Zhang       pv[diag] = sctx.pv;
71035aab85fSBarry Smith     }
711d2276718SHong Zhang 
71207b50cabSHong Zhang     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
7136cc28720Svictorle       /*
7146c037d1bSvictorle        * if no shift in this attempt & shifting & started shifting & can refine,
7156cc28720Svictorle        * then try lower shift
7166cc28720Svictorle        */
7170481f469SBarry Smith       sctx.shift_hi       = sctx.shift_fraction;
7180481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
7190481f469SBarry Smith       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
72007b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
721d2276718SHong Zhang       sctx.nshift++;
7226cc28720Svictorle     }
72307b50cabSHong Zhang   } while (sctx.newshift);
7240f11f9aeSSatish Balay 
72535aab85fSBarry Smith   /* invert diagonal entries for simplier triangular solves */
72635aab85fSBarry Smith   for (i=0; i<n; i++) {
727010a20caSHong Zhang     b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]];
72835aab85fSBarry Smith   }
729606d414cSSatish Balay   ierr = PetscFree(rtmp);CHKERRQ(ierr);
73078b31e54SBarry Smith   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
73178b31e54SBarry Smith   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
732d3ac4fa3SBarry Smith 
7338d8c7c43SBarry Smith   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
7348d8c7c43SBarry Smith   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
7358d8c7c43SBarry Smith   if (row_identity && col_identity) {
736ad04f41aSHong Zhang     C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_inplace;
7378d8c7c43SBarry Smith   } else {
738ad04f41aSHong Zhang     C->ops->solve = MatSolve_SeqAIJ_inplace;
739db4efbfdSBarry Smith   }
740ad04f41aSHong Zhang   C->ops->solveadd          = MatSolveAdd_SeqAIJ_inplace;
741ad04f41aSHong Zhang   C->ops->solvetranspose    = MatSolveTranspose_SeqAIJ_inplace;
742ad04f41aSHong Zhang   C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace;
743ad04f41aSHong Zhang   C->ops->matsolve          = MatMatSolve_SeqAIJ_inplace;
7442205254eSKarl Rupp 
745c456f294SBarry Smith   C->assembled    = PETSC_TRUE;
7465c9eb25fSBarry Smith   C->preallocated = PETSC_TRUE;
7472205254eSKarl Rupp 
748d0f46423SBarry Smith   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
749d2276718SHong Zhang   if (sctx.nshift) {
750f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
75157622a8eSBarry 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);
752f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
75357622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
7546cc28720Svictorle     }
7550697b6caSHong Zhang   }
756d3ac4fa3SBarry Smith   (C)->ops->solve          = MatSolve_SeqAIJ_inplace;
757d3ac4fa3SBarry Smith   (C)->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace;
7582205254eSKarl Rupp 
759a02bda8eSBarry Smith   ierr = MatSeqAIJCheckInode(C);CHKERRQ(ierr);
7603a40ed3dSBarry Smith   PetscFunctionReturn(0);
761289bc588SBarry Smith }
7620889b5dcSvictorle 
763137fb511SHong Zhang /*
764137fb511SHong Zhang    This routine implements inplace ILU(0) with row or/and column permutations.
765137fb511SHong Zhang    Input:
766137fb511SHong Zhang      A - original matrix
767137fb511SHong Zhang    Output;
768137fb511SHong Zhang      A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i]
769137fb511SHong Zhang          a->j (col index) is permuted by the inverse of colperm, then sorted
770137fb511SHong Zhang          a->a reordered accordingly with a->j
771137fb511SHong Zhang          a->diag (ptr to diagonal elements) is updated.
772137fb511SHong Zhang */
7730481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info)
774137fb511SHong Zhang {
775b051339dSHong Zhang   Mat_SeqAIJ      *a    =(Mat_SeqAIJ*)A->data;
776b051339dSHong Zhang   IS              isrow = a->row,isicol = a->icol;
777137fb511SHong Zhang   PetscErrorCode  ierr;
7785d0c19d7SBarry Smith   const PetscInt  *r,*ic,*ics;
7795d0c19d7SBarry Smith   PetscInt        i,j,n=A->rmap->n,*ai=a->i,*aj=a->j;
7805d0c19d7SBarry Smith   PetscInt        *ajtmp,nz,row;
781b051339dSHong Zhang   PetscInt        *diag = a->diag,nbdiag,*pj;
782a77337e4SBarry Smith   PetscScalar     *rtmp,*pc,multiplier,d;
783504a3fadSHong Zhang   MatScalar       *pv,*v;
784137fb511SHong Zhang   PetscReal       rs;
785d90ac83dSHong Zhang   FactorShiftCtx  sctx;
786504a3fadSHong Zhang   const MatScalar *aa=a->a,*vtmp;
787137fb511SHong Zhang 
788137fb511SHong Zhang   PetscFunctionBegin;
789e32f2f54SBarry Smith   if (A != B) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address");
790504a3fadSHong Zhang 
791504a3fadSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
792504a3fadSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
793504a3fadSHong Zhang 
794504a3fadSHong Zhang   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
795504a3fadSHong Zhang     const PetscInt *ddiag = a->diag;
796504a3fadSHong Zhang     sctx.shift_top = info->zeropivot;
797504a3fadSHong Zhang     for (i=0; i<n; i++) {
798504a3fadSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
799504a3fadSHong Zhang       d    = (aa)[ddiag[i]];
800504a3fadSHong Zhang       rs   = -PetscAbsScalar(d) - PetscRealPart(d);
801504a3fadSHong Zhang       vtmp = aa+ai[i];
802504a3fadSHong Zhang       nz   = ai[i+1] - ai[i];
8032205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(vtmp[j]);
804504a3fadSHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
805504a3fadSHong Zhang     }
806504a3fadSHong Zhang     sctx.shift_top *= 1.1;
807504a3fadSHong Zhang     sctx.nshift_max = 5;
808504a3fadSHong Zhang     sctx.shift_lo   = 0.;
809504a3fadSHong Zhang     sctx.shift_hi   = 1.;
810504a3fadSHong Zhang   }
811504a3fadSHong Zhang 
812137fb511SHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
813137fb511SHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
814854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&rtmp);CHKERRQ(ierr);
815580bdb30SBarry Smith   ierr = PetscArrayzero(rtmp,n+1);CHKERRQ(ierr);
816b051339dSHong Zhang   ics  = ic;
817137fb511SHong Zhang 
818504a3fadSHong Zhang #if defined(MV)
81975567043SBarry Smith   sctx.shift_top      = 0.;
820e60cf9a0SBarry Smith   sctx.nshift_max     = 0;
82175567043SBarry Smith   sctx.shift_lo       = 0.;
82275567043SBarry Smith   sctx.shift_hi       = 0.;
82375567043SBarry Smith   sctx.shift_fraction = 0.;
824137fb511SHong Zhang 
825f4db908eSBarry Smith   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
82675567043SBarry Smith     sctx.shift_top = 0.;
827137fb511SHong Zhang     for (i=0; i<n; i++) {
828137fb511SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
829b051339dSHong Zhang       d  = (a->a)[diag[i]];
830137fb511SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
831b051339dSHong Zhang       v  = a->a+ai[i];
832b051339dSHong Zhang       nz = ai[i+1] - ai[i];
8332205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
834137fb511SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
835137fb511SHong Zhang     }
836137fb511SHong Zhang     if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot;
837137fb511SHong Zhang     sctx.shift_top *= 1.1;
838137fb511SHong Zhang     sctx.nshift_max = 5;
839137fb511SHong Zhang     sctx.shift_lo   = 0.;
840137fb511SHong Zhang     sctx.shift_hi   = 1.;
841137fb511SHong Zhang   }
842137fb511SHong Zhang 
84375567043SBarry Smith   sctx.shift_amount = 0.;
844137fb511SHong Zhang   sctx.nshift       = 0;
845504a3fadSHong Zhang #endif
846504a3fadSHong Zhang 
847137fb511SHong Zhang   do {
84807b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
849137fb511SHong Zhang     for (i=0; i<n; i++) {
850b051339dSHong Zhang       /* load in initial unfactored row */
851b051339dSHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
852b051339dSHong Zhang       ajtmp = aj + ai[r[i]];
853b051339dSHong Zhang       v     = a->a + ai[r[i]];
854137fb511SHong Zhang       /* sort permuted ajtmp and values v accordingly */
855b051339dSHong Zhang       for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]];
856137fb511SHong Zhang       ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr);
857137fb511SHong Zhang 
858b051339dSHong Zhang       diag[r[i]] = ai[r[i]];
859137fb511SHong Zhang       for (j=0; j<nz; j++) {
860137fb511SHong Zhang         rtmp[ajtmp[j]] = v[j];
861b051339dSHong Zhang         if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */
862137fb511SHong Zhang       }
863137fb511SHong Zhang       rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */
864137fb511SHong Zhang 
865b051339dSHong Zhang       row = *ajtmp++;
866137fb511SHong Zhang       while  (row < i) {
867137fb511SHong Zhang         pc = rtmp + row;
868137fb511SHong Zhang         if (*pc != 0.0) {
869b051339dSHong Zhang           pv = a->a + diag[r[row]];
870b051339dSHong Zhang           pj = aj + diag[r[row]] + 1;
871137fb511SHong Zhang 
872137fb511SHong Zhang           multiplier = *pc / *pv++;
873137fb511SHong Zhang           *pc        = multiplier;
874b051339dSHong Zhang           nz         = ai[r[row]+1] - diag[r[row]] - 1;
875b051339dSHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
87628b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
877137fb511SHong Zhang         }
878b051339dSHong Zhang         row = *ajtmp++;
879137fb511SHong Zhang       }
880b051339dSHong Zhang       /* finished row so overwrite it onto a->a */
881b051339dSHong Zhang       pv     = a->a + ai[r[i]];
882b051339dSHong Zhang       pj     = aj + ai[r[i]];
883b051339dSHong Zhang       nz     = ai[r[i]+1] - ai[r[i]];
884b051339dSHong Zhang       nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */
885137fb511SHong Zhang 
886137fb511SHong Zhang       rs = 0.0;
887137fb511SHong Zhang       for (j=0; j<nz; j++) {
888b051339dSHong Zhang         pv[j] = rtmp[pj[j]];
889b051339dSHong Zhang         if (j != nbdiag) rs += PetscAbsScalar(pv[j]);
890137fb511SHong Zhang       }
891137fb511SHong Zhang 
892137fb511SHong Zhang       sctx.rs = rs;
893b051339dSHong Zhang       sctx.pv = pv[nbdiag];
8944cccfbddSHong Zhang       ierr    = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr);
89507b50cabSHong Zhang       if (sctx.newshift) break;
896504a3fadSHong Zhang       pv[nbdiag] = sctx.pv;
897137fb511SHong Zhang     }
898137fb511SHong Zhang 
89907b50cabSHong Zhang     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
900137fb511SHong Zhang       /*
901137fb511SHong Zhang        * if no shift in this attempt & shifting & started shifting & can refine,
902137fb511SHong Zhang        * then try lower shift
903137fb511SHong Zhang        */
9040481f469SBarry Smith       sctx.shift_hi       = sctx.shift_fraction;
9050481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
9060481f469SBarry Smith       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
90707b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
908137fb511SHong Zhang       sctx.nshift++;
909137fb511SHong Zhang     }
91007b50cabSHong Zhang   } while (sctx.newshift);
911137fb511SHong Zhang 
912137fb511SHong Zhang   /* invert diagonal entries for simplier triangular solves */
913137fb511SHong Zhang   for (i=0; i<n; i++) {
914b051339dSHong Zhang     a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]];
915137fb511SHong Zhang   }
916137fb511SHong Zhang 
917137fb511SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
918137fb511SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
919137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
9202205254eSKarl Rupp 
921b051339dSHong Zhang   A->ops->solve             = MatSolve_SeqAIJ_InplaceWithPerm;
922ad04f41aSHong Zhang   A->ops->solveadd          = MatSolveAdd_SeqAIJ_inplace;
923ad04f41aSHong Zhang   A->ops->solvetranspose    = MatSolveTranspose_SeqAIJ_inplace;
924ad04f41aSHong Zhang   A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace;
9252205254eSKarl Rupp 
926b051339dSHong Zhang   A->assembled    = PETSC_TRUE;
9275c9eb25fSBarry Smith   A->preallocated = PETSC_TRUE;
9282205254eSKarl Rupp 
929d0f46423SBarry Smith   ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr);
930137fb511SHong Zhang   if (sctx.nshift) {
931f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
93257622a8eSBarry 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);
933f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
93457622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
935137fb511SHong Zhang     }
936137fb511SHong Zhang   }
937137fb511SHong Zhang   PetscFunctionReturn(0);
938137fb511SHong Zhang }
939137fb511SHong Zhang 
9400a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
9410481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info)
942da3a660dSBarry Smith {
943dfbe8321SBarry Smith   PetscErrorCode ierr;
944416022c9SBarry Smith   Mat            C;
945416022c9SBarry Smith 
9463a40ed3dSBarry Smith   PetscFunctionBegin;
9472692d6eeSBarry Smith   ierr = MatGetFactor(A,MATSOLVERPETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr);
948719d5645SBarry Smith   ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr);
949719d5645SBarry Smith   ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr);
9502205254eSKarl Rupp 
951db4efbfdSBarry Smith   A->ops->solve          = C->ops->solve;
952db4efbfdSBarry Smith   A->ops->solvetranspose = C->ops->solvetranspose;
9532205254eSKarl Rupp 
95428be2f97SBarry Smith   ierr = MatHeaderMerge(A,&C);CHKERRQ(ierr);
9553bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)A,(PetscObject)((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr);
9563a40ed3dSBarry Smith   PetscFunctionReturn(0);
957da3a660dSBarry Smith }
9580a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
9591d098868SHong Zhang 
960889a8dedSBarry Smith 
961ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_inplace(Mat A,Vec bb,Vec xx)
9628c37ef55SBarry Smith {
963416022c9SBarry Smith   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
964416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
9656849ba73SBarry Smith   PetscErrorCode    ierr;
9665d0c19d7SBarry Smith   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
9675d0c19d7SBarry Smith   PetscInt          nz;
9685d0c19d7SBarry Smith   const PetscInt    *rout,*cout,*r,*c;
969dd6ea824SBarry Smith   PetscScalar       *x,*tmp,*tmps,sum;
970d9fead3dSBarry Smith   const PetscScalar *b;
971dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
9728c37ef55SBarry Smith 
9733a40ed3dSBarry Smith   PetscFunctionBegin;
9743a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
97591bf9adeSBarry Smith 
9763649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
977fdc842d1SBarry Smith   ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr);
978416022c9SBarry Smith   tmp  = a->solve_work;
9798c37ef55SBarry Smith 
9803b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
9813b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
9828c37ef55SBarry Smith 
9838c37ef55SBarry Smith   /* forward solve the lower triangular */
9848c37ef55SBarry Smith   tmp[0] = b[*r++];
985010a20caSHong Zhang   tmps   = tmp;
9868c37ef55SBarry Smith   for (i=1; i<n; i++) {
987010a20caSHong Zhang     v   = aa + ai[i];
988010a20caSHong Zhang     vi  = aj + ai[i];
98953e7425aSBarry Smith     nz  = a->diag[i] - ai[i];
99053e7425aSBarry Smith     sum = b[*r++];
991003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
9928c37ef55SBarry Smith     tmp[i] = sum;
9938c37ef55SBarry Smith   }
9948c37ef55SBarry Smith 
9958c37ef55SBarry Smith   /* backward solve the upper triangular */
9968c37ef55SBarry Smith   for (i=n-1; i>=0; i--) {
997010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
998010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
999416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
10008c37ef55SBarry Smith     sum = tmp[i];
1001003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1002010a20caSHong Zhang     x[*c--] = tmp[i] = sum*aa[a->diag[i]];
10038c37ef55SBarry Smith   }
10048c37ef55SBarry Smith 
10053b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
10063b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
10073649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1008fdc842d1SBarry Smith   ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr);
1009dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
10103a40ed3dSBarry Smith   PetscFunctionReturn(0);
10118c37ef55SBarry Smith }
1012026e39d0SSatish Balay 
1013ad04f41aSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ_inplace(Mat A,Mat B,Mat X)
10142bebee5dSHong Zhang {
10152bebee5dSHong Zhang   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
10162bebee5dSHong Zhang   IS                iscol = a->col,isrow = a->row;
10172bebee5dSHong Zhang   PetscErrorCode    ierr;
10185d0c19d7SBarry Smith   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
10195d0c19d7SBarry Smith   PetscInt          nz,neq;
10205d0c19d7SBarry Smith   const PetscInt    *rout,*cout,*r,*c;
10211683a169SBarry Smith   PetscScalar       *x,*tmp,*tmps,sum;
10221683a169SBarry Smith   const PetscScalar *aa = a->a,*v;
10231683a169SBarry Smith   const PetscScalar *b;
1024ace3abfcSBarry Smith   PetscBool         bisdense,xisdense;
10252bebee5dSHong Zhang 
10262bebee5dSHong Zhang   PetscFunctionBegin;
10272bebee5dSHong Zhang   if (!n) PetscFunctionReturn(0);
10282bebee5dSHong Zhang 
1029251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr);
1030e32f2f54SBarry Smith   if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix");
1031251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr);
1032e32f2f54SBarry Smith   if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix");
1033db4efbfdSBarry Smith 
10341683a169SBarry Smith   ierr = MatDenseGetArrayRead(B,&b);CHKERRQ(ierr);
10358c778c55SBarry Smith   ierr = MatDenseGetArray(X,&x);CHKERRQ(ierr);
10362bebee5dSHong Zhang 
10372bebee5dSHong Zhang   tmp  = a->solve_work;
10382bebee5dSHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
10392bebee5dSHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
10402bebee5dSHong Zhang 
1041a36b22ccSSatish Balay   for (neq=0; neq<B->cmap->n; neq++) {
10422bebee5dSHong Zhang     /* forward solve the lower triangular */
10432bebee5dSHong Zhang     tmp[0] = b[r[0]];
10442bebee5dSHong Zhang     tmps   = tmp;
10452bebee5dSHong Zhang     for (i=1; i<n; i++) {
10462bebee5dSHong Zhang       v   = aa + ai[i];
10472bebee5dSHong Zhang       vi  = aj + ai[i];
10482bebee5dSHong Zhang       nz  = a->diag[i] - ai[i];
10492bebee5dSHong Zhang       sum = b[r[i]];
1050003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10512bebee5dSHong Zhang       tmp[i] = sum;
10522bebee5dSHong Zhang     }
10532bebee5dSHong Zhang     /* backward solve the upper triangular */
10542bebee5dSHong Zhang     for (i=n-1; i>=0; i--) {
10552bebee5dSHong Zhang       v   = aa + a->diag[i] + 1;
10562bebee5dSHong Zhang       vi  = aj + a->diag[i] + 1;
10572bebee5dSHong Zhang       nz  = ai[i+1] - a->diag[i] - 1;
10582bebee5dSHong Zhang       sum = tmp[i];
1059003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10602bebee5dSHong Zhang       x[c[i]] = tmp[i] = sum*aa[a->diag[i]];
10612bebee5dSHong Zhang     }
10622bebee5dSHong Zhang 
10632bebee5dSHong Zhang     b += n;
10642bebee5dSHong Zhang     x += n;
10652bebee5dSHong Zhang   }
10662bebee5dSHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
10672bebee5dSHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
10681683a169SBarry Smith   ierr = MatDenseRestoreArrayRead(B,&b);CHKERRQ(ierr);
10698c778c55SBarry Smith   ierr = MatDenseRestoreArray(X,&x);CHKERRQ(ierr);
1070dc0b31edSSatish Balay   ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr);
10712bebee5dSHong Zhang   PetscFunctionReturn(0);
10722bebee5dSHong Zhang }
10732bebee5dSHong Zhang 
107435233d99SShri Abhyankar PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X)
10759bd0847aSShri Abhyankar {
10769bd0847aSShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
10779bd0847aSShri Abhyankar   IS                iscol = a->col,isrow = a->row;
10789bd0847aSShri Abhyankar   PetscErrorCode    ierr;
10799bd0847aSShri Abhyankar   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag;
10809bd0847aSShri Abhyankar   PetscInt          nz,neq;
10819bd0847aSShri Abhyankar   const PetscInt    *rout,*cout,*r,*c;
10821683a169SBarry Smith   PetscScalar       *x,*tmp,sum;
10831683a169SBarry Smith   const PetscScalar *b;
10841683a169SBarry Smith   const PetscScalar *aa = a->a,*v;
1085ace3abfcSBarry Smith   PetscBool         bisdense,xisdense;
10869bd0847aSShri Abhyankar 
10879bd0847aSShri Abhyankar   PetscFunctionBegin;
10889bd0847aSShri Abhyankar   if (!n) PetscFunctionReturn(0);
10899bd0847aSShri Abhyankar 
1090251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr);
1091e32f2f54SBarry Smith   if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix");
1092251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr);
1093e32f2f54SBarry Smith   if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix");
10949bd0847aSShri Abhyankar 
10951683a169SBarry Smith   ierr = MatDenseGetArrayRead(B,&b);CHKERRQ(ierr);
10968c778c55SBarry Smith   ierr = MatDenseGetArray(X,&x);CHKERRQ(ierr);
10979bd0847aSShri Abhyankar 
10989bd0847aSShri Abhyankar   tmp  = a->solve_work;
10999bd0847aSShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
11009bd0847aSShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
11019bd0847aSShri Abhyankar 
11029bd0847aSShri Abhyankar   for (neq=0; neq<B->cmap->n; neq++) {
11039bd0847aSShri Abhyankar     /* forward solve the lower triangular */
11049bd0847aSShri Abhyankar     tmp[0] = b[r[0]];
11059bd0847aSShri Abhyankar     v      = aa;
11069bd0847aSShri Abhyankar     vi     = aj;
11079bd0847aSShri Abhyankar     for (i=1; i<n; i++) {
11089bd0847aSShri Abhyankar       nz  = ai[i+1] - ai[i];
11099bd0847aSShri Abhyankar       sum = b[r[i]];
11109bd0847aSShri Abhyankar       PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
11119bd0847aSShri Abhyankar       tmp[i] = sum;
11129bd0847aSShri Abhyankar       v     += nz; vi += nz;
11139bd0847aSShri Abhyankar     }
11149bd0847aSShri Abhyankar 
11159bd0847aSShri Abhyankar     /* backward solve the upper triangular */
11169bd0847aSShri Abhyankar     for (i=n-1; i>=0; i--) {
11179bd0847aSShri Abhyankar       v   = aa + adiag[i+1]+1;
11189bd0847aSShri Abhyankar       vi  = aj + adiag[i+1]+1;
11199bd0847aSShri Abhyankar       nz  = adiag[i]-adiag[i+1]-1;
11209bd0847aSShri Abhyankar       sum = tmp[i];
11219bd0847aSShri Abhyankar       PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
11229bd0847aSShri Abhyankar       x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */
11239bd0847aSShri Abhyankar     }
11249bd0847aSShri Abhyankar 
11259bd0847aSShri Abhyankar     b += n;
11269bd0847aSShri Abhyankar     x += n;
11279bd0847aSShri Abhyankar   }
11289bd0847aSShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
11299bd0847aSShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
11301683a169SBarry Smith   ierr = MatDenseRestoreArrayRead(B,&b);CHKERRQ(ierr);
11318c778c55SBarry Smith   ierr = MatDenseRestoreArray(X,&x);CHKERRQ(ierr);
11329bd0847aSShri Abhyankar   ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr);
11339bd0847aSShri Abhyankar   PetscFunctionReturn(0);
11349bd0847aSShri Abhyankar }
11359bd0847aSShri Abhyankar 
1136137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx)
1137137fb511SHong Zhang {
1138137fb511SHong Zhang   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
1139137fb511SHong Zhang   IS                iscol = a->col,isrow = a->row;
1140137fb511SHong Zhang   PetscErrorCode    ierr;
11415d0c19d7SBarry Smith   const PetscInt    *r,*c,*rout,*cout;
11425d0c19d7SBarry Smith   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
11435d0c19d7SBarry Smith   PetscInt          nz,row;
1144fdc842d1SBarry Smith   PetscScalar       *x,*tmp,*tmps,sum;
1145fdc842d1SBarry Smith   const PetscScalar *b;
1146dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
1147137fb511SHong Zhang 
1148137fb511SHong Zhang   PetscFunctionBegin;
1149137fb511SHong Zhang   if (!n) PetscFunctionReturn(0);
1150137fb511SHong Zhang 
1151fdc842d1SBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
1152fdc842d1SBarry Smith   ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr);
1153137fb511SHong Zhang   tmp  = a->solve_work;
1154137fb511SHong Zhang 
1155137fb511SHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1156137fb511SHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1157137fb511SHong Zhang 
1158137fb511SHong Zhang   /* forward solve the lower triangular */
1159137fb511SHong Zhang   tmp[0] = b[*r++];
1160137fb511SHong Zhang   tmps   = tmp;
1161137fb511SHong Zhang   for (row=1; row<n; row++) {
1162137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1163137fb511SHong Zhang     v   = aa + ai[i];
1164137fb511SHong Zhang     vi  = aj + ai[i];
1165137fb511SHong Zhang     nz  = a->diag[i] - ai[i];
1166137fb511SHong Zhang     sum = b[*r++];
1167003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1168137fb511SHong Zhang     tmp[row] = sum;
1169137fb511SHong Zhang   }
1170137fb511SHong Zhang 
1171137fb511SHong Zhang   /* backward solve the upper triangular */
1172137fb511SHong Zhang   for (row=n-1; row>=0; row--) {
1173137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1174137fb511SHong Zhang     v   = aa + a->diag[i] + 1;
1175137fb511SHong Zhang     vi  = aj + a->diag[i] + 1;
1176137fb511SHong Zhang     nz  = ai[i+1] - a->diag[i] - 1;
1177137fb511SHong Zhang     sum = tmp[row];
1178003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1179137fb511SHong Zhang     x[*c--] = tmp[row] = sum*aa[a->diag[i]];
1180137fb511SHong Zhang   }
1181137fb511SHong Zhang 
1182137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1183137fb511SHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1184fdc842d1SBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1185fdc842d1SBarry Smith   ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr);
1186dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
1187137fb511SHong Zhang   PetscFunctionReturn(0);
1188137fb511SHong Zhang }
1189137fb511SHong Zhang 
1190930ae53cSBarry Smith /* ----------------------------------------------------------- */
1191c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/ftn-kernels/fsolve.h>
1192ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_inplace(Mat A,Vec bb,Vec xx)
1193930ae53cSBarry Smith {
1194930ae53cSBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
11956849ba73SBarry Smith   PetscErrorCode    ierr;
1196d0f46423SBarry Smith   PetscInt          n   = A->rmap->n;
1197b377110cSBarry Smith   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag;
1198356650c2SBarry Smith   PetscScalar       *x;
1199356650c2SBarry Smith   const PetscScalar *b;
1200dd6ea824SBarry Smith   const MatScalar   *aa = a->a;
1201aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
1202356650c2SBarry Smith   PetscInt        adiag_i,i,nz,ai_i;
1203b377110cSBarry Smith   const PetscInt  *vi;
1204dd6ea824SBarry Smith   const MatScalar *v;
1205dd6ea824SBarry Smith   PetscScalar     sum;
1206d85afc42SSatish Balay #endif
1207930ae53cSBarry Smith 
12083a40ed3dSBarry Smith   PetscFunctionBegin;
12093a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
1210930ae53cSBarry Smith 
12113649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
1212fdc842d1SBarry Smith   ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr);
1213930ae53cSBarry Smith 
1214aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
12151c4feecaSSatish Balay   fortransolveaij_(&n,x,ai,aj,adiag,aa,b);
12161c4feecaSSatish Balay #else
1217930ae53cSBarry Smith   /* forward solve the lower triangular */
1218930ae53cSBarry Smith   x[0] = b[0];
1219930ae53cSBarry Smith   for (i=1; i<n; i++) {
1220930ae53cSBarry Smith     ai_i = ai[i];
1221930ae53cSBarry Smith     v    = aa + ai_i;
1222930ae53cSBarry Smith     vi   = aj + ai_i;
1223930ae53cSBarry Smith     nz   = adiag[i] - ai_i;
1224930ae53cSBarry Smith     sum  = b[i];
1225003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1226930ae53cSBarry Smith     x[i] = sum;
1227930ae53cSBarry Smith   }
1228930ae53cSBarry Smith 
1229930ae53cSBarry Smith   /* backward solve the upper triangular */
1230930ae53cSBarry Smith   for (i=n-1; i>=0; i--) {
1231930ae53cSBarry Smith     adiag_i = adiag[i];
1232d708749eSBarry Smith     v       = aa + adiag_i + 1;
1233d708749eSBarry Smith     vi      = aj + adiag_i + 1;
1234930ae53cSBarry Smith     nz      = ai[i+1] - adiag_i - 1;
1235930ae53cSBarry Smith     sum     = x[i];
1236003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1237930ae53cSBarry Smith     x[i] = sum*aa[adiag_i];
1238930ae53cSBarry Smith   }
12391c4feecaSSatish Balay #endif
1240dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
12413649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1242fdc842d1SBarry Smith   ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr);
12433a40ed3dSBarry Smith   PetscFunctionReturn(0);
1244930ae53cSBarry Smith }
1245930ae53cSBarry Smith 
1246ad04f41aSHong Zhang PetscErrorCode MatSolveAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec yy,Vec xx)
1247da3a660dSBarry Smith {
1248416022c9SBarry Smith   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
1249416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
12506849ba73SBarry Smith   PetscErrorCode    ierr;
125104fbf559SBarry Smith   PetscInt          i, n = A->rmap->n,j;
12525d0c19d7SBarry Smith   PetscInt          nz;
125304fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j;
125404fbf559SBarry Smith   PetscScalar       *x,*tmp,sum;
125504fbf559SBarry Smith   const PetscScalar *b;
1256dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
1257da3a660dSBarry Smith 
12583a40ed3dSBarry Smith   PetscFunctionBegin;
125978b31e54SBarry Smith   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
1260da3a660dSBarry Smith 
12613649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
12621ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1263416022c9SBarry Smith   tmp  = a->solve_work;
1264da3a660dSBarry Smith 
12653b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
12663b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1267da3a660dSBarry Smith 
1268da3a660dSBarry Smith   /* forward solve the lower triangular */
1269da3a660dSBarry Smith   tmp[0] = b[*r++];
1270da3a660dSBarry Smith   for (i=1; i<n; i++) {
1271010a20caSHong Zhang     v   = aa + ai[i];
1272010a20caSHong Zhang     vi  = aj + ai[i];
1273416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
1274da3a660dSBarry Smith     sum = b[*r++];
127504fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1276da3a660dSBarry Smith     tmp[i] = sum;
1277da3a660dSBarry Smith   }
1278da3a660dSBarry Smith 
1279da3a660dSBarry Smith   /* backward solve the upper triangular */
1280da3a660dSBarry Smith   for (i=n-1; i>=0; i--) {
1281010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
1282010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
1283416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
1284da3a660dSBarry Smith     sum = tmp[i];
128504fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1286010a20caSHong Zhang     tmp[i]   = sum*aa[a->diag[i]];
1287da3a660dSBarry Smith     x[*c--] += tmp[i];
1288da3a660dSBarry Smith   }
1289da3a660dSBarry Smith 
12903b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
12913b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
12923649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
12931ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1294dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
12953a40ed3dSBarry Smith   PetscFunctionReturn(0);
1296da3a660dSBarry Smith }
129704fbf559SBarry Smith 
129835233d99SShri Abhyankar PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx)
12993c0119dfSShri Abhyankar {
13003c0119dfSShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
13013c0119dfSShri Abhyankar   IS                iscol = a->col,isrow = a->row;
13023c0119dfSShri Abhyankar   PetscErrorCode    ierr;
13033c0119dfSShri Abhyankar   PetscInt          i, n = A->rmap->n,j;
13043c0119dfSShri Abhyankar   PetscInt          nz;
13053c0119dfSShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag;
13063c0119dfSShri Abhyankar   PetscScalar       *x,*tmp,sum;
13073c0119dfSShri Abhyankar   const PetscScalar *b;
13083c0119dfSShri Abhyankar   const MatScalar   *aa = a->a,*v;
13093c0119dfSShri Abhyankar 
13103c0119dfSShri Abhyankar   PetscFunctionBegin;
13113c0119dfSShri Abhyankar   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
13123c0119dfSShri Abhyankar 
13133649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
13143c0119dfSShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
13153c0119dfSShri Abhyankar   tmp  = a->solve_work;
13163c0119dfSShri Abhyankar 
13173c0119dfSShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
13183c0119dfSShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
13193c0119dfSShri Abhyankar 
13203c0119dfSShri Abhyankar   /* forward solve the lower triangular */
13213c0119dfSShri Abhyankar   tmp[0] = b[r[0]];
13223c0119dfSShri Abhyankar   v      = aa;
13233c0119dfSShri Abhyankar   vi     = aj;
13243c0119dfSShri Abhyankar   for (i=1; i<n; i++) {
13253c0119dfSShri Abhyankar     nz  = ai[i+1] - ai[i];
13263c0119dfSShri Abhyankar     sum = b[r[i]];
13273c0119dfSShri Abhyankar     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
13283c0119dfSShri Abhyankar     tmp[i] = sum;
13292205254eSKarl Rupp     v     += nz;
13302205254eSKarl Rupp     vi    += nz;
13313c0119dfSShri Abhyankar   }
13323c0119dfSShri Abhyankar 
13333c0119dfSShri Abhyankar   /* backward solve the upper triangular */
13343c0119dfSShri Abhyankar   v  = aa + adiag[n-1];
13353c0119dfSShri Abhyankar   vi = aj + adiag[n-1];
13363c0119dfSShri Abhyankar   for (i=n-1; i>=0; i--) {
13373c0119dfSShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
13383c0119dfSShri Abhyankar     sum = tmp[i];
13393c0119dfSShri Abhyankar     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
13403c0119dfSShri Abhyankar     tmp[i]   = sum*v[nz];
13413c0119dfSShri Abhyankar     x[c[i]] += tmp[i];
13423c0119dfSShri Abhyankar     v       += nz+1; vi += nz+1;
13433c0119dfSShri Abhyankar   }
13443c0119dfSShri Abhyankar 
13453c0119dfSShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
13463c0119dfSShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
13473649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
13483c0119dfSShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
13493c0119dfSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
13503c0119dfSShri Abhyankar   PetscFunctionReturn(0);
13513c0119dfSShri Abhyankar }
13523c0119dfSShri Abhyankar 
1353ad04f41aSHong Zhang PetscErrorCode MatSolveTranspose_SeqAIJ_inplace(Mat A,Vec bb,Vec xx)
1354da3a660dSBarry Smith {
1355416022c9SBarry Smith   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
13562235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
13576849ba73SBarry Smith   PetscErrorCode    ierr;
135804fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
135904fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
136004fbf559SBarry Smith   PetscInt          nz;
136104fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1362dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
136304fbf559SBarry Smith   const PetscScalar *b;
1364da3a660dSBarry Smith 
13653a40ed3dSBarry Smith   PetscFunctionBegin;
13663649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
1367fdc842d1SBarry Smith   ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr);
1368416022c9SBarry Smith   tmp  = a->solve_work;
1369da3a660dSBarry Smith 
13702235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
13712235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1372da3a660dSBarry Smith 
1373da3a660dSBarry Smith   /* copy the b into temp work space according to permutation */
13742235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1375da3a660dSBarry Smith 
1376da3a660dSBarry Smith   /* forward solve the U^T */
1377da3a660dSBarry Smith   for (i=0; i<n; i++) {
1378010a20caSHong Zhang     v   = aa + diag[i];
1379010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1380f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
1381c38d4ed2SBarry Smith     s1  = tmp[i];
1382ef66eb69SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
138304fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1384c38d4ed2SBarry Smith     tmp[i] = s1;
1385da3a660dSBarry Smith   }
1386da3a660dSBarry Smith 
1387da3a660dSBarry Smith   /* backward solve the L^T */
1388da3a660dSBarry Smith   for (i=n-1; i>=0; i--) {
1389010a20caSHong Zhang     v  = aa + diag[i] - 1;
1390010a20caSHong Zhang     vi = aj + diag[i] - 1;
1391f1af5d2fSBarry Smith     nz = diag[i] - ai[i];
1392f1af5d2fSBarry Smith     s1 = tmp[i];
139304fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
1394da3a660dSBarry Smith   }
1395da3a660dSBarry Smith 
1396da3a660dSBarry Smith   /* copy tmp into x according to permutation */
1397591294e4SBarry Smith   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1398da3a660dSBarry Smith 
13992235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
14002235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
14013649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1402fdc842d1SBarry Smith   ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr);
1403da3a660dSBarry Smith 
1404dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
14053a40ed3dSBarry Smith   PetscFunctionReturn(0);
1406da3a660dSBarry Smith }
1407da3a660dSBarry Smith 
140835233d99SShri Abhyankar PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx)
1409d1fa9404SShri Abhyankar {
1410d1fa9404SShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
1411d1fa9404SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
1412d1fa9404SShri Abhyankar   PetscErrorCode    ierr;
1413d1fa9404SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi;
1414d1fa9404SShri Abhyankar   PetscInt          i,n = A->rmap->n,j;
1415d1fa9404SShri Abhyankar   PetscInt          nz;
1416d1fa9404SShri Abhyankar   PetscScalar       *x,*tmp,s1;
1417d1fa9404SShri Abhyankar   const MatScalar   *aa = a->a,*v;
1418d1fa9404SShri Abhyankar   const PetscScalar *b;
1419d1fa9404SShri Abhyankar 
1420d1fa9404SShri Abhyankar   PetscFunctionBegin;
14213649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
1422fdc842d1SBarry Smith   ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr);
1423d1fa9404SShri Abhyankar   tmp  = a->solve_work;
1424d1fa9404SShri Abhyankar 
1425d1fa9404SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1426d1fa9404SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1427d1fa9404SShri Abhyankar 
1428d1fa9404SShri Abhyankar   /* copy the b into temp work space according to permutation */
1429d1fa9404SShri Abhyankar   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1430d1fa9404SShri Abhyankar 
1431d1fa9404SShri Abhyankar   /* forward solve the U^T */
1432d1fa9404SShri Abhyankar   for (i=0; i<n; i++) {
1433d1fa9404SShri Abhyankar     v   = aa + adiag[i+1] + 1;
1434d1fa9404SShri Abhyankar     vi  = aj + adiag[i+1] + 1;
1435d1fa9404SShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
1436d1fa9404SShri Abhyankar     s1  = tmp[i];
1437d1fa9404SShri Abhyankar     s1 *= v[nz];  /* multiply by inverse of diagonal entry */
1438d1fa9404SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1439d1fa9404SShri Abhyankar     tmp[i] = s1;
1440d1fa9404SShri Abhyankar   }
1441d1fa9404SShri Abhyankar 
1442d1fa9404SShri Abhyankar   /* backward solve the L^T */
1443d1fa9404SShri Abhyankar   for (i=n-1; i>=0; i--) {
1444d1fa9404SShri Abhyankar     v  = aa + ai[i];
1445d1fa9404SShri Abhyankar     vi = aj + ai[i];
1446d1fa9404SShri Abhyankar     nz = ai[i+1] - ai[i];
1447d1fa9404SShri Abhyankar     s1 = tmp[i];
1448d1fa9404SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1449d1fa9404SShri Abhyankar   }
1450d1fa9404SShri Abhyankar 
1451d1fa9404SShri Abhyankar   /* copy tmp into x according to permutation */
1452d1fa9404SShri Abhyankar   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1453d1fa9404SShri Abhyankar 
1454d1fa9404SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1455d1fa9404SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
14563649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1457fdc842d1SBarry Smith   ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr);
1458d1fa9404SShri Abhyankar 
1459d1fa9404SShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
1460d1fa9404SShri Abhyankar   PetscFunctionReturn(0);
1461d1fa9404SShri Abhyankar }
1462d1fa9404SShri Abhyankar 
1463ad04f41aSHong Zhang PetscErrorCode MatSolveTransposeAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec zz,Vec xx)
1464da3a660dSBarry Smith {
1465416022c9SBarry Smith   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
14662235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
14676849ba73SBarry Smith   PetscErrorCode    ierr;
146804fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
146904fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
147004fbf559SBarry Smith   PetscInt          nz;
147104fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1472dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
147304fbf559SBarry Smith   const PetscScalar *b;
14746abc6512SBarry Smith 
14753a40ed3dSBarry Smith   PetscFunctionBegin;
147623bc6035SMatthew Knepley   if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);}
14773649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
14781ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1479416022c9SBarry Smith   tmp  = a->solve_work;
14806abc6512SBarry Smith 
14812235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
14822235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
14836abc6512SBarry Smith 
14846abc6512SBarry Smith   /* copy the b into temp work space according to permutation */
14852235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
14866abc6512SBarry Smith 
14876abc6512SBarry Smith   /* forward solve the U^T */
14886abc6512SBarry Smith   for (i=0; i<n; i++) {
1489010a20caSHong Zhang     v   = aa + diag[i];
1490010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1491f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
149204fbf559SBarry Smith     s1  = tmp[i];
149304fbf559SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
149404fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
149504fbf559SBarry Smith     tmp[i] = s1;
14966abc6512SBarry Smith   }
14976abc6512SBarry Smith 
14986abc6512SBarry Smith   /* backward solve the L^T */
14996abc6512SBarry Smith   for (i=n-1; i>=0; i--) {
1500010a20caSHong Zhang     v  = aa + diag[i] - 1;
1501010a20caSHong Zhang     vi = aj + diag[i] - 1;
1502f1af5d2fSBarry Smith     nz = diag[i] - ai[i];
150304fbf559SBarry Smith     s1 = tmp[i];
150404fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
15056abc6512SBarry Smith   }
15066abc6512SBarry Smith 
15076abc6512SBarry Smith   /* copy tmp into x according to permutation */
15086abc6512SBarry Smith   for (i=0; i<n; i++) x[r[i]] += tmp[i];
15096abc6512SBarry Smith 
15102235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
15112235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
15123649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
15131ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
15146abc6512SBarry Smith 
151504fbf559SBarry Smith   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
15163a40ed3dSBarry Smith   PetscFunctionReturn(0);
1517da3a660dSBarry Smith }
151804fbf559SBarry Smith 
151935233d99SShri Abhyankar PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx)
1520533e6011SShri Abhyankar {
1521533e6011SShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
1522533e6011SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
1523533e6011SShri Abhyankar   PetscErrorCode    ierr;
1524533e6011SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi;
1525533e6011SShri Abhyankar   PetscInt          i,n = A->rmap->n,j;
1526533e6011SShri Abhyankar   PetscInt          nz;
1527533e6011SShri Abhyankar   PetscScalar       *x,*tmp,s1;
1528533e6011SShri Abhyankar   const MatScalar   *aa = a->a,*v;
1529533e6011SShri Abhyankar   const PetscScalar *b;
1530533e6011SShri Abhyankar 
1531533e6011SShri Abhyankar   PetscFunctionBegin;
1532533e6011SShri Abhyankar   if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);}
15333649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
1534533e6011SShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1535533e6011SShri Abhyankar   tmp  = a->solve_work;
1536533e6011SShri Abhyankar 
1537533e6011SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1538533e6011SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1539533e6011SShri Abhyankar 
1540533e6011SShri Abhyankar   /* copy the b into temp work space according to permutation */
1541533e6011SShri Abhyankar   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1542533e6011SShri Abhyankar 
1543533e6011SShri Abhyankar   /* forward solve the U^T */
1544533e6011SShri Abhyankar   for (i=0; i<n; i++) {
1545533e6011SShri Abhyankar     v   = aa + adiag[i+1] + 1;
1546533e6011SShri Abhyankar     vi  = aj + adiag[i+1] + 1;
1547533e6011SShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
1548533e6011SShri Abhyankar     s1  = tmp[i];
1549533e6011SShri Abhyankar     s1 *= v[nz];  /* multiply by inverse of diagonal entry */
1550533e6011SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1551533e6011SShri Abhyankar     tmp[i] = s1;
1552533e6011SShri Abhyankar   }
1553533e6011SShri Abhyankar 
1554533e6011SShri Abhyankar 
1555533e6011SShri Abhyankar   /* backward solve the L^T */
1556533e6011SShri Abhyankar   for (i=n-1; i>=0; i--) {
1557533e6011SShri Abhyankar     v  = aa + ai[i];
1558533e6011SShri Abhyankar     vi = aj + ai[i];
1559533e6011SShri Abhyankar     nz = ai[i+1] - ai[i];
1560533e6011SShri Abhyankar     s1 = tmp[i];
1561c5e3b2a3SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1562533e6011SShri Abhyankar   }
1563533e6011SShri Abhyankar 
1564533e6011SShri Abhyankar   /* copy tmp into x according to permutation */
1565533e6011SShri Abhyankar   for (i=0; i<n; i++) x[r[i]] += tmp[i];
1566533e6011SShri Abhyankar 
1567533e6011SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1568533e6011SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
15693649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1570533e6011SShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1571533e6011SShri Abhyankar 
1572533e6011SShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
1573533e6011SShri Abhyankar   PetscFunctionReturn(0);
1574533e6011SShri Abhyankar }
1575533e6011SShri Abhyankar 
15769e25ed09SBarry Smith /* ----------------------------------------------------------------*/
1577b47f5a65SHong Zhang 
15788fc3a347SHong Zhang /*
15798a76ed4fSHong Zhang    ilu() under revised new data structure.
1580813a1f2bSShri Abhyankar    Factored arrays bj and ba are stored as
1581813a1f2bSShri Abhyankar      L(0,:), L(1,:), ...,L(n-1,:),  U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:)
1582813a1f2bSShri Abhyankar 
1583813a1f2bSShri Abhyankar    bi=fact->i is an array of size n+1, in which
1584813a1f2bSShri Abhyankar    bi+
1585baabb860SHong Zhang      bi[i]:  points to 1st entry of L(i,:),i=0,...,n-1
1586baabb860SHong Zhang      bi[n]:  points to L(n-1,n-1)+1
1587baabb860SHong Zhang 
1588813a1f2bSShri Abhyankar   bdiag=fact->diag is an array of size n+1,in which
1589baabb860SHong Zhang      bdiag[i]: points to diagonal of U(i,:), i=0,...,n-1
1590a24f213cSHong Zhang      bdiag[n]: points to entry of U(n-1,0)-1
1591813a1f2bSShri Abhyankar 
1592813a1f2bSShri Abhyankar    U(i,:) contains bdiag[i] as its last entry, i.e.,
1593813a1f2bSShri Abhyankar     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
1594813a1f2bSShri Abhyankar */
159535233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
1596813a1f2bSShri Abhyankar {
1597813a1f2bSShri Abhyankar   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data,*b;
1598813a1f2bSShri Abhyankar   PetscErrorCode ierr;
1599bbd65245SShri Abhyankar   const PetscInt n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag;
1600bbd65245SShri Abhyankar   PetscInt       i,j,k=0,nz,*bi,*bj,*bdiag;
16011df811f5SHong Zhang   IS             isicol;
1602813a1f2bSShri Abhyankar 
1603813a1f2bSShri Abhyankar   PetscFunctionBegin;
16041df811f5SHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
1605813a1f2bSShri Abhyankar   ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr);
1606813a1f2bSShri Abhyankar   b    = (Mat_SeqAIJ*)(fact)->data;
1607813a1f2bSShri Abhyankar 
1608813a1f2bSShri Abhyankar   /* allocate matrix arrays for new data structure */
1609dcca6d9dSJed Brown   ierr = PetscMalloc3(ai[n]+1,&b->a,ai[n]+1,&b->j,n+1,&b->i);CHKERRQ(ierr);
16103bb1ff40SBarry Smith   ierr = PetscLogObjectMemory((PetscObject)fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(n+1)*sizeof(PetscInt));CHKERRQ(ierr);
16112205254eSKarl Rupp 
1612813a1f2bSShri Abhyankar   b->singlemalloc = PETSC_TRUE;
1613813a1f2bSShri Abhyankar   if (!b->diag) {
1614854ce69bSBarry Smith     ierr = PetscMalloc1(n+1,&b->diag);CHKERRQ(ierr);
16153bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)fact,(n+1)*sizeof(PetscInt));CHKERRQ(ierr);
1616813a1f2bSShri Abhyankar   }
1617813a1f2bSShri Abhyankar   bdiag = b->diag;
1618813a1f2bSShri Abhyankar 
1619813a1f2bSShri Abhyankar   if (n > 0) {
1620580bdb30SBarry Smith     ierr = PetscArrayzero(b->a,ai[n]);CHKERRQ(ierr);
1621813a1f2bSShri Abhyankar   }
1622813a1f2bSShri Abhyankar 
1623813a1f2bSShri Abhyankar   /* set bi and bj with new data structure */
1624813a1f2bSShri Abhyankar   bi = b->i;
1625813a1f2bSShri Abhyankar   bj = b->j;
1626813a1f2bSShri Abhyankar 
1627813a1f2bSShri Abhyankar   /* L part */
1628e60cf9a0SBarry Smith   bi[0] = 0;
1629813a1f2bSShri Abhyankar   for (i=0; i<n; i++) {
1630813a1f2bSShri Abhyankar     nz      = adiag[i] - ai[i];
1631813a1f2bSShri Abhyankar     bi[i+1] = bi[i] + nz;
1632813a1f2bSShri Abhyankar     aj      = a->j + ai[i];
1633813a1f2bSShri Abhyankar     for (j=0; j<nz; j++) {
1634bbd65245SShri Abhyankar       /*   *bj = aj[j]; bj++; */
1635bbd65245SShri Abhyankar       bj[k++] = aj[j];
1636813a1f2bSShri Abhyankar     }
1637813a1f2bSShri Abhyankar   }
1638813a1f2bSShri Abhyankar 
1639813a1f2bSShri Abhyankar   /* U part */
164062fbd6afSShri Abhyankar   bdiag[n] = bi[n]-1;
1641813a1f2bSShri Abhyankar   for (i=n-1; i>=0; i--) {
1642813a1f2bSShri Abhyankar     nz = ai[i+1] - adiag[i] - 1;
1643813a1f2bSShri Abhyankar     aj = a->j + adiag[i] + 1;
1644813a1f2bSShri Abhyankar     for (j=0; j<nz; j++) {
1645bbd65245SShri Abhyankar       /*      *bj = aj[j]; bj++; */
1646bbd65245SShri Abhyankar       bj[k++] = aj[j];
1647813a1f2bSShri Abhyankar     }
1648813a1f2bSShri Abhyankar     /* diag[i] */
1649bbd65245SShri Abhyankar     /*    *bj = i; bj++; */
1650bbd65245SShri Abhyankar     bj[k++]  = i;
1651a24f213cSHong Zhang     bdiag[i] = bdiag[i+1] + nz + 1;
1652813a1f2bSShri Abhyankar   }
16531df811f5SHong Zhang 
1654d5f3da31SBarry Smith   fact->factortype             = MAT_FACTOR_ILU;
16551df811f5SHong Zhang   fact->info.factor_mallocs    = 0;
16561df811f5SHong Zhang   fact->info.fill_ratio_given  = info->fill;
16571df811f5SHong Zhang   fact->info.fill_ratio_needed = 1.0;
165835233d99SShri Abhyankar   fact->ops->lufactornumeric   = MatLUFactorNumeric_SeqAIJ;
1659a02bda8eSBarry Smith   ierr = MatSeqAIJCheckInode_FactorLU(fact);CHKERRQ(ierr);
16601df811f5SHong Zhang 
16611df811f5SHong Zhang   b       = (Mat_SeqAIJ*)(fact)->data;
16621df811f5SHong Zhang   b->row  = isrow;
16631df811f5SHong Zhang   b->col  = iscol;
16641df811f5SHong Zhang   b->icol = isicol;
1665854ce69bSBarry Smith   ierr    = PetscMalloc1(fact->rmap->n+1,&b->solve_work);CHKERRQ(ierr);
16661df811f5SHong Zhang   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
16671df811f5SHong Zhang   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
1668813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
1669813a1f2bSShri Abhyankar }
1670813a1f2bSShri Abhyankar 
167135233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
1672813a1f2bSShri Abhyankar {
1673813a1f2bSShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
1674813a1f2bSShri Abhyankar   IS                 isicol;
1675813a1f2bSShri Abhyankar   PetscErrorCode     ierr;
1676813a1f2bSShri Abhyankar   const PetscInt     *r,*ic;
16771df811f5SHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j;
1678813a1f2bSShri Abhyankar   PetscInt           *bi,*cols,nnz,*cols_lvl;
1679813a1f2bSShri Abhyankar   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
1680813a1f2bSShri Abhyankar   PetscInt           i,levels,diagonal_fill;
1681035f4963SHong Zhang   PetscBool          col_identity,row_identity,missing;
1682813a1f2bSShri Abhyankar   PetscReal          f;
16830298fd71SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=NULL;
1684813a1f2bSShri Abhyankar   PetscBT            lnkbt;
1685813a1f2bSShri Abhyankar   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
16860298fd71SBarry Smith   PetscFreeSpaceList free_space    =NULL,current_space=NULL;
16870298fd71SBarry Smith   PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL;
1688813a1f2bSShri Abhyankar 
1689813a1f2bSShri Abhyankar   PetscFunctionBegin;
1690035f4963SHong 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);
1691035f4963SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
1692035f4963SHong Zhang   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
1693ad04f41aSHong Zhang 
1694813a1f2bSShri Abhyankar   levels = (PetscInt)info->levels;
1695813a1f2bSShri Abhyankar   ierr   = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
1696813a1f2bSShri Abhyankar   ierr   = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
1697813a1f2bSShri Abhyankar   if (!levels && row_identity && col_identity) {
1698813a1f2bSShri Abhyankar     /* special case: ilu(0) with natural ordering */
169935233d99SShri Abhyankar     ierr = MatILUFactorSymbolic_SeqAIJ_ilu0(fact,A,isrow,iscol,info);CHKERRQ(ierr);
1700124a3351SHong Zhang     if (a->inode.size) {
1701124a3351SHong Zhang       fact->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode;
1702124a3351SHong Zhang     }
1703813a1f2bSShri Abhyankar     PetscFunctionReturn(0);
1704813a1f2bSShri Abhyankar   }
1705813a1f2bSShri Abhyankar 
17061df811f5SHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
1707813a1f2bSShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1708813a1f2bSShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
1709813a1f2bSShri Abhyankar 
17101bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
1711854ce69bSBarry Smith   ierr  = PetscMalloc1(n+1,&bi);CHKERRQ(ierr);
1712854ce69bSBarry Smith   ierr  = PetscMalloc1(n+1,&bdiag);CHKERRQ(ierr);
1713e60cf9a0SBarry Smith   bi[0] = bdiag[0] = 0;
1714dcca6d9dSJed Brown   ierr  = PetscMalloc2(n,&bj_ptr,n,&bjlvl_ptr);CHKERRQ(ierr);
1715813a1f2bSShri Abhyankar 
1716813a1f2bSShri Abhyankar   /* create a linked list for storing column indices of the active row */
1717813a1f2bSShri Abhyankar   nlnk = n + 1;
1718813a1f2bSShri Abhyankar   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
1719813a1f2bSShri Abhyankar 
1720813a1f2bSShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
17211df811f5SHong Zhang   f                 = info->fill;
17221df811f5SHong Zhang   diagonal_fill     = (PetscInt)info->diagonal_fill;
17232f18eb33SBarry Smith   ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space);CHKERRQ(ierr);
1724813a1f2bSShri Abhyankar   current_space     = free_space;
17252f18eb33SBarry Smith   ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space_lvl);CHKERRQ(ierr);
1726813a1f2bSShri Abhyankar   current_space_lvl = free_space_lvl;
1727813a1f2bSShri Abhyankar   for (i=0; i<n; i++) {
1728813a1f2bSShri Abhyankar     nzi = 0;
1729813a1f2bSShri Abhyankar     /* copy current row into linked list */
1730813a1f2bSShri Abhyankar     nnz = ai[r[i]+1] - ai[r[i]];
1731e32f2f54SBarry 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);
1732813a1f2bSShri Abhyankar     cols   = aj + ai[r[i]];
1733813a1f2bSShri Abhyankar     lnk[i] = -1; /* marker to indicate if diagonal exists */
1734813a1f2bSShri Abhyankar     ierr   = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
1735813a1f2bSShri Abhyankar     nzi   += nlnk;
1736813a1f2bSShri Abhyankar 
1737813a1f2bSShri Abhyankar     /* make sure diagonal entry is included */
1738813a1f2bSShri Abhyankar     if (diagonal_fill && lnk[i] == -1) {
1739813a1f2bSShri Abhyankar       fm = n;
1740813a1f2bSShri Abhyankar       while (lnk[fm] < i) fm = lnk[fm];
1741813a1f2bSShri Abhyankar       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
1742813a1f2bSShri Abhyankar       lnk[fm]    = i;
1743813a1f2bSShri Abhyankar       lnk_lvl[i] = 0;
1744813a1f2bSShri Abhyankar       nzi++; dcount++;
1745813a1f2bSShri Abhyankar     }
1746813a1f2bSShri Abhyankar 
1747813a1f2bSShri Abhyankar     /* add pivot rows into the active row */
1748813a1f2bSShri Abhyankar     nzbd = 0;
1749813a1f2bSShri Abhyankar     prow = lnk[n];
1750813a1f2bSShri Abhyankar     while (prow < i) {
1751813a1f2bSShri Abhyankar       nnz      = bdiag[prow];
1752813a1f2bSShri Abhyankar       cols     = bj_ptr[prow] + nnz + 1;
1753813a1f2bSShri Abhyankar       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
1754813a1f2bSShri Abhyankar       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
1755813a1f2bSShri Abhyankar       ierr     = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
1756813a1f2bSShri Abhyankar       nzi     += nlnk;
1757813a1f2bSShri Abhyankar       prow     = lnk[prow];
1758813a1f2bSShri Abhyankar       nzbd++;
1759813a1f2bSShri Abhyankar     }
1760813a1f2bSShri Abhyankar     bdiag[i] = nzbd;
1761813a1f2bSShri Abhyankar     bi[i+1]  = bi[i] + nzi;
1762813a1f2bSShri Abhyankar     /* if free space is not available, make more free space */
1763813a1f2bSShri Abhyankar     if (current_space->local_remaining<nzi) {
1764f91af8c7SBarry Smith       nnz  = PetscIntMultTruncate(2,PetscIntMultTruncate(nzi,n - i)); /* estimated and max additional space needed */
1765813a1f2bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
1766813a1f2bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
1767813a1f2bSShri Abhyankar       reallocs++;
1768813a1f2bSShri Abhyankar     }
1769813a1f2bSShri Abhyankar 
1770813a1f2bSShri Abhyankar     /* copy data into free_space and free_space_lvl, then initialize lnk */
1771813a1f2bSShri Abhyankar     ierr         = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
1772813a1f2bSShri Abhyankar     bj_ptr[i]    = current_space->array;
1773813a1f2bSShri Abhyankar     bjlvl_ptr[i] = current_space_lvl->array;
1774813a1f2bSShri Abhyankar 
1775813a1f2bSShri Abhyankar     /* make sure the active row i has diagonal entry */
177665e19b50SBarry 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);
1777813a1f2bSShri Abhyankar 
1778813a1f2bSShri Abhyankar     current_space->array               += nzi;
1779813a1f2bSShri Abhyankar     current_space->local_used          += nzi;
1780813a1f2bSShri Abhyankar     current_space->local_remaining     -= nzi;
1781813a1f2bSShri Abhyankar     current_space_lvl->array           += nzi;
1782813a1f2bSShri Abhyankar     current_space_lvl->local_used      += nzi;
1783813a1f2bSShri Abhyankar     current_space_lvl->local_remaining -= nzi;
1784813a1f2bSShri Abhyankar   }
1785813a1f2bSShri Abhyankar 
1786813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
1787813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
1788813a1f2bSShri Abhyankar   /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */
1789854ce69bSBarry Smith   ierr = PetscMalloc1(bi[n]+1,&bj);CHKERRQ(ierr);
17902ce24eb6SHong Zhang   ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
1791813a1f2bSShri Abhyankar 
1792813a1f2bSShri Abhyankar   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
1793813a1f2bSShri Abhyankar   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
17940e83c824SBarry Smith   ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr);
1795813a1f2bSShri Abhyankar 
1796813a1f2bSShri Abhyankar #if defined(PETSC_USE_INFO)
1797813a1f2bSShri Abhyankar   {
1798aef85c9fSShri Abhyankar     PetscReal af = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
179957622a8eSBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)f,(double)af);CHKERRQ(ierr);
180057622a8eSBarry Smith     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
180157622a8eSBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%g);\n",(double)af);CHKERRQ(ierr);
1802813a1f2bSShri Abhyankar     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
1803813a1f2bSShri Abhyankar     if (diagonal_fill) {
1804955c1f14SBarry Smith       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals\n",dcount);CHKERRQ(ierr);
1805813a1f2bSShri Abhyankar     }
1806813a1f2bSShri Abhyankar   }
1807813a1f2bSShri Abhyankar #endif
1808813a1f2bSShri Abhyankar   /* put together the new matrix */
18090298fd71SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
18103bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)fact,(PetscObject)isicol);CHKERRQ(ierr);
1811813a1f2bSShri Abhyankar   b    = (Mat_SeqAIJ*)(fact)->data;
18122205254eSKarl Rupp 
1813813a1f2bSShri Abhyankar   b->free_a       = PETSC_TRUE;
1814813a1f2bSShri Abhyankar   b->free_ij      = PETSC_TRUE;
1815813a1f2bSShri Abhyankar   b->singlemalloc = PETSC_FALSE;
18162205254eSKarl Rupp 
1817854ce69bSBarry Smith   ierr = PetscMalloc1(bdiag[0]+1,&b->a);CHKERRQ(ierr);
18182205254eSKarl Rupp 
1819813a1f2bSShri Abhyankar   b->j    = bj;
1820813a1f2bSShri Abhyankar   b->i    = bi;
1821813a1f2bSShri Abhyankar   b->diag = bdiag;
1822813a1f2bSShri Abhyankar   b->ilen = 0;
1823813a1f2bSShri Abhyankar   b->imax = 0;
1824813a1f2bSShri Abhyankar   b->row  = isrow;
1825813a1f2bSShri Abhyankar   b->col  = iscol;
1826813a1f2bSShri Abhyankar   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1827813a1f2bSShri Abhyankar   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
1828813a1f2bSShri Abhyankar   b->icol = isicol;
18292205254eSKarl Rupp 
1830854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&b->solve_work);CHKERRQ(ierr);
1831813a1f2bSShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.
1832813a1f2bSShri Abhyankar      Allocate bdiag, solve_work, new a, new j */
18333bb1ff40SBarry Smith   ierr     = PetscLogObjectMemory((PetscObject)fact,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
1834f268cba8SShri Abhyankar   b->maxnz = b->nz = bdiag[0]+1;
18352205254eSKarl Rupp 
1836813a1f2bSShri Abhyankar   (fact)->info.factor_mallocs    = reallocs;
1837813a1f2bSShri Abhyankar   (fact)->info.fill_ratio_given  = f;
1838f268cba8SShri Abhyankar   (fact)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
183935233d99SShri Abhyankar   (fact)->ops->lufactornumeric   = MatLUFactorNumeric_SeqAIJ;
1840f2344125SBarry Smith   if (a->inode.size) {
1841f2344125SBarry Smith     (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode;
1842f2344125SBarry Smith   }
1843a02bda8eSBarry Smith   ierr = MatSeqAIJCheckInode_FactorLU(fact);CHKERRQ(ierr);
1844813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
1845813a1f2bSShri Abhyankar }
1846813a1f2bSShri Abhyankar 
1847ad04f41aSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
18486516239fSHong Zhang {
18496516239fSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
18506516239fSHong Zhang   IS                 isicol;
18516516239fSHong Zhang   PetscErrorCode     ierr;
18526516239fSHong Zhang   const PetscInt     *r,*ic;
1853ece542f9SHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j;
18546516239fSHong Zhang   PetscInt           *bi,*cols,nnz,*cols_lvl;
18556516239fSHong Zhang   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
18566516239fSHong Zhang   PetscInt           i,levels,diagonal_fill;
1857ace3abfcSBarry Smith   PetscBool          col_identity,row_identity;
18586516239fSHong Zhang   PetscReal          f;
18590298fd71SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=NULL;
18606516239fSHong Zhang   PetscBT            lnkbt;
18616516239fSHong Zhang   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
18620298fd71SBarry Smith   PetscFreeSpaceList free_space    =NULL,current_space=NULL;
18630298fd71SBarry Smith   PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL;
1864ace3abfcSBarry Smith   PetscBool          missing;
18656516239fSHong Zhang 
18666516239fSHong Zhang   PetscFunctionBegin;
1867e32f2f54SBarry 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);
1868ece542f9SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
1869ece542f9SHong Zhang   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
1870ece542f9SHong Zhang 
18716516239fSHong Zhang   f             = info->fill;
18726516239fSHong Zhang   levels        = (PetscInt)info->levels;
18736516239fSHong Zhang   diagonal_fill = (PetscInt)info->diagonal_fill;
18742205254eSKarl Rupp 
18756516239fSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
18766516239fSHong Zhang 
18776516239fSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
18786516239fSHong Zhang   ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
18796516239fSHong Zhang   if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */
1880f77e22a1SHong Zhang     ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr);
18812205254eSKarl Rupp 
1882ad04f41aSHong Zhang     (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ_inplace;
1883f2344125SBarry Smith     if (a->inode.size) {
1884f2344125SBarry Smith       (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
1885f2344125SBarry Smith     }
1886d5f3da31SBarry Smith     fact->factortype               = MAT_FACTOR_ILU;
1887719d5645SBarry Smith     (fact)->info.factor_mallocs    = 0;
1888719d5645SBarry Smith     (fact)->info.fill_ratio_given  = info->fill;
1889719d5645SBarry Smith     (fact)->info.fill_ratio_needed = 1.0;
18902205254eSKarl Rupp 
1891719d5645SBarry Smith     b    = (Mat_SeqAIJ*)(fact)->data;
189208480c60SBarry Smith     b->row  = isrow;
189308480c60SBarry Smith     b->col  = iscol;
189482bf6240SBarry Smith     b->icol = isicol;
1895854ce69bSBarry Smith     ierr    = PetscMalloc1((fact)->rmap->n+1,&b->solve_work);CHKERRQ(ierr);
1896c38d4ed2SBarry Smith     ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1897c38d4ed2SBarry Smith     ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
18983a40ed3dSBarry Smith     PetscFunctionReturn(0);
189908480c60SBarry Smith   }
190008480c60SBarry Smith 
1901898183ecSLois Curfman McInnes   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1902898183ecSLois Curfman McInnes   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
19039e25ed09SBarry Smith 
19041bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
1905854ce69bSBarry Smith   ierr  = PetscMalloc1(n+1,&bi);CHKERRQ(ierr);
1906854ce69bSBarry Smith   ierr  = PetscMalloc1(n+1,&bdiag);CHKERRQ(ierr);
1907a6df321fSShri Abhyankar   bi[0] = bdiag[0] = 0;
19086cf997b0SBarry Smith 
1909dcca6d9dSJed Brown   ierr = PetscMalloc2(n,&bj_ptr,n,&bjlvl_ptr);CHKERRQ(ierr);
19105a8e39fbSHong Zhang 
19115a8e39fbSHong Zhang   /* create a linked list for storing column indices of the active row */
19125a8e39fbSHong Zhang   nlnk = n + 1;
19135a8e39fbSHong Zhang   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
19145a8e39fbSHong Zhang 
19155a8e39fbSHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
19162f18eb33SBarry Smith   ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space);CHKERRQ(ierr);
19175a8e39fbSHong Zhang   current_space     = free_space;
19182f18eb33SBarry Smith   ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space_lvl);CHKERRQ(ierr);
19195a8e39fbSHong Zhang   current_space_lvl = free_space_lvl;
19205a8e39fbSHong Zhang 
19215a8e39fbSHong Zhang   for (i=0; i<n; i++) {
19225a8e39fbSHong Zhang     nzi = 0;
19236cf997b0SBarry Smith     /* copy current row into linked list */
19245a8e39fbSHong Zhang     nnz = ai[r[i]+1] - ai[r[i]];
1925e32f2f54SBarry 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);
19265a8e39fbSHong Zhang     cols   = aj + ai[r[i]];
19275a8e39fbSHong Zhang     lnk[i] = -1; /* marker to indicate if diagonal exists */
19285a8e39fbSHong Zhang     ierr   = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
19295a8e39fbSHong Zhang     nzi   += nlnk;
19306cf997b0SBarry Smith 
19316cf997b0SBarry Smith     /* make sure diagonal entry is included */
19325a8e39fbSHong Zhang     if (diagonal_fill && lnk[i] == -1) {
19336cf997b0SBarry Smith       fm = n;
19345a8e39fbSHong Zhang       while (lnk[fm] < i) fm = lnk[fm];
19355a8e39fbSHong Zhang       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
19365a8e39fbSHong Zhang       lnk[fm]    = i;
19375a8e39fbSHong Zhang       lnk_lvl[i] = 0;
19385a8e39fbSHong Zhang       nzi++; dcount++;
19396cf997b0SBarry Smith     }
19406cf997b0SBarry Smith 
19415a8e39fbSHong Zhang     /* add pivot rows into the active row */
19425a8e39fbSHong Zhang     nzbd = 0;
19435a8e39fbSHong Zhang     prow = lnk[n];
19445a8e39fbSHong Zhang     while (prow < i) {
19455a8e39fbSHong Zhang       nnz      = bdiag[prow];
19465a8e39fbSHong Zhang       cols     = bj_ptr[prow] + nnz + 1;
19475a8e39fbSHong Zhang       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
19485a8e39fbSHong Zhang       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
19495a8e39fbSHong Zhang       ierr     = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
19505a8e39fbSHong Zhang       nzi     += nlnk;
19515a8e39fbSHong Zhang       prow     = lnk[prow];
19525a8e39fbSHong Zhang       nzbd++;
195356beaf04SBarry Smith     }
19545a8e39fbSHong Zhang     bdiag[i] = nzbd;
19555a8e39fbSHong Zhang     bi[i+1]  = bi[i] + nzi;
1956ecf371e4SBarry Smith 
19575a8e39fbSHong Zhang     /* if free space is not available, make more free space */
19585a8e39fbSHong Zhang     if (current_space->local_remaining<nzi) {
1959f91af8c7SBarry Smith       nnz  = PetscIntMultTruncate(nzi,n - i); /* estimated and max additional space needed */
1960a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
1961a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
19625a8e39fbSHong Zhang       reallocs++;
19635a8e39fbSHong Zhang     }
1964ecf371e4SBarry Smith 
19655a8e39fbSHong Zhang     /* copy data into free_space and free_space_lvl, then initialize lnk */
19665a8e39fbSHong Zhang     ierr         = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
19675a8e39fbSHong Zhang     bj_ptr[i]    = current_space->array;
19685a8e39fbSHong Zhang     bjlvl_ptr[i] = current_space_lvl->array;
19695a8e39fbSHong Zhang 
19705a8e39fbSHong Zhang     /* make sure the active row i has diagonal entry */
197165e19b50SBarry 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);
19725a8e39fbSHong Zhang 
19735a8e39fbSHong Zhang     current_space->array               += nzi;
19745a8e39fbSHong Zhang     current_space->local_used          += nzi;
19755a8e39fbSHong Zhang     current_space->local_remaining     -= nzi;
19765a8e39fbSHong Zhang     current_space_lvl->array           += nzi;
19775a8e39fbSHong Zhang     current_space_lvl->local_used      += nzi;
19785a8e39fbSHong Zhang     current_space_lvl->local_remaining -= nzi;
19799e25ed09SBarry Smith   }
19805a8e39fbSHong Zhang 
1981898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
1982898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
19835a8e39fbSHong Zhang 
19845a8e39fbSHong Zhang   /* destroy list of free space and other temporary arrays */
1985854ce69bSBarry Smith   ierr = PetscMalloc1(bi[n]+1,&bj);CHKERRQ(ierr);
19866516239fSHong Zhang   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */
19875a8e39fbSHong Zhang   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
1988a1a86e44SBarry Smith   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
19890e83c824SBarry Smith   ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr);
19909e25ed09SBarry Smith 
19916cf91177SBarry Smith #if defined(PETSC_USE_INFO)
1992f64065bbSBarry Smith   {
19935a8e39fbSHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
199457622a8eSBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)f,(double)af);CHKERRQ(ierr);
199557622a8eSBarry Smith     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
199657622a8eSBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%g);\n",(double)af);CHKERRQ(ierr);
1997ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
1998335d9088SBarry Smith     if (diagonal_fill) {
1999955c1f14SBarry Smith       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals\n",dcount);CHKERRQ(ierr);
2000335d9088SBarry Smith     }
2001f64065bbSBarry Smith   }
200263ba0a88SBarry Smith #endif
2003bbb6d6a8SBarry Smith 
20049e25ed09SBarry Smith   /* put together the new matrix */
20050298fd71SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
20063bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)fact,(PetscObject)isicol);CHKERRQ(ierr);
2007719d5645SBarry Smith   b    = (Mat_SeqAIJ*)(fact)->data;
20082205254eSKarl Rupp 
2009e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
2010e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
20117c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
20122205254eSKarl Rupp 
2013785e854fSJed Brown   ierr = PetscMalloc1(bi[n],&b->a);CHKERRQ(ierr);
20145a8e39fbSHong Zhang   b->j = bj;
20155a8e39fbSHong Zhang   b->i = bi;
20165a8e39fbSHong Zhang   for (i=0; i<n; i++) bdiag[i] += bi[i];
20175a8e39fbSHong Zhang   b->diag = bdiag;
2018416022c9SBarry Smith   b->ilen = 0;
2019416022c9SBarry Smith   b->imax = 0;
2020416022c9SBarry Smith   b->row  = isrow;
2021416022c9SBarry Smith   b->col  = iscol;
2022c38d4ed2SBarry Smith   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
2023c38d4ed2SBarry Smith   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
202482bf6240SBarry Smith   b->icol = isicol;
2025854ce69bSBarry Smith   ierr    = PetscMalloc1(n+1,&b->solve_work);CHKERRQ(ierr);
2026416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
20275a8e39fbSHong Zhang      Allocate bdiag, solve_work, new a, new j */
20283bb1ff40SBarry Smith   ierr     = PetscLogObjectMemory((PetscObject)fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
20295a8e39fbSHong Zhang   b->maxnz = b->nz = bi[n];
20302205254eSKarl Rupp 
2031719d5645SBarry Smith   (fact)->info.factor_mallocs    = reallocs;
2032719d5645SBarry Smith   (fact)->info.fill_ratio_given  = f;
2033719d5645SBarry Smith   (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
2034ad04f41aSHong Zhang   (fact)->ops->lufactornumeric   =  MatLUFactorNumeric_SeqAIJ_inplace;
2035f2344125SBarry Smith   if (a->inode.size) {
2036f2344125SBarry Smith     (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
2037d3ac4fa3SBarry Smith   }
20383a40ed3dSBarry Smith   PetscFunctionReturn(0);
20399e25ed09SBarry Smith }
2040d5d45c9bSBarry Smith 
204135233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
20425f44c854SHong Zhang {
20435f44c854SHong Zhang   Mat            C = B;
20445f44c854SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
20455f44c854SHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
20465f44c854SHong Zhang   IS             ip=b->row,iip = b->icol;
20475f44c854SHong Zhang   PetscErrorCode ierr;
20485f44c854SHong Zhang   const PetscInt *rip,*riip;
20495f44c854SHong Zhang   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp;
20505f44c854SHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
20515f44c854SHong Zhang   PetscInt       k,jmin,jmax,*c2r,*il,col,nexti,ili,nz;
20525f44c854SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
2053ace3abfcSBarry Smith   PetscBool      perm_identity;
2054d90ac83dSHong Zhang   FactorShiftCtx sctx;
205598a93161SHong Zhang   PetscReal      rs;
205698a93161SHong Zhang   MatScalar      d,*v;
205798a93161SHong Zhang 
20585f44c854SHong Zhang   PetscFunctionBegin;
205998a93161SHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
2060d90ac83dSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
206198a93161SHong Zhang 
2062f4db908eSBarry Smith   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
206398a93161SHong Zhang     sctx.shift_top = info->zeropivot;
206498a93161SHong Zhang     for (i=0; i<mbs; i++) {
206598a93161SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
206698a93161SHong Zhang       d  = (aa)[a->diag[i]];
206798a93161SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
206898a93161SHong Zhang       v  = aa+ai[i];
206998a93161SHong Zhang       nz = ai[i+1] - ai[i];
20702205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
207198a93161SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
207298a93161SHong Zhang     }
207398a93161SHong Zhang     sctx.shift_top *= 1.1;
207498a93161SHong Zhang     sctx.nshift_max = 5;
207598a93161SHong Zhang     sctx.shift_lo   = 0.;
207698a93161SHong Zhang     sctx.shift_hi   = 1.;
207798a93161SHong Zhang   }
20785f44c854SHong Zhang 
20795f44c854SHong Zhang   ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr);
20805f44c854SHong Zhang   ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr);
20815f44c854SHong Zhang 
20825f44c854SHong Zhang   /* allocate working arrays
20835f44c854SHong Zhang      c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col
20845f44c854SHong 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
20855f44c854SHong Zhang   */
2086dcca6d9dSJed Brown   ierr = PetscMalloc3(mbs,&rtmp,mbs,&il,mbs,&c2r);CHKERRQ(ierr);
20875f44c854SHong Zhang 
208898a93161SHong Zhang   do {
208907b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
209098a93161SHong Zhang 
2091c5546cabSHong Zhang     for (i=0; i<mbs; i++) c2r[i] = mbs;
20922e987568SSatish Balay     if (mbs) il[0] = 0;
20935f44c854SHong Zhang 
20945f44c854SHong Zhang     for (k = 0; k<mbs; k++) {
20955f44c854SHong Zhang       /* zero rtmp */
20965f44c854SHong Zhang       nz    = bi[k+1] - bi[k];
20975f44c854SHong Zhang       bjtmp = bj + bi[k];
20985f44c854SHong Zhang       for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
20995f44c854SHong Zhang 
21005f44c854SHong Zhang       /* load in initial unfactored row */
21015f44c854SHong Zhang       bval = ba + bi[k];
21025f44c854SHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
21035f44c854SHong Zhang       for (j = jmin; j < jmax; j++) {
21045f44c854SHong Zhang         col = riip[aj[j]];
21055f44c854SHong Zhang         if (col >= k) { /* only take upper triangular entry */
21065f44c854SHong Zhang           rtmp[col] = aa[j];
21075f44c854SHong Zhang           *bval++   = 0.0; /* for in-place factorization */
21085f44c854SHong Zhang         }
21095f44c854SHong Zhang       }
211098a93161SHong Zhang       /* shift the diagonal of the matrix: ZeropivotApply() */
211198a93161SHong Zhang       rtmp[k] += sctx.shift_amount;  /* shift the diagonal of the matrix */
21125f44c854SHong Zhang 
21135f44c854SHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
21145f44c854SHong Zhang       dk = rtmp[k];
21155f44c854SHong Zhang       i  = c2r[k]; /* first row to be added to k_th row  */
21165f44c854SHong Zhang 
21175f44c854SHong Zhang       while (i < k) {
21185f44c854SHong Zhang         nexti = c2r[i]; /* next row to be added to k_th row */
21195f44c854SHong Zhang 
21205f44c854SHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
21215f44c854SHong Zhang         ili     = il[i]; /* index of first nonzero element in U(i,k:bms-1) */
21225f44c854SHong Zhang         uikdi   = -ba[ili]*ba[bdiag[i]]; /* diagonal(k) */
21235f44c854SHong Zhang         dk     += uikdi*ba[ili]; /* update diag[k] */
21245f44c854SHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
21255f44c854SHong Zhang 
21265f44c854SHong Zhang         /* add multiple of row i to k-th row */
21275f44c854SHong Zhang         jmin = ili + 1; jmax = bi[i+1];
21285f44c854SHong Zhang         if (jmin < jmax) {
21295f44c854SHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
21305f44c854SHong Zhang           /* update il and c2r for row i */
21315f44c854SHong Zhang           il[i] = jmin;
21325f44c854SHong Zhang           j     = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i;
21335f44c854SHong Zhang         }
21345f44c854SHong Zhang         i = nexti;
21355f44c854SHong Zhang       }
21365f44c854SHong Zhang 
21375f44c854SHong Zhang       /* copy data into U(k,:) */
213898a93161SHong Zhang       rs   = 0.0;
21395f44c854SHong Zhang       jmin = bi[k]; jmax = bi[k+1]-1;
21405f44c854SHong Zhang       if (jmin < jmax) {
21415f44c854SHong Zhang         for (j=jmin; j<jmax; j++) {
214298a93161SHong Zhang           col = bj[j]; ba[j] = rtmp[col]; rs += PetscAbsScalar(ba[j]);
21435f44c854SHong Zhang         }
21445f44c854SHong Zhang         /* add the k-th row into il and c2r */
21455f44c854SHong Zhang         il[k] = jmin;
21465f44c854SHong Zhang         i     = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k;
21475f44c854SHong Zhang       }
214898a93161SHong Zhang 
214998a93161SHong Zhang       /* MatPivotCheck() */
215098a93161SHong Zhang       sctx.rs = rs;
215198a93161SHong Zhang       sctx.pv = dk;
21524cccfbddSHong Zhang       ierr    = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr);
215307b50cabSHong Zhang       if (sctx.newshift) break;
215498a93161SHong Zhang       dk = sctx.pv;
215598a93161SHong Zhang 
215698a93161SHong Zhang       ba[bdiag[k]] = 1.0/dk; /* U(k,k) */
215798a93161SHong Zhang     }
215807b50cabSHong Zhang   } while (sctx.newshift);
21595f44c854SHong Zhang 
21600e83c824SBarry Smith   ierr = PetscFree3(rtmp,il,c2r);CHKERRQ(ierr);
21615f44c854SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
21625f44c854SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
21635f44c854SHong Zhang 
21645f44c854SHong Zhang   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
21655f44c854SHong Zhang   if (perm_identity) {
216635233d99SShri Abhyankar     B->ops->solve          = MatSolve_SeqSBAIJ_1_NaturalOrdering;
216735233d99SShri Abhyankar     B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering;
21682169352eSHong Zhang     B->ops->forwardsolve   = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering;
21692169352eSHong Zhang     B->ops->backwardsolve  = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering;
21705f44c854SHong Zhang   } else {
217135233d99SShri Abhyankar     B->ops->solve          = MatSolve_SeqSBAIJ_1;
217235233d99SShri Abhyankar     B->ops->solvetranspose = MatSolve_SeqSBAIJ_1;
217335233d99SShri Abhyankar     B->ops->forwardsolve   = MatForwardSolve_SeqSBAIJ_1;
217435233d99SShri Abhyankar     B->ops->backwardsolve  = MatBackwardSolve_SeqSBAIJ_1;
21755f44c854SHong Zhang   }
21765f44c854SHong Zhang 
21775f44c854SHong Zhang   C->assembled    = PETSC_TRUE;
21785f44c854SHong Zhang   C->preallocated = PETSC_TRUE;
21792205254eSKarl Rupp 
21805f44c854SHong Zhang   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
218198a93161SHong Zhang 
218298a93161SHong Zhang   /* MatPivotView() */
218398a93161SHong Zhang   if (sctx.nshift) {
2184f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
218557622a8eSBarry 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);
2186f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
218757622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
2188f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS) {
218957622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %g\n",sctx.nshift,(double)info->shiftamount);CHKERRQ(ierr);
219098a93161SHong Zhang     }
219198a93161SHong Zhang   }
21925f44c854SHong Zhang   PetscFunctionReturn(0);
21935f44c854SHong Zhang }
21945f44c854SHong Zhang 
21950a3c351aSHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info)
2196a6175056SHong Zhang {
2197719d5645SBarry Smith   Mat            C = B;
21980968510aSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
21992ed133dbSHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
22009bfd6278SHong Zhang   IS             ip=b->row,iip = b->icol;
22012ed133dbSHong Zhang   PetscErrorCode ierr;
22025d0c19d7SBarry Smith   const PetscInt *rip,*riip;
2203c5546cabSHong Zhang   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol,*bjtmp;
22042ed133dbSHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
22052ed133dbSHong Zhang   PetscInt       k,jmin,jmax,*jl,*il,col,nexti,ili,nz;
2206622e2028SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
2207ace3abfcSBarry Smith   PetscBool      perm_identity;
22080e95ead3SHong Zhang   FactorShiftCtx sctx;
22090e95ead3SHong Zhang   PetscReal      rs;
22100e95ead3SHong Zhang   MatScalar      d,*v;
2211d5d45c9bSBarry Smith 
2212a6175056SHong Zhang   PetscFunctionBegin;
22130e95ead3SHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
22140e95ead3SHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
22150e95ead3SHong Zhang 
22160e95ead3SHong Zhang   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
22170e95ead3SHong Zhang     sctx.shift_top = info->zeropivot;
22180e95ead3SHong Zhang     for (i=0; i<mbs; i++) {
22190e95ead3SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
22200e95ead3SHong Zhang       d  = (aa)[a->diag[i]];
22210e95ead3SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
22220e95ead3SHong Zhang       v  = aa+ai[i];
22230e95ead3SHong Zhang       nz = ai[i+1] - ai[i];
22242205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
22250e95ead3SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
22260e95ead3SHong Zhang     }
22270e95ead3SHong Zhang     sctx.shift_top *= 1.1;
22280e95ead3SHong Zhang     sctx.nshift_max = 5;
22290e95ead3SHong Zhang     sctx.shift_lo   = 0.;
22300e95ead3SHong Zhang     sctx.shift_hi   = 1.;
22310e95ead3SHong Zhang   }
2232ee45ca4aSHong Zhang 
22332ed133dbSHong Zhang   ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr);
22349bfd6278SHong Zhang   ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr);
22352ed133dbSHong Zhang 
22362ed133dbSHong Zhang   /* initialization */
2237dcca6d9dSJed Brown   ierr = PetscMalloc3(mbs,&rtmp,mbs,&il,mbs,&jl);CHKERRQ(ierr);
22380e95ead3SHong Zhang 
22392ed133dbSHong Zhang   do {
224007b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
22410e95ead3SHong Zhang 
2242c5546cabSHong Zhang     for (i=0; i<mbs; i++) jl[i] = mbs;
2243c5546cabSHong Zhang     il[0] = 0;
22442ed133dbSHong Zhang 
22452ed133dbSHong Zhang     for (k = 0; k<mbs; k++) {
2246c5546cabSHong Zhang       /* zero rtmp */
2247c5546cabSHong Zhang       nz    = bi[k+1] - bi[k];
2248c5546cabSHong Zhang       bjtmp = bj + bi[k];
2249c5546cabSHong Zhang       for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
2250c5546cabSHong Zhang 
2251c05c3958SHong Zhang       bval = ba + bi[k];
22522ed133dbSHong Zhang       /* initialize k-th row by the perm[k]-th row of A */
22532ed133dbSHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
22542ed133dbSHong Zhang       for (j = jmin; j < jmax; j++) {
22559bfd6278SHong Zhang         col = riip[aj[j]];
22562ed133dbSHong Zhang         if (col >= k) { /* only take upper triangular entry */
22572ed133dbSHong Zhang           rtmp[col] = aa[j];
2258c05c3958SHong Zhang           *bval++   = 0.0; /* for in-place factorization */
22592ed133dbSHong Zhang         }
22602ed133dbSHong Zhang       }
226139e3d630SHong Zhang       /* shift the diagonal of the matrix */
2262540703acSHong Zhang       if (sctx.nshift) rtmp[k] += sctx.shift_amount;
22632ed133dbSHong Zhang 
22642ed133dbSHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
22652ed133dbSHong Zhang       dk = rtmp[k];
22662ed133dbSHong Zhang       i  = jl[k]; /* first row to be added to k_th row  */
22672ed133dbSHong Zhang 
22682ed133dbSHong Zhang       while (i < k) {
22692ed133dbSHong Zhang         nexti = jl[i]; /* next row to be added to k_th row */
22702ed133dbSHong Zhang 
22712ed133dbSHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
22722ed133dbSHong Zhang         ili     = il[i]; /* index of first nonzero element in U(i,k:bms-1) */
22732ed133dbSHong Zhang         uikdi   = -ba[ili]*ba[bi[i]]; /* diagonal(k) */
22742ed133dbSHong Zhang         dk     += uikdi*ba[ili];
22752ed133dbSHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
22762ed133dbSHong Zhang 
22772ed133dbSHong Zhang         /* add multiple of row i to k-th row */
22782ed133dbSHong Zhang         jmin = ili + 1; jmax = bi[i+1];
22792ed133dbSHong Zhang         if (jmin < jmax) {
22802ed133dbSHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
22812ed133dbSHong Zhang           /* update il and jl for row i */
22822ed133dbSHong Zhang           il[i] = jmin;
22832ed133dbSHong Zhang           j     = bj[jmin]; jl[i] = jl[j]; jl[j] = i;
22842ed133dbSHong Zhang         }
22852ed133dbSHong Zhang         i = nexti;
22862ed133dbSHong Zhang       }
22872ed133dbSHong Zhang 
22880697b6caSHong Zhang       /* shift the diagonals when zero pivot is detected */
22890697b6caSHong Zhang       /* compute rs=sum of abs(off-diagonal) */
22900697b6caSHong Zhang       rs   = 0.0;
22913c9e8598SHong Zhang       jmin = bi[k]+1;
22923c9e8598SHong Zhang       nz   = bi[k+1] - jmin;
22933c9e8598SHong Zhang       bcol = bj + jmin;
229404fbf559SBarry Smith       for (j=0; j<nz; j++) {
229504fbf559SBarry Smith         rs += PetscAbsScalar(rtmp[bcol[j]]);
22963c9e8598SHong Zhang       }
2297540703acSHong Zhang 
2298540703acSHong Zhang       sctx.rs = rs;
2299540703acSHong Zhang       sctx.pv = dk;
23004cccfbddSHong Zhang       ierr    = MatPivotCheck(B,A,info,&sctx,k);CHKERRQ(ierr);
230107b50cabSHong Zhang       if (sctx.newshift) break;
23020e95ead3SHong Zhang       dk = sctx.pv;
23033c9e8598SHong Zhang 
23043c9e8598SHong Zhang       /* copy data into U(k,:) */
230539e3d630SHong Zhang       ba[bi[k]] = 1.0/dk; /* U(k,k) */
230639e3d630SHong Zhang       jmin      = bi[k]+1; jmax = bi[k+1];
230739e3d630SHong Zhang       if (jmin < jmax) {
230839e3d630SHong Zhang         for (j=jmin; j<jmax; j++) {
2309c6ae9bfcSHong Zhang           col = bj[j]; ba[j] = rtmp[col];
23103c9e8598SHong Zhang         }
231139e3d630SHong Zhang         /* add the k-th row into il and jl */
23123c9e8598SHong Zhang         il[k] = jmin;
23133c9e8598SHong Zhang         i     = bj[jmin]; jl[k] = jl[i]; jl[i] = k;
23143c9e8598SHong Zhang       }
23153c9e8598SHong Zhang     }
231607b50cabSHong Zhang   } while (sctx.newshift);
23170e95ead3SHong Zhang 
23180e83c824SBarry Smith   ierr = PetscFree3(rtmp,il,jl);CHKERRQ(ierr);
231939e3d630SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
23209bfd6278SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
2321db4efbfdSBarry Smith 
2322db4efbfdSBarry Smith   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
2323db4efbfdSBarry Smith   if (perm_identity) {
23240a3c351aSHong Zhang     B->ops->solve          = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23250a3c351aSHong Zhang     B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23260a3c351aSHong Zhang     B->ops->forwardsolve   = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23270a3c351aSHong Zhang     B->ops->backwardsolve  = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
2328db4efbfdSBarry Smith   } else {
23290a3c351aSHong Zhang     B->ops->solve          = MatSolve_SeqSBAIJ_1_inplace;
23300a3c351aSHong Zhang     B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_inplace;
23310a3c351aSHong Zhang     B->ops->forwardsolve   = MatForwardSolve_SeqSBAIJ_1_inplace;
23320a3c351aSHong Zhang     B->ops->backwardsolve  = MatBackwardSolve_SeqSBAIJ_1_inplace;
2333db4efbfdSBarry Smith   }
2334db4efbfdSBarry Smith 
23353c9e8598SHong Zhang   C->assembled    = PETSC_TRUE;
23363c9e8598SHong Zhang   C->preallocated = PETSC_TRUE;
23372205254eSKarl Rupp 
2338d0f46423SBarry Smith   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
2339540703acSHong Zhang   if (sctx.nshift) {
2340f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
234157622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
2342f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
234357622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
23440697b6caSHong Zhang     }
23453c9e8598SHong Zhang   }
23463c9e8598SHong Zhang   PetscFunctionReturn(0);
23473c9e8598SHong Zhang }
2348a6175056SHong Zhang 
23498a76ed4fSHong Zhang /*
23508a76ed4fSHong Zhang    icc() under revised new data structure.
23518a76ed4fSHong Zhang    Factored arrays bj and ba are stored as
23528a76ed4fSHong Zhang      U(0,:),...,U(i,:),U(n-1,:)
23538a76ed4fSHong Zhang 
23548a76ed4fSHong Zhang    ui=fact->i is an array of size n+1, in which
23558a76ed4fSHong Zhang    ui+
23568a76ed4fSHong Zhang      ui[i]:  points to 1st entry of U(i,:),i=0,...,n-1
23578a76ed4fSHong Zhang      ui[n]:  points to U(n-1,n-1)+1
23588a76ed4fSHong Zhang 
23598a76ed4fSHong Zhang   udiag=fact->diag is an array of size n,in which
23608a76ed4fSHong Zhang      udiag[i]: points to diagonal of U(i,:), i=0,...,n-1
23618a76ed4fSHong Zhang 
23628a76ed4fSHong Zhang    U(i,:) contains udiag[i] as its last entry, i.e.,
23638a76ed4fSHong Zhang     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
23648a76ed4fSHong Zhang */
23658a76ed4fSHong Zhang 
236635233d99SShri Abhyankar PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2367a6175056SHong Zhang {
23680968510aSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
2369ed59401aSHong Zhang   Mat_SeqSBAIJ       *b;
2370dfbe8321SBarry Smith   PetscErrorCode     ierr;
2371ace3abfcSBarry Smith   PetscBool          perm_identity,missing;
23725f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
23735d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
2374ed59401aSHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
23750298fd71SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=NULL,d;
23765a8e39fbSHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
2377ed59401aSHong Zhang   PetscReal          fill          =info->fill,levels=info->levels;
23780298fd71SBarry Smith   PetscFreeSpaceList free_space    =NULL,current_space=NULL;
23790298fd71SBarry Smith   PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL;
23807a48dd6fSHong Zhang   PetscBT            lnkbt;
2381b635d36bSHong Zhang   IS                 iperm;
2382a6175056SHong Zhang 
2383a6175056SHong Zhang   PetscFunctionBegin;
2384e32f2f54SBarry 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);
238558ebbce7SBarry Smith   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
2386e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
2387653a6975SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
2388b635d36bSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
23897a48dd6fSHong Zhang 
2390854ce69bSBarry Smith   ierr  = PetscMalloc1(am+1,&ui);CHKERRQ(ierr);
2391854ce69bSBarry Smith   ierr  = PetscMalloc1(am+1,&udiag);CHKERRQ(ierr);
239239e3d630SHong Zhang   ui[0] = 0;
239339e3d630SHong Zhang 
23948a76ed4fSHong Zhang   /* ICC(0) without matrix ordering: simply rearrange column indices */
2395622e2028SHong Zhang   if (!levels && perm_identity) {
23965f44c854SHong Zhang     for (i=0; i<am; i++) {
2397c5546cabSHong Zhang       ncols    = ai[i+1] - a->diag[i];
2398c5546cabSHong Zhang       ui[i+1]  = ui[i] + ncols;
2399c5546cabSHong Zhang       udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */
2400dc1e2a3fSHong Zhang     }
2401854ce69bSBarry Smith     ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr);
2402dc1e2a3fSHong Zhang     cols = uj;
2403dc1e2a3fSHong Zhang     for (i=0; i<am; i++) {
24045f44c854SHong Zhang       aj    = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */
2405dc1e2a3fSHong Zhang       ncols = ai[i+1] - a->diag[i] -1;
24065f44c854SHong Zhang       for (j=0; j<ncols; j++) *cols++ = aj[j];
24075f44c854SHong Zhang       *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */
24085f44c854SHong Zhang     }
24095f44c854SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
24105f44c854SHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
24115f44c854SHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
24125f44c854SHong Zhang 
24135f44c854SHong Zhang     /* initialization */
2414854ce69bSBarry Smith     ierr = PetscMalloc1(am+1,&ajtmp);CHKERRQ(ierr);
24155f44c854SHong Zhang 
24165f44c854SHong Zhang     /* jl: linked list for storing indices of the pivot rows
24175f44c854SHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
2418dcca6d9dSJed Brown     ierr = PetscMalloc4(am,&uj_ptr,am,&uj_lvl_ptr,am,&jl,am,&il);CHKERRQ(ierr);
24195f44c854SHong Zhang     for (i=0; i<am; i++) {
24205f44c854SHong Zhang       jl[i] = am; il[i] = 0;
24215f44c854SHong Zhang     }
24225f44c854SHong Zhang 
24235f44c854SHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
24245f44c854SHong Zhang     nlnk = am + 1;
24255f44c854SHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
24265f44c854SHong Zhang 
242795b5ac22SHong Zhang     /* initial FreeSpace size is fill*(ai[am]+am)/2 */
24282f18eb33SBarry Smith     ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,(ai[am]+am)/2),&free_space);CHKERRQ(ierr);
24295f44c854SHong Zhang     current_space     = free_space;
24302f18eb33SBarry Smith     ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,(ai[am]+am)/2),&free_space_lvl);CHKERRQ(ierr);
24315f44c854SHong Zhang     current_space_lvl = free_space_lvl;
24325f44c854SHong Zhang 
24335f44c854SHong Zhang     for (k=0; k<am; k++) {  /* for each active row k */
24345f44c854SHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
24355f44c854SHong Zhang       nzk   = 0;
24365f44c854SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
2437e32f2f54SBarry 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);
24385f44c854SHong Zhang       ncols_upper = 0;
24395f44c854SHong Zhang       for (j=0; j<ncols; j++) {
24405f44c854SHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
24415f44c854SHong Zhang         if (riip[i] >= k) { /* only take upper triangular entry */
24425f44c854SHong Zhang           ajtmp[ncols_upper] = i;
24435f44c854SHong Zhang           ncols_upper++;
24445f44c854SHong Zhang         }
24455f44c854SHong Zhang       }
24465f44c854SHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
24475f44c854SHong Zhang       nzk += nlnk;
24485f44c854SHong Zhang 
24495f44c854SHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
24505f44c854SHong Zhang       prow = jl[k]; /* 1st pivot row */
24515f44c854SHong Zhang 
24525f44c854SHong Zhang       while (prow < k) {
24535f44c854SHong Zhang         nextprow = jl[prow];
24545f44c854SHong Zhang 
24555f44c854SHong Zhang         /* merge prow into k-th row */
24565f44c854SHong Zhang         jmin  = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */
24575f44c854SHong Zhang         jmax  = ui[prow+1];
24585f44c854SHong Zhang         ncols = jmax-jmin;
24595f44c854SHong Zhang         i     = jmin - ui[prow];
24605f44c854SHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
24615f44c854SHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
24625f44c854SHong Zhang         j     = *(uj - 1);
24635f44c854SHong Zhang         ierr  = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
24645f44c854SHong Zhang         nzk  += nlnk;
24655f44c854SHong Zhang 
24665f44c854SHong Zhang         /* update il and jl for prow */
24675f44c854SHong Zhang         if (jmin < jmax) {
24685f44c854SHong Zhang           il[prow] = jmin;
24695f44c854SHong Zhang           j        = *cols; jl[prow] = jl[j]; jl[j] = prow;
24705f44c854SHong Zhang         }
24715f44c854SHong Zhang         prow = nextprow;
24725f44c854SHong Zhang       }
24735f44c854SHong Zhang 
24745f44c854SHong Zhang       /* if free space is not available, make more free space */
24755f44c854SHong Zhang       if (current_space->local_remaining<nzk) {
24765f44c854SHong Zhang         i    = am - k + 1; /* num of unfactored rows */
2477f91af8c7SBarry Smith         i    = PetscIntMultTruncate(i,PetscMin(nzk, i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
24785f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
24795f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
24805f44c854SHong Zhang         reallocs++;
24815f44c854SHong Zhang       }
24825f44c854SHong Zhang 
24835f44c854SHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
2484e32f2f54SBarry Smith       if (nzk == 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
24855f44c854SHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
24865f44c854SHong Zhang 
24875f44c854SHong Zhang       /* add the k-th row into il and jl */
24885f44c854SHong Zhang       if (nzk > 1) {
24895f44c854SHong Zhang         i     = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
24905f44c854SHong Zhang         jl[k] = jl[i]; jl[i] = k;
24915f44c854SHong Zhang         il[k] = ui[k] + 1;
24925f44c854SHong Zhang       }
24935f44c854SHong Zhang       uj_ptr[k]     = current_space->array;
24945f44c854SHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
24955f44c854SHong Zhang 
24965f44c854SHong Zhang       current_space->array           += nzk;
24975f44c854SHong Zhang       current_space->local_used      += nzk;
24985f44c854SHong Zhang       current_space->local_remaining -= nzk;
24995f44c854SHong Zhang 
25005f44c854SHong Zhang       current_space_lvl->array           += nzk;
25015f44c854SHong Zhang       current_space_lvl->local_used      += nzk;
25025f44c854SHong Zhang       current_space_lvl->local_remaining -= nzk;
25035f44c854SHong Zhang 
25045f44c854SHong Zhang       ui[k+1] = ui[k] + nzk;
25055f44c854SHong Zhang     }
25065f44c854SHong Zhang 
25075f44c854SHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
25085f44c854SHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
25090e83c824SBarry Smith     ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr);
25105f44c854SHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
25115f44c854SHong Zhang 
25129263d837SHong Zhang     /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */
2513854ce69bSBarry Smith     ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr);
251435233d99SShri Abhyankar     ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor  */
25155f44c854SHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
25165f44c854SHong Zhang     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
25175f44c854SHong Zhang 
25185f44c854SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
25195f44c854SHong Zhang 
25205f44c854SHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
25215f44c854SHong Zhang   b               = (Mat_SeqSBAIJ*)(fact)->data;
25225f44c854SHong Zhang   b->singlemalloc = PETSC_FALSE;
25232205254eSKarl Rupp 
2524854ce69bSBarry Smith   ierr = PetscMalloc1(ui[am]+1,&b->a);CHKERRQ(ierr);
25252205254eSKarl Rupp 
25265f44c854SHong Zhang   b->j             = uj;
25275f44c854SHong Zhang   b->i             = ui;
25285f44c854SHong Zhang   b->diag          = udiag;
25297f53bb6cSHong Zhang   b->free_diag     = PETSC_TRUE;
25305f44c854SHong Zhang   b->ilen          = 0;
25315f44c854SHong Zhang   b->imax          = 0;
25325f44c854SHong Zhang   b->row           = perm;
25335f44c854SHong Zhang   b->col           = perm;
25345f44c854SHong Zhang   ierr             = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
25355f44c854SHong Zhang   ierr             = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
25365f44c854SHong Zhang   b->icol          = iperm;
25375f44c854SHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
25382205254eSKarl Rupp 
2539854ce69bSBarry Smith   ierr = PetscMalloc1(am+1,&b->solve_work);CHKERRQ(ierr);
25403bb1ff40SBarry Smith   ierr = PetscLogObjectMemory((PetscObject)fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
25412205254eSKarl Rupp 
25425f44c854SHong Zhang   b->maxnz   = b->nz = ui[am];
25435f44c854SHong Zhang   b->free_a  = PETSC_TRUE;
25445f44c854SHong Zhang   b->free_ij = PETSC_TRUE;
25455f44c854SHong Zhang 
2546f284f12aSHong Zhang   fact->info.factor_mallocs   = reallocs;
2547f284f12aSHong Zhang   fact->info.fill_ratio_given = fill;
25485f44c854SHong Zhang   if (ai[am] != 0) {
25496a69fef8SHong Zhang     /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */
255095b5ac22SHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am);
25515f44c854SHong Zhang   } else {
2552f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
25535f44c854SHong Zhang   }
25549263d837SHong Zhang #if defined(PETSC_USE_INFO)
25559263d837SHong Zhang   if (ai[am] != 0) {
25569263d837SHong Zhang     PetscReal af = fact->info.fill_ratio_needed;
255757622a8eSBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)fill,(double)af);CHKERRQ(ierr);
255857622a8eSBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
255957622a8eSBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g) for best performance.\n",(double)af);CHKERRQ(ierr);
25609263d837SHong Zhang   } else {
2561dbf6bb8dSprj-     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
25629263d837SHong Zhang   }
25639263d837SHong Zhang #endif
256435233d99SShri Abhyankar   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
25655f44c854SHong Zhang   PetscFunctionReturn(0);
25665f44c854SHong Zhang }
25675f44c854SHong Zhang 
2568ad04f41aSHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
25695f44c854SHong Zhang {
25705f44c854SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
25715f44c854SHong Zhang   Mat_SeqSBAIJ       *b;
25725f44c854SHong Zhang   PetscErrorCode     ierr;
2573ace3abfcSBarry Smith   PetscBool          perm_identity,missing;
25745f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
25755f44c854SHong Zhang   const PetscInt     *rip,*riip;
25765f44c854SHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
25770298fd71SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=NULL,d;
25785f44c854SHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
25795f44c854SHong Zhang   PetscReal          fill          =info->fill,levels=info->levels;
25800298fd71SBarry Smith   PetscFreeSpaceList free_space    =NULL,current_space=NULL;
25810298fd71SBarry Smith   PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL;
25825f44c854SHong Zhang   PetscBT            lnkbt;
25835f44c854SHong Zhang   IS                 iperm;
25845f44c854SHong Zhang 
25855f44c854SHong Zhang   PetscFunctionBegin;
2586e32f2f54SBarry 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);
25875f44c854SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
2588e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
25895f44c854SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
25905f44c854SHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
25915f44c854SHong Zhang 
2592854ce69bSBarry Smith   ierr  = PetscMalloc1(am+1,&ui);CHKERRQ(ierr);
2593854ce69bSBarry Smith   ierr  = PetscMalloc1(am+1,&udiag);CHKERRQ(ierr);
25945f44c854SHong Zhang   ui[0] = 0;
25955f44c854SHong Zhang 
25965f44c854SHong Zhang   /* ICC(0) without matrix ordering: simply copies fill pattern */
25975f44c854SHong Zhang   if (!levels && perm_identity) {
25985f44c854SHong Zhang 
25995f44c854SHong Zhang     for (i=0; i<am; i++) {
26005f44c854SHong Zhang       ui[i+1]  = ui[i] + ai[i+1] - a->diag[i];
26015f44c854SHong Zhang       udiag[i] = ui[i];
2602ed59401aSHong Zhang     }
2603854ce69bSBarry Smith     ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr);
260439e3d630SHong Zhang     cols = uj;
2605ed59401aSHong Zhang     for (i=0; i<am; i++) {
2606ed59401aSHong Zhang       aj    = a->j + a->diag[i];
260739e3d630SHong Zhang       ncols = ui[i+1] - ui[i];
260839e3d630SHong Zhang       for (j=0; j<ncols; j++) *cols++ = *aj++;
2609ed59401aSHong Zhang     }
261039e3d630SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
2611b635d36bSHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
2612b635d36bSHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
2613b635d36bSHong Zhang 
26147a48dd6fSHong Zhang     /* initialization */
2615854ce69bSBarry Smith     ierr = PetscMalloc1(am+1,&ajtmp);CHKERRQ(ierr);
26167a48dd6fSHong Zhang 
26177a48dd6fSHong Zhang     /* jl: linked list for storing indices of the pivot rows
26187a48dd6fSHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
2619dcca6d9dSJed Brown     ierr = PetscMalloc4(am,&uj_ptr,am,&uj_lvl_ptr,am,&jl,am,&il);CHKERRQ(ierr);
26207a48dd6fSHong Zhang     for (i=0; i<am; i++) {
26217a48dd6fSHong Zhang       jl[i] = am; il[i] = 0;
26227a48dd6fSHong Zhang     }
26237a48dd6fSHong Zhang 
26247a48dd6fSHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
26257a48dd6fSHong Zhang     nlnk = am + 1;
26262ed133dbSHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
26277a48dd6fSHong Zhang 
26287a48dd6fSHong Zhang     /* initial FreeSpace size is fill*(ai[am]+1) */
26292f18eb33SBarry Smith     ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,ai[am]+1),&free_space);CHKERRQ(ierr);
26307a48dd6fSHong Zhang     current_space     = free_space;
26312f18eb33SBarry Smith     ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,ai[am]+1),&free_space_lvl);CHKERRQ(ierr);
26327a48dd6fSHong Zhang     current_space_lvl = free_space_lvl;
26337a48dd6fSHong Zhang 
26347a48dd6fSHong Zhang     for (k=0; k<am; k++) {  /* for each active row k */
26357a48dd6fSHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
26367a48dd6fSHong Zhang       nzk   = 0;
2637622e2028SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
2638e32f2f54SBarry 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);
2639622e2028SHong Zhang       ncols_upper = 0;
2640622e2028SHong Zhang       for (j=0; j<ncols; j++) {
2641b635d36bSHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
2642b635d36bSHong Zhang         if (riip[i] >= k) { /* only take upper triangular entry */
26435a8e39fbSHong Zhang           ajtmp[ncols_upper] = i;
2644622e2028SHong Zhang           ncols_upper++;
2645622e2028SHong Zhang         }
2646622e2028SHong Zhang       }
2647b635d36bSHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
26487a48dd6fSHong Zhang       nzk += nlnk;
26497a48dd6fSHong Zhang 
26507a48dd6fSHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
26517a48dd6fSHong Zhang       prow = jl[k]; /* 1st pivot row */
26527a48dd6fSHong Zhang 
26537a48dd6fSHong Zhang       while (prow < k) {
26547a48dd6fSHong Zhang         nextprow = jl[prow];
26557a48dd6fSHong Zhang 
26567a48dd6fSHong Zhang         /* merge prow into k-th row */
26577a48dd6fSHong Zhang         jmin  = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */
26587a48dd6fSHong Zhang         jmax  = ui[prow+1];
26597a48dd6fSHong Zhang         ncols = jmax-jmin;
2660ed59401aSHong Zhang         i     = jmin - ui[prow];
2661ed59401aSHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
26625a8e39fbSHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
26635a8e39fbSHong Zhang         j     = *(uj - 1);
26645a8e39fbSHong Zhang         ierr  = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
26657a48dd6fSHong Zhang         nzk  += nlnk;
26667a48dd6fSHong Zhang 
26677a48dd6fSHong Zhang         /* update il and jl for prow */
26687a48dd6fSHong Zhang         if (jmin < jmax) {
26697a48dd6fSHong Zhang           il[prow] = jmin;
26707a48dd6fSHong Zhang           j        = *cols; jl[prow] = jl[j]; jl[j] = prow;
26717a48dd6fSHong Zhang         }
26727a48dd6fSHong Zhang         prow = nextprow;
26737a48dd6fSHong Zhang       }
26747a48dd6fSHong Zhang 
26757a48dd6fSHong Zhang       /* if free space is not available, make more free space */
26767a48dd6fSHong Zhang       if (current_space->local_remaining<nzk) {
26777a48dd6fSHong Zhang         i    = am - k + 1; /* num of unfactored rows */
2678f91af8c7SBarry Smith         i    = PetscIntMultTruncate(i,PetscMin(nzk, i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
2679a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
2680a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
26817a48dd6fSHong Zhang         reallocs++;
26827a48dd6fSHong Zhang       }
26837a48dd6fSHong Zhang 
26847a48dd6fSHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
268565e19b50SBarry Smith       if (!nzk) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
26862ed133dbSHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
26877a48dd6fSHong Zhang 
26887a48dd6fSHong Zhang       /* add the k-th row into il and jl */
268939e3d630SHong Zhang       if (nzk > 1) {
26907a48dd6fSHong Zhang         i     = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
26917a48dd6fSHong Zhang         jl[k] = jl[i]; jl[i] = k;
26927a48dd6fSHong Zhang         il[k] = ui[k] + 1;
26937a48dd6fSHong Zhang       }
26947a48dd6fSHong Zhang       uj_ptr[k]     = current_space->array;
26957a48dd6fSHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
26967a48dd6fSHong Zhang 
26977a48dd6fSHong Zhang       current_space->array           += nzk;
26987a48dd6fSHong Zhang       current_space->local_used      += nzk;
26997a48dd6fSHong Zhang       current_space->local_remaining -= nzk;
27007a48dd6fSHong Zhang 
27017a48dd6fSHong Zhang       current_space_lvl->array           += nzk;
27027a48dd6fSHong Zhang       current_space_lvl->local_used      += nzk;
27037a48dd6fSHong Zhang       current_space_lvl->local_remaining -= nzk;
27047a48dd6fSHong Zhang 
27057a48dd6fSHong Zhang       ui[k+1] = ui[k] + nzk;
27067a48dd6fSHong Zhang     }
27077a48dd6fSHong Zhang 
27086cf91177SBarry Smith #if defined(PETSC_USE_INFO)
27097a48dd6fSHong Zhang     if (ai[am] != 0) {
271039e3d630SHong Zhang       PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]);
271157622a8eSBarry Smith       ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)fill,(double)af);CHKERRQ(ierr);
271257622a8eSBarry Smith       ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
271357622a8eSBarry Smith       ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g) for best performance.\n",(double)af);CHKERRQ(ierr);
27147a48dd6fSHong Zhang     } else {
2715dbf6bb8dSprj-       ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
27167a48dd6fSHong Zhang     }
271763ba0a88SBarry Smith #endif
27187a48dd6fSHong Zhang 
27197a48dd6fSHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
2720b635d36bSHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
27210e83c824SBarry Smith     ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr);
27225a8e39fbSHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
27237a48dd6fSHong Zhang 
27247a48dd6fSHong Zhang     /* destroy list of free space and other temporary array(s) */
2725854ce69bSBarry Smith     ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr);
2726a1a86e44SBarry Smith     ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
27272ed133dbSHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
2728a1a86e44SBarry Smith     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
27297a48dd6fSHong Zhang 
273039e3d630SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
273139e3d630SHong Zhang 
27327a48dd6fSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
27337a48dd6fSHong Zhang 
2734f284f12aSHong Zhang   b               = (Mat_SeqSBAIJ*)fact->data;
27357a48dd6fSHong Zhang   b->singlemalloc = PETSC_FALSE;
27362205254eSKarl Rupp 
2737854ce69bSBarry Smith   ierr = PetscMalloc1(ui[am]+1,&b->a);CHKERRQ(ierr);
27382205254eSKarl Rupp 
27397a48dd6fSHong Zhang   b->j         = uj;
27407a48dd6fSHong Zhang   b->i         = ui;
27415f44c854SHong Zhang   b->diag      = udiag;
27427f53bb6cSHong Zhang   b->free_diag = PETSC_TRUE;
27437a48dd6fSHong Zhang   b->ilen      = 0;
27447a48dd6fSHong Zhang   b->imax      = 0;
27457a48dd6fSHong Zhang   b->row       = perm;
2746b635d36bSHong Zhang   b->col       = perm;
27472205254eSKarl Rupp 
2748b635d36bSHong Zhang   ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
2749b635d36bSHong Zhang   ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
27502205254eSKarl Rupp 
2751b635d36bSHong Zhang   b->icol          = iperm;
27527a48dd6fSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
2753854ce69bSBarry Smith   ierr             = PetscMalloc1(am+1,&b->solve_work);CHKERRQ(ierr);
27543bb1ff40SBarry Smith   ierr             = PetscLogObjectMemory((PetscObject)fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
27557a48dd6fSHong Zhang   b->maxnz         = b->nz = ui[am];
2756e6b907acSBarry Smith   b->free_a        = PETSC_TRUE;
2757e6b907acSBarry Smith   b->free_ij       = PETSC_TRUE;
27587a48dd6fSHong Zhang 
2759f284f12aSHong Zhang   fact->info.factor_mallocs   = reallocs;
2760f284f12aSHong Zhang   fact->info.fill_ratio_given = fill;
27617a48dd6fSHong Zhang   if (ai[am] != 0) {
2762f284f12aSHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
27637a48dd6fSHong Zhang   } else {
2764f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
27657a48dd6fSHong Zhang   }
27660a3c351aSHong Zhang   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace;
2767a6175056SHong Zhang   PetscFunctionReturn(0);
2768a6175056SHong Zhang }
2769d5d45c9bSBarry Smith 
277035233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
27710be760fbSHong Zhang {
27720be760fbSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
27730be760fbSHong Zhang   Mat_SeqSBAIJ       *b;
27740be760fbSHong Zhang   PetscErrorCode     ierr;
27759186b105SHong Zhang   PetscBool          perm_identity,missing;
27760be760fbSHong Zhang   PetscReal          fill = info->fill;
27770be760fbSHong Zhang   const PetscInt     *rip,*riip;
27780be760fbSHong Zhang   PetscInt           i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
27790be760fbSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
27800be760fbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr,*udiag;
27810298fd71SBarry Smith   PetscFreeSpaceList free_space=NULL,current_space=NULL;
27820be760fbSHong Zhang   PetscBT            lnkbt;
27830be760fbSHong Zhang   IS                 iperm;
27840be760fbSHong Zhang 
27850be760fbSHong Zhang   PetscFunctionBegin;
2786e32f2f54SBarry 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);
27879186b105SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
27889186b105SHong Zhang   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
27899186b105SHong Zhang 
27900be760fbSHong Zhang   /* check whether perm is the identity mapping */
27910be760fbSHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
27920be760fbSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
27930be760fbSHong Zhang   ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
27940be760fbSHong Zhang   ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
27950be760fbSHong Zhang 
27960be760fbSHong Zhang   /* initialization */
2797854ce69bSBarry Smith   ierr  = PetscMalloc1(am+1,&ui);CHKERRQ(ierr);
2798854ce69bSBarry Smith   ierr  = PetscMalloc1(am+1,&udiag);CHKERRQ(ierr);
27990be760fbSHong Zhang   ui[0] = 0;
28000be760fbSHong Zhang 
28010be760fbSHong Zhang   /* jl: linked list for storing indices of the pivot rows
28020be760fbSHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
2803dcca6d9dSJed Brown   ierr = PetscMalloc4(am,&ui_ptr,am,&jl,am,&il,am,&cols);CHKERRQ(ierr);
28040be760fbSHong Zhang   for (i=0; i<am; i++) {
28050be760fbSHong Zhang     jl[i] = am; il[i] = 0;
28060be760fbSHong Zhang   }
28070be760fbSHong Zhang 
28080be760fbSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
28090be760fbSHong Zhang   nlnk = am + 1;
28100be760fbSHong Zhang   ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28110be760fbSHong Zhang 
281295b5ac22SHong Zhang   /* initial FreeSpace size is fill*(ai[am]+am)/2 */
28132f18eb33SBarry Smith   ierr          = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,(ai[am]+am)/2),&free_space);CHKERRQ(ierr);
28140be760fbSHong Zhang   current_space = free_space;
28150be760fbSHong Zhang 
28160be760fbSHong Zhang   for (k=0; k<am; k++) {  /* for each active row k */
28170be760fbSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
28180be760fbSHong Zhang     nzk   = 0;
28190be760fbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
2820e32f2f54SBarry 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);
28210be760fbSHong Zhang     ncols_upper = 0;
28220be760fbSHong Zhang     for (j=0; j<ncols; j++) {
28230be760fbSHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
28240be760fbSHong Zhang       if (i >= k) { /* only take upper triangular entry */
28250be760fbSHong Zhang         cols[ncols_upper] = i;
28260be760fbSHong Zhang         ncols_upper++;
28270be760fbSHong Zhang       }
28280be760fbSHong Zhang     }
28290be760fbSHong Zhang     ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28300be760fbSHong Zhang     nzk += nlnk;
28310be760fbSHong Zhang 
28320be760fbSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
28330be760fbSHong Zhang     prow = jl[k]; /* 1st pivot row */
28340be760fbSHong Zhang 
28350be760fbSHong Zhang     while (prow < k) {
28360be760fbSHong Zhang       nextprow = jl[prow];
28370be760fbSHong Zhang       /* merge prow into k-th row */
28380be760fbSHong Zhang       jmin   = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */
28390be760fbSHong Zhang       jmax   = ui[prow+1];
28400be760fbSHong Zhang       ncols  = jmax-jmin;
28410be760fbSHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
28420be760fbSHong Zhang       ierr   = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28430be760fbSHong Zhang       nzk   += nlnk;
28440be760fbSHong Zhang 
28450be760fbSHong Zhang       /* update il and jl for prow */
28460be760fbSHong Zhang       if (jmin < jmax) {
28470be760fbSHong Zhang         il[prow] = jmin;
28482205254eSKarl Rupp         j        = *uj_ptr;
28492205254eSKarl Rupp         jl[prow] = jl[j];
28502205254eSKarl Rupp         jl[j]    = prow;
28510be760fbSHong Zhang       }
28520be760fbSHong Zhang       prow = nextprow;
28530be760fbSHong Zhang     }
28540be760fbSHong Zhang 
28550be760fbSHong Zhang     /* if free space is not available, make more free space */
28560be760fbSHong Zhang     if (current_space->local_remaining<nzk) {
28570be760fbSHong Zhang       i    = am - k + 1; /* num of unfactored rows */
2858f91af8c7SBarry Smith       i    = PetscIntMultTruncate(i,PetscMin(nzk,i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
28590be760fbSHong Zhang       ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
28600be760fbSHong Zhang       reallocs++;
28610be760fbSHong Zhang     }
28620be760fbSHong Zhang 
28630be760fbSHong Zhang     /* copy data into free space, then initialize lnk */
28640be760fbSHong Zhang     ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
28650be760fbSHong Zhang 
28660be760fbSHong Zhang     /* add the k-th row into il and jl */
28677b056e98SHong Zhang     if (nzk > 1) {
28680be760fbSHong Zhang       i     = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
28690be760fbSHong Zhang       jl[k] = jl[i]; jl[i] = k;
28700be760fbSHong Zhang       il[k] = ui[k] + 1;
28710be760fbSHong Zhang     }
28720be760fbSHong Zhang     ui_ptr[k] = current_space->array;
28732205254eSKarl Rupp 
28740be760fbSHong Zhang     current_space->array           += nzk;
28750be760fbSHong Zhang     current_space->local_used      += nzk;
28760be760fbSHong Zhang     current_space->local_remaining -= nzk;
28770be760fbSHong Zhang 
28780be760fbSHong Zhang     ui[k+1] = ui[k] + nzk;
28790be760fbSHong Zhang   }
28800be760fbSHong Zhang 
28810be760fbSHong Zhang   ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
28820be760fbSHong Zhang   ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
28830be760fbSHong Zhang   ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr);
28840be760fbSHong Zhang 
28859263d837SHong Zhang   /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */
2886854ce69bSBarry Smith   ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr);
288735233d99SShri Abhyankar   ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor */
28880be760fbSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
28890be760fbSHong Zhang 
28900be760fbSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
28910be760fbSHong Zhang 
2892f284f12aSHong Zhang   b               = (Mat_SeqSBAIJ*)fact->data;
28930be760fbSHong Zhang   b->singlemalloc = PETSC_FALSE;
28940be760fbSHong Zhang   b->free_a       = PETSC_TRUE;
28950be760fbSHong Zhang   b->free_ij      = PETSC_TRUE;
28962205254eSKarl Rupp 
2897854ce69bSBarry Smith   ierr = PetscMalloc1(ui[am]+1,&b->a);CHKERRQ(ierr);
28982205254eSKarl Rupp 
28990be760fbSHong Zhang   b->j         = uj;
29000be760fbSHong Zhang   b->i         = ui;
29010be760fbSHong Zhang   b->diag      = udiag;
29020be760fbSHong Zhang   b->free_diag = PETSC_TRUE;
29030be760fbSHong Zhang   b->ilen      = 0;
29040be760fbSHong Zhang   b->imax      = 0;
29050be760fbSHong Zhang   b->row       = perm;
29060be760fbSHong Zhang   b->col       = perm;
290726fbe8dcSKarl Rupp 
29080be760fbSHong Zhang   ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
29090be760fbSHong Zhang   ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
29102205254eSKarl Rupp 
29110be760fbSHong Zhang   b->icol          = iperm;
29120be760fbSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
291326fbe8dcSKarl Rupp 
2914854ce69bSBarry Smith   ierr = PetscMalloc1(am+1,&b->solve_work);CHKERRQ(ierr);
29153bb1ff40SBarry Smith   ierr = PetscLogObjectMemory((PetscObject)fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
29162205254eSKarl Rupp 
29170be760fbSHong Zhang   b->maxnz = b->nz = ui[am];
29180be760fbSHong Zhang 
2919f284f12aSHong Zhang   fact->info.factor_mallocs   = reallocs;
2920f284f12aSHong Zhang   fact->info.fill_ratio_given = fill;
29210be760fbSHong Zhang   if (ai[am] != 0) {
29226a69fef8SHong Zhang     /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */
292395b5ac22SHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am);
29240be760fbSHong Zhang   } else {
2925f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
29260be760fbSHong Zhang   }
29279263d837SHong Zhang #if defined(PETSC_USE_INFO)
29289263d837SHong Zhang   if (ai[am] != 0) {
29299263d837SHong Zhang     PetscReal af = fact->info.fill_ratio_needed;
293057622a8eSBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)fill,(double)af);CHKERRQ(ierr);
293157622a8eSBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
293257622a8eSBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g) for best performance.\n",(double)af);CHKERRQ(ierr);
29339263d837SHong Zhang   } else {
2934dbf6bb8dSprj-     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
29359263d837SHong Zhang   }
29369263d837SHong Zhang #endif
293735233d99SShri Abhyankar   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
29380be760fbSHong Zhang   PetscFunctionReturn(0);
29390be760fbSHong Zhang }
29400be760fbSHong Zhang 
2941ad04f41aSHong Zhang PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2942f76d2b81SHong Zhang {
2943f76d2b81SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
294410c27e3dSHong Zhang   Mat_SeqSBAIJ       *b;
2945dfbe8321SBarry Smith   PetscErrorCode     ierr;
29469186b105SHong Zhang   PetscBool          perm_identity,missing;
294710c27e3dSHong Zhang   PetscReal          fill = info->fill;
29485d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
29495d0c19d7SBarry Smith   PetscInt           i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
295010c27e3dSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
29512ed133dbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr;
29520298fd71SBarry Smith   PetscFreeSpaceList free_space=NULL,current_space=NULL;
295310c27e3dSHong Zhang   PetscBT            lnkbt;
29542ed133dbSHong Zhang   IS                 iperm;
2955f76d2b81SHong Zhang 
2956f76d2b81SHong Zhang   PetscFunctionBegin;
2957e32f2f54SBarry 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);
29589186b105SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
29599186b105SHong Zhang   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
29609186b105SHong Zhang 
29612ed133dbSHong Zhang   /* check whether perm is the identity mapping */
2962f76d2b81SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
29632ed133dbSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
29642ed133dbSHong Zhang   ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
29659bfd6278SHong Zhang   ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
296610c27e3dSHong Zhang 
296710c27e3dSHong Zhang   /* initialization */
2968854ce69bSBarry Smith   ierr  = PetscMalloc1(am+1,&ui);CHKERRQ(ierr);
296910c27e3dSHong Zhang   ui[0] = 0;
297010c27e3dSHong Zhang 
297110c27e3dSHong Zhang   /* jl: linked list for storing indices of the pivot rows
29721393bc97SHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
2973dcca6d9dSJed Brown   ierr = PetscMalloc4(am,&ui_ptr,am,&jl,am,&il,am,&cols);CHKERRQ(ierr);
29741393bc97SHong Zhang   for (i=0; i<am; i++) {
29751393bc97SHong Zhang     jl[i] = am; il[i] = 0;
2976f76d2b81SHong Zhang   }
297710c27e3dSHong Zhang 
297810c27e3dSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
29791393bc97SHong Zhang   nlnk = am + 1;
29801393bc97SHong Zhang   ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
298110c27e3dSHong Zhang 
29821393bc97SHong Zhang   /* initial FreeSpace size is fill*(ai[am]+1) */
29832f18eb33SBarry Smith   ierr          = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,ai[am]+1),&free_space);CHKERRQ(ierr);
298410c27e3dSHong Zhang   current_space = free_space;
298510c27e3dSHong Zhang 
29861393bc97SHong Zhang   for (k=0; k<am; k++) {  /* for each active row k */
298710c27e3dSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
298810c27e3dSHong Zhang     nzk   = 0;
29892ed133dbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
2990e32f2f54SBarry 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);
29912ed133dbSHong Zhang     ncols_upper = 0;
29922ed133dbSHong Zhang     for (j=0; j<ncols; j++) {
29939bfd6278SHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
29942ed133dbSHong Zhang       if (i >= k) { /* only take upper triangular entry */
29952ed133dbSHong Zhang         cols[ncols_upper] = i;
29962ed133dbSHong Zhang         ncols_upper++;
29972ed133dbSHong Zhang       }
29982ed133dbSHong Zhang     }
29991393bc97SHong Zhang     ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
300010c27e3dSHong Zhang     nzk += nlnk;
300110c27e3dSHong Zhang 
300210c27e3dSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
300310c27e3dSHong Zhang     prow = jl[k]; /* 1st pivot row */
300410c27e3dSHong Zhang 
300510c27e3dSHong Zhang     while (prow < k) {
300610c27e3dSHong Zhang       nextprow = jl[prow];
300710c27e3dSHong Zhang       /* merge prow into k-th row */
30081393bc97SHong Zhang       jmin   = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */
300910c27e3dSHong Zhang       jmax   = ui[prow+1];
301010c27e3dSHong Zhang       ncols  = jmax-jmin;
30111393bc97SHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
30125a8e39fbSHong Zhang       ierr   = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
301310c27e3dSHong Zhang       nzk   += nlnk;
301410c27e3dSHong Zhang 
301510c27e3dSHong Zhang       /* update il and jl for prow */
301610c27e3dSHong Zhang       if (jmin < jmax) {
301710c27e3dSHong Zhang         il[prow] = jmin;
30182ed133dbSHong Zhang         j        = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow;
301910c27e3dSHong Zhang       }
302010c27e3dSHong Zhang       prow = nextprow;
302110c27e3dSHong Zhang     }
302210c27e3dSHong Zhang 
302310c27e3dSHong Zhang     /* if free space is not available, make more free space */
302410c27e3dSHong Zhang     if (current_space->local_remaining<nzk) {
30251393bc97SHong Zhang       i    = am - k + 1; /* num of unfactored rows */
302610c27e3dSHong Zhang       i    = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
3027a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
302810c27e3dSHong Zhang       reallocs++;
302910c27e3dSHong Zhang     }
303010c27e3dSHong Zhang 
303110c27e3dSHong Zhang     /* copy data into free space, then initialize lnk */
30321393bc97SHong Zhang     ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
303310c27e3dSHong Zhang 
303410c27e3dSHong Zhang     /* add the k-th row into il and jl */
303510c27e3dSHong Zhang     if (nzk-1 > 0) {
30361393bc97SHong Zhang       i     = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
303710c27e3dSHong Zhang       jl[k] = jl[i]; jl[i] = k;
303810c27e3dSHong Zhang       il[k] = ui[k] + 1;
303910c27e3dSHong Zhang     }
30402ed133dbSHong Zhang     ui_ptr[k] = current_space->array;
30412205254eSKarl Rupp 
304210c27e3dSHong Zhang     current_space->array           += nzk;
304310c27e3dSHong Zhang     current_space->local_used      += nzk;
304410c27e3dSHong Zhang     current_space->local_remaining -= nzk;
304510c27e3dSHong Zhang 
304610c27e3dSHong Zhang     ui[k+1] = ui[k] + nzk;
304710c27e3dSHong Zhang   }
304810c27e3dSHong Zhang 
30496cf91177SBarry Smith #if defined(PETSC_USE_INFO)
30501393bc97SHong Zhang   if (ai[am] != 0) {
30511393bc97SHong Zhang     PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]);
305257622a8eSBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)fill,(double)af);CHKERRQ(ierr);
305357622a8eSBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
305457622a8eSBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g) for best performance.\n",(double)af);CHKERRQ(ierr);
305510c27e3dSHong Zhang   } else {
3056dbf6bb8dSprj-     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
305710c27e3dSHong Zhang   }
305863ba0a88SBarry Smith #endif
305910c27e3dSHong Zhang 
306010c27e3dSHong Zhang   ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
30619bfd6278SHong Zhang   ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
30620e83c824SBarry Smith   ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr);
306310c27e3dSHong Zhang 
306410c27e3dSHong Zhang   /* destroy list of free space and other temporary array(s) */
3065854ce69bSBarry Smith   ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr);
3066a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
306710c27e3dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
306810c27e3dSHong Zhang 
306910c27e3dSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
307010c27e3dSHong Zhang 
3071f284f12aSHong Zhang   b               = (Mat_SeqSBAIJ*)fact->data;
307210c27e3dSHong Zhang   b->singlemalloc = PETSC_FALSE;
3073e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
3074e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
30752205254eSKarl Rupp 
3076854ce69bSBarry Smith   ierr = PetscMalloc1(ui[am]+1,&b->a);CHKERRQ(ierr);
30772205254eSKarl Rupp 
307810c27e3dSHong Zhang   b->j    = uj;
307910c27e3dSHong Zhang   b->i    = ui;
308010c27e3dSHong Zhang   b->diag = 0;
308110c27e3dSHong Zhang   b->ilen = 0;
308210c27e3dSHong Zhang   b->imax = 0;
308310c27e3dSHong Zhang   b->row  = perm;
30849bfd6278SHong Zhang   b->col  = perm;
30852205254eSKarl Rupp 
30869bfd6278SHong Zhang   ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
30879bfd6278SHong Zhang   ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
30882205254eSKarl Rupp 
30899bfd6278SHong Zhang   b->icol          = iperm;
309010c27e3dSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
30912205254eSKarl Rupp 
3092854ce69bSBarry Smith   ierr     = PetscMalloc1(am+1,&b->solve_work);CHKERRQ(ierr);
30933bb1ff40SBarry Smith   ierr     = PetscLogObjectMemory((PetscObject)fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
30941393bc97SHong Zhang   b->maxnz = b->nz = ui[am];
309510c27e3dSHong Zhang 
3096f284f12aSHong Zhang   fact->info.factor_mallocs   = reallocs;
3097f284f12aSHong Zhang   fact->info.fill_ratio_given = fill;
30981393bc97SHong Zhang   if (ai[am] != 0) {
3099f284f12aSHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
310010c27e3dSHong Zhang   } else {
3101f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
310210c27e3dSHong Zhang   }
31030a3c351aSHong Zhang   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace;
3104f76d2b81SHong Zhang   PetscFunctionReturn(0);
3105f76d2b81SHong Zhang }
3106599ef60dSHong Zhang 
310735233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx)
3108813a1f2bSShri Abhyankar {
3109813a1f2bSShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
3110813a1f2bSShri Abhyankar   PetscErrorCode    ierr;
3111813a1f2bSShri Abhyankar   PetscInt          n   = A->rmap->n;
3112813a1f2bSShri Abhyankar   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag,*vi;
3113813a1f2bSShri Abhyankar   PetscScalar       *x,sum;
3114813a1f2bSShri Abhyankar   const PetscScalar *b;
3115813a1f2bSShri Abhyankar   const MatScalar   *aa = a->a,*v;
3116813a1f2bSShri Abhyankar   PetscInt          i,nz;
3117813a1f2bSShri Abhyankar 
3118813a1f2bSShri Abhyankar   PetscFunctionBegin;
3119813a1f2bSShri Abhyankar   if (!n) PetscFunctionReturn(0);
3120813a1f2bSShri Abhyankar 
31213649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
3122fdc842d1SBarry Smith   ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr);
3123813a1f2bSShri Abhyankar 
3124813a1f2bSShri Abhyankar   /* forward solve the lower triangular */
3125813a1f2bSShri Abhyankar   x[0] = b[0];
3126813a1f2bSShri Abhyankar   v    = aa;
3127813a1f2bSShri Abhyankar   vi   = aj;
3128813a1f2bSShri Abhyankar   for (i=1; i<n; i++) {
3129813a1f2bSShri Abhyankar     nz  = ai[i+1] - ai[i];
3130813a1f2bSShri Abhyankar     sum = b[i];
3131813a1f2bSShri Abhyankar     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
3132813a1f2bSShri Abhyankar     v   += nz;
3133813a1f2bSShri Abhyankar     vi  += nz;
3134813a1f2bSShri Abhyankar     x[i] = sum;
3135813a1f2bSShri Abhyankar   }
3136813a1f2bSShri Abhyankar 
3137813a1f2bSShri Abhyankar   /* backward solve the upper triangular */
313862fbd6afSShri Abhyankar   for (i=n-1; i>=0; i--) {
31393c0119dfSShri Abhyankar     v   = aa + adiag[i+1] + 1;
31403c0119dfSShri Abhyankar     vi  = aj + adiag[i+1] + 1;
3141813a1f2bSShri Abhyankar     nz  = adiag[i] - adiag[i+1]-1;
3142813a1f2bSShri Abhyankar     sum = x[i];
3143813a1f2bSShri Abhyankar     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
31443c0119dfSShri Abhyankar     x[i] = sum*v[nz]; /* x[i]=aa[adiag[i]]*sum; v++; */
3145813a1f2bSShri Abhyankar   }
3146813a1f2bSShri Abhyankar 
3147813a1f2bSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
31483649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
3149fdc842d1SBarry Smith   ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr);
3150813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
3151813a1f2bSShri Abhyankar }
3152813a1f2bSShri Abhyankar 
315335233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx)
3154f268cba8SShri Abhyankar {
3155f268cba8SShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
3156f268cba8SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
3157f268cba8SShri Abhyankar   PetscErrorCode    ierr;
3158f268cba8SShri Abhyankar   PetscInt          i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,*adiag = a->diag,nz;
3159f268cba8SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c;
3160f268cba8SShri Abhyankar   PetscScalar       *x,*tmp,sum;
3161f268cba8SShri Abhyankar   const PetscScalar *b;
3162f268cba8SShri Abhyankar   const MatScalar   *aa = a->a,*v;
3163f268cba8SShri Abhyankar 
3164f268cba8SShri Abhyankar   PetscFunctionBegin;
3165f268cba8SShri Abhyankar   if (!n) PetscFunctionReturn(0);
3166f268cba8SShri Abhyankar 
31673649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
3168fdc842d1SBarry Smith   ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr);
3169f268cba8SShri Abhyankar   tmp  = a->solve_work;
3170f268cba8SShri Abhyankar 
3171f268cba8SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
3172f268cba8SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
3173f268cba8SShri Abhyankar 
3174f268cba8SShri Abhyankar   /* forward solve the lower triangular */
3175f268cba8SShri Abhyankar   tmp[0] = b[r[0]];
3176f268cba8SShri Abhyankar   v      = aa;
3177f268cba8SShri Abhyankar   vi     = aj;
3178f268cba8SShri Abhyankar   for (i=1; i<n; i++) {
3179f268cba8SShri Abhyankar     nz  = ai[i+1] - ai[i];
3180f268cba8SShri Abhyankar     sum = b[r[i]];
3181f268cba8SShri Abhyankar     PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
3182f268cba8SShri Abhyankar     tmp[i] = sum;
3183f268cba8SShri Abhyankar     v     += nz; vi += nz;
3184f268cba8SShri Abhyankar   }
3185f268cba8SShri Abhyankar 
3186f268cba8SShri Abhyankar   /* backward solve the upper triangular */
3187f268cba8SShri Abhyankar   for (i=n-1; i>=0; i--) {
31883c0119dfSShri Abhyankar     v   = aa + adiag[i+1]+1;
31893c0119dfSShri Abhyankar     vi  = aj + adiag[i+1]+1;
3190f268cba8SShri Abhyankar     nz  = adiag[i]-adiag[i+1]-1;
3191f268cba8SShri Abhyankar     sum = tmp[i];
3192f268cba8SShri Abhyankar     PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
3193f268cba8SShri Abhyankar     x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */
3194f268cba8SShri Abhyankar   }
3195f268cba8SShri Abhyankar 
3196f268cba8SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
3197f268cba8SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
31983649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
3199fdc842d1SBarry Smith   ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr);
3200f268cba8SShri Abhyankar   ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr);
3201f268cba8SShri Abhyankar   PetscFunctionReturn(0);
3202f268cba8SShri Abhyankar }
3203f268cba8SShri Abhyankar 
3204fe97e370SBarry Smith /*
3205bebe2cf6SSatish 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
3206fe97e370SBarry Smith */
32071d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact)
32081d098868SHong Zhang {
32091d098868SHong Zhang   Mat            B = *fact;
3210599ef60dSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*b;
3211599ef60dSHong Zhang   IS             isicol;
3212599ef60dSHong Zhang   PetscErrorCode ierr;
32131d098868SHong Zhang   const PetscInt *r,*ic;
32141d098868SHong Zhang   PetscInt       i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag;
3215dc3a2fd3SHong Zhang   PetscInt       *bi,*bj,*bdiag,*bdiag_rev;
3216f61a948aSLisandro Dalcin   PetscInt       row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au;
32171d098868SHong Zhang   PetscInt       nlnk,*lnk;
32181d098868SHong Zhang   PetscBT        lnkbt;
3219ace3abfcSBarry Smith   PetscBool      row_identity,icol_identity;
32208aee2decSHong Zhang   MatScalar      *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp;
32211d098868SHong Zhang   const PetscInt *ics;
32221d098868SHong Zhang   PetscInt       j,nz,*pj,*bjtmp,k,ncut,*jtmp;
3223a2ea699eSBarry Smith   PetscReal      dt     =info->dt,shift=info->shiftamount;
3224f61a948aSLisandro Dalcin   PetscInt       dtcount=(PetscInt)info->dtcount,nnz_max;
3225ace3abfcSBarry Smith   PetscBool      missing;
3226599ef60dSHong Zhang 
3227599ef60dSHong Zhang   PetscFunctionBegin;
3228f61a948aSLisandro Dalcin   if (dt      == PETSC_DEFAULT) dt = 0.005;
3229f61a948aSLisandro Dalcin   if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax);
3230f61a948aSLisandro Dalcin 
32311d098868SHong Zhang   /* ------- symbolic factorization, can be reused ---------*/
32321d098868SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
3233e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
32341d098868SHong Zhang   adiag=a->diag;
3235599ef60dSHong Zhang 
3236599ef60dSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
3237599ef60dSHong Zhang 
3238599ef60dSHong Zhang   /* bdiag is location of diagonal in factor */
3239854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&bdiag);CHKERRQ(ierr);     /* becomes b->diag */
3240854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&bdiag_rev);CHKERRQ(ierr); /* temporary */
3241599ef60dSHong Zhang 
32421d098868SHong Zhang   /* allocate row pointers bi */
3243854ce69bSBarry Smith   ierr = PetscMalloc1(2*n+2,&bi);CHKERRQ(ierr);
32441d098868SHong Zhang 
32451d098868SHong Zhang   /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */
3246393d3a68SHong Zhang   if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */
32471d098868SHong Zhang   nnz_max = ai[n]+2*n*dtcount+2;
32488fc3a347SHong Zhang 
3249854ce69bSBarry Smith   ierr = PetscMalloc1(nnz_max+1,&bj);CHKERRQ(ierr);
3250854ce69bSBarry Smith   ierr = PetscMalloc1(nnz_max+1,&ba);CHKERRQ(ierr);
3251599ef60dSHong Zhang 
3252599ef60dSHong Zhang   /* put together the new matrix */
32530298fd71SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
32543bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)B,(PetscObject)isicol);CHKERRQ(ierr);
3255f284f12aSHong Zhang   b    = (Mat_SeqAIJ*)B->data;
32562205254eSKarl Rupp 
3257599ef60dSHong Zhang   b->free_a       = PETSC_TRUE;
3258599ef60dSHong Zhang   b->free_ij      = PETSC_TRUE;
3259599ef60dSHong Zhang   b->singlemalloc = PETSC_FALSE;
32602205254eSKarl Rupp 
3261599ef60dSHong Zhang   b->a    = ba;
3262599ef60dSHong Zhang   b->j    = bj;
3263599ef60dSHong Zhang   b->i    = bi;
3264599ef60dSHong Zhang   b->diag = bdiag;
3265599ef60dSHong Zhang   b->ilen = 0;
3266599ef60dSHong Zhang   b->imax = 0;
3267599ef60dSHong Zhang   b->row  = isrow;
3268599ef60dSHong Zhang   b->col  = iscol;
3269599ef60dSHong Zhang   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
3270599ef60dSHong Zhang   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
3271599ef60dSHong Zhang   b->icol = isicol;
3272599ef60dSHong Zhang 
3273854ce69bSBarry Smith   ierr     = PetscMalloc1(n+1,&b->solve_work);CHKERRQ(ierr);
32743bb1ff40SBarry Smith   ierr     = PetscLogObjectMemory((PetscObject)B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
32751d098868SHong Zhang   b->maxnz = nnz_max;
3276599ef60dSHong Zhang 
3277d5f3da31SBarry Smith   B->factortype            = MAT_FACTOR_ILUDT;
3278f284f12aSHong Zhang   B->info.factor_mallocs   = 0;
3279f284f12aSHong Zhang   B->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]);
32801d098868SHong Zhang   /* ------- end of symbolic factorization ---------*/
3281599ef60dSHong Zhang 
3282599ef60dSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
3283599ef60dSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
3284599ef60dSHong Zhang   ics  = ic;
3285599ef60dSHong Zhang 
3286599ef60dSHong Zhang   /* linked list for storing column indices of the active row */
3287599ef60dSHong Zhang   nlnk = n + 1;
3288599ef60dSHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
32891d098868SHong Zhang 
32901d098868SHong Zhang   /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */
3291dcca6d9dSJed Brown   ierr = PetscMalloc2(n,&im,n,&jtmp);CHKERRQ(ierr);
32921d098868SHong Zhang   /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */
3293dcca6d9dSJed Brown   ierr = PetscMalloc2(n,&rtmp,n,&vtmp);CHKERRQ(ierr);
3294580bdb30SBarry Smith   ierr = PetscArrayzero(rtmp,n);CHKERRQ(ierr);
3295599ef60dSHong Zhang 
3296599ef60dSHong Zhang   bi[0]        = 0;
32978fc3a347SHong Zhang   bdiag[0]     = nnz_max-1; /* location of diag[0] in factor B */
3298dc3a2fd3SHong Zhang   bdiag_rev[n] = bdiag[0];
32998fc3a347SHong Zhang   bi[2*n+1]    = bdiag[0]+1; /* endof bj and ba array */
3300599ef60dSHong Zhang   for (i=0; i<n; i++) {
3301599ef60dSHong Zhang     /* copy initial fill into linked list */
33028369b28dSHong Zhang     nzi = ai[r[i]+1] - ai[r[i]];
3303e32f2f54SBarry 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);
33048369b28dSHong Zhang     nzi_al = adiag[r[i]] - ai[r[i]];
33058369b28dSHong Zhang     nzi_au = ai[r[i]+1] - adiag[r[i]] -1;
3306599ef60dSHong Zhang     ajtmp  = aj + ai[r[i]];
33078369b28dSHong Zhang     ierr   = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3308599ef60dSHong Zhang 
3309599ef60dSHong Zhang     /* load in initial (unfactored row) */
33101d098868SHong Zhang     aatmp = a->a + ai[r[i]];
33118369b28dSHong Zhang     for (j=0; j<nzi; j++) {
33121d098868SHong Zhang       rtmp[ics[*ajtmp++]] = *aatmp++;
3313599ef60dSHong Zhang     }
3314599ef60dSHong Zhang 
3315599ef60dSHong Zhang     /* add pivot rows into linked list */
3316599ef60dSHong Zhang     row = lnk[n];
3317599ef60dSHong Zhang     while (row < i) {
33181d098868SHong Zhang       nzi_bl = bi[row+1] - bi[row] + 1;
33191d098868SHong Zhang       bjtmp  = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */
33201d098868SHong Zhang       ierr   = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr);
3321599ef60dSHong Zhang       nzi   += nlnk;
3322599ef60dSHong Zhang       row    = lnk[row];
3323599ef60dSHong Zhang     }
3324599ef60dSHong Zhang 
33258369b28dSHong Zhang     /* copy data from lnk into jtmp, then initialize lnk */
33268369b28dSHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr);
3327599ef60dSHong Zhang 
3328599ef60dSHong Zhang     /* numerical factorization */
33298369b28dSHong Zhang     bjtmp = jtmp;
3330599ef60dSHong Zhang     row   = *bjtmp++; /* 1st pivot row */
3331599ef60dSHong Zhang     while (row < i) {
3332599ef60dSHong Zhang       pc         = rtmp + row;
33333c2a7987SHong Zhang       pv         = ba + bdiag[row]; /* 1./(diag of the pivot row) */
33343c2a7987SHong Zhang       multiplier = (*pc) * (*pv);
3335599ef60dSHong Zhang       *pc        = multiplier;
33363c2a7987SHong Zhang       if (PetscAbsScalar(*pc) > dt) { /* apply tolerance dropping rule */
33371d098868SHong Zhang         pj = bj + bdiag[row+1] + 1;         /* point to 1st entry of U(row,:) */
33381d098868SHong Zhang         pv = ba + bdiag[row+1] + 1;
33391d098868SHong Zhang         /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */
33401d098868SHong Zhang         nz = bdiag[row] - bdiag[row+1] - 1;         /* num of entries in U(row,:), excluding diagonal */
33411d098868SHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
334228b1a77fSLisandro Dalcin         ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
3343599ef60dSHong Zhang       }
3344599ef60dSHong Zhang       row = *bjtmp++;
3345599ef60dSHong Zhang     }
3346599ef60dSHong Zhang 
33478369b28dSHong Zhang     /* copy sparse rtmp into contiguous vtmp; separate L and U part */
33488aee2decSHong Zhang     diag_tmp = rtmp[i];  /* save diagonal value - may not needed?? */
33498369b28dSHong Zhang     nzi_bl   = 0; j = 0;
33508369b28dSHong Zhang     while (jtmp[j] < i) { /* Note: jtmp is sorted */
33518aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
33528369b28dSHong Zhang       nzi_bl++; j++;
33538369b28dSHong Zhang     }
33548369b28dSHong Zhang     nzi_bu = nzi - nzi_bl -1;
33558369b28dSHong Zhang     while (j < nzi) {
33568aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
33578369b28dSHong Zhang       j++;
33588369b28dSHong Zhang     }
33598369b28dSHong Zhang 
3360599ef60dSHong Zhang     bjtmp = bj + bi[i];
3361599ef60dSHong Zhang     batmp = ba + bi[i];
33628369b28dSHong Zhang     /* apply level dropping rule to L part */
33638369b28dSHong Zhang     ncut = nzi_al + dtcount;
33648369b28dSHong Zhang     if (ncut < nzi_bl) {
33658369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr);
33668369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr);
3367599ef60dSHong Zhang     } else {
33688369b28dSHong Zhang       ncut = nzi_bl;
33698369b28dSHong Zhang     }
33708369b28dSHong Zhang     for (j=0; j<ncut; j++) {
33718369b28dSHong Zhang       bjtmp[j] = jtmp[j];
33728369b28dSHong Zhang       batmp[j] = vtmp[j];
33736da515a1SHong Zhang       /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */
33748369b28dSHong Zhang     }
33751d098868SHong Zhang     bi[i+1] = bi[i] + ncut;
33768369b28dSHong Zhang     nzi     = ncut + 1;
33778369b28dSHong Zhang 
33788369b28dSHong Zhang     /* apply level dropping rule to U part */
33798369b28dSHong Zhang     ncut = nzi_au + dtcount;
33808369b28dSHong Zhang     if (ncut < nzi_bu) {
33818369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr);
33828369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr);
33838369b28dSHong Zhang     } else {
33848369b28dSHong Zhang       ncut = nzi_bu;
33858369b28dSHong Zhang     }
33868369b28dSHong Zhang     nzi += ncut;
33871d098868SHong Zhang 
33881d098868SHong Zhang     /* mark bdiagonal */
33891d098868SHong Zhang     bdiag[i+1]       = bdiag[i] - (ncut + 1);
3390dc3a2fd3SHong Zhang     bdiag_rev[n-i-1] = bdiag[i+1];
33918fc3a347SHong Zhang     bi[2*n - i]      = bi[2*n - i +1] - (ncut + 1);
33921d098868SHong Zhang     bjtmp            = bj + bdiag[i];
33931d098868SHong Zhang     batmp            = ba + bdiag[i];
33941d098868SHong Zhang     *bjtmp           = i;
33958aee2decSHong Zhang     *batmp           = diag_tmp; /* rtmp[i]; */
3396c9c72f8fSHong Zhang     if (*batmp == 0.0) {
3397c9c72f8fSHong Zhang       *batmp = dt+shift;
3398bd1bc851SBarry Smith       /* printf(" row %d add shift %g\n",i,shift); */
3399c9c72f8fSHong Zhang     }
3400b78a477dSHong Zhang     *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */
34016da515a1SHong Zhang     /* printf(" (%d,%g),",*bjtmp,*batmp); */
34021d098868SHong Zhang 
34031d098868SHong Zhang     bjtmp = bj + bdiag[i+1]+1;
34041d098868SHong Zhang     batmp = ba + bdiag[i+1]+1;
34058369b28dSHong Zhang     for (k=0; k<ncut; k++) {
34061d098868SHong Zhang       bjtmp[k] = jtmp[nzi_bl+1+k];
34071d098868SHong Zhang       batmp[k] = vtmp[nzi_bl+1+k];
34086da515a1SHong Zhang       /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */
3409599ef60dSHong Zhang     }
34106da515a1SHong Zhang     /* printf("\n"); */
3411599ef60dSHong Zhang 
3412599ef60dSHong Zhang     im[i] = nzi;   /* used by PetscLLAddSortedLU() */
3413599ef60dSHong Zhang     /*
34141d098868SHong Zhang     printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]);
34158369b28dSHong Zhang     printf(" ----------------------------\n");
3416599ef60dSHong Zhang     */
34178369b28dSHong Zhang   } /* for (i=0; i<n; i++) */
34181d098868SHong Zhang     /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */
3419e32f2f54SBarry 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]);
3420599ef60dSHong Zhang 
3421599ef60dSHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3422599ef60dSHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3423599ef60dSHong Zhang 
3424599ef60dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
34250e83c824SBarry Smith   ierr = PetscFree2(im,jtmp);CHKERRQ(ierr);
34260e83c824SBarry Smith   ierr = PetscFree2(rtmp,vtmp);CHKERRQ(ierr);
34271bfa06eaSJed Brown   ierr = PetscFree(bdiag_rev);CHKERRQ(ierr);
3428599ef60dSHong Zhang 
3429599ef60dSHong Zhang   ierr     = PetscLogFlops(B->cmap->n);CHKERRQ(ierr);
34301d098868SHong Zhang   b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n];
3431599ef60dSHong Zhang 
3432599ef60dSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
3433599ef60dSHong Zhang   ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr);
3434599ef60dSHong Zhang   if (row_identity && icol_identity) {
343535233d99SShri Abhyankar     B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
3436599ef60dSHong Zhang   } else {
343735233d99SShri Abhyankar     B->ops->solve = MatSolve_SeqAIJ;
3438599ef60dSHong Zhang   }
3439599ef60dSHong Zhang 
34401d098868SHong Zhang   B->ops->solveadd          = 0;
34411d098868SHong Zhang   B->ops->solvetranspose    = 0;
34421d098868SHong Zhang   B->ops->solvetransposeadd = 0;
34431d098868SHong Zhang   B->ops->matsolve          = 0;
3444599ef60dSHong Zhang   B->assembled              = PETSC_TRUE;
3445599ef60dSHong Zhang   B->preallocated           = PETSC_TRUE;
3446599ef60dSHong Zhang   PetscFunctionReturn(0);
3447599ef60dSHong Zhang }
3448599ef60dSHong Zhang 
34493c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */
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  MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info)
34553c2a7987SHong Zhang {
34563c2a7987SHong Zhang   PetscErrorCode ierr;
34573c2a7987SHong Zhang 
34583c2a7987SHong Zhang   PetscFunctionBegin;
34593c2a7987SHong Zhang   ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr);
34603c2a7987SHong Zhang   PetscFunctionReturn(0);
34613c2a7987SHong Zhang }
34623c2a7987SHong Zhang 
34633c2a7987SHong Zhang /*
34643c2a7987SHong Zhang    same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors
34653c2a7987SHong Zhang    - intend to replace existing MatLUFactorNumeric_SeqAIJ()
34663c2a7987SHong Zhang */
3467fe97e370SBarry Smith /*
3468bebe2cf6SSatish 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
3469fe97e370SBarry Smith */
3470fe97e370SBarry Smith 
34717087cfbeSBarry Smith PetscErrorCode  MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info)
34723c2a7987SHong Zhang {
34733c2a7987SHong Zhang   Mat            C     =fact;
34743c2a7987SHong Zhang   Mat_SeqAIJ     *a    =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data;
34753c2a7987SHong Zhang   IS             isrow = b->row,isicol = b->icol;
34763c2a7987SHong Zhang   PetscErrorCode ierr;
34773c2a7987SHong Zhang   const PetscInt *r,*ic,*ics;
3478b78a477dSHong Zhang   PetscInt       i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
3479b78a477dSHong Zhang   PetscInt       *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj;
34803c2a7987SHong Zhang   MatScalar      *rtmp,*pc,multiplier,*v,*pv,*aa=a->a;
3481182b8fbaSHong Zhang   PetscReal      dt=info->dt,shift=info->shiftamount;
3482ace3abfcSBarry Smith   PetscBool      row_identity, col_identity;
34833c2a7987SHong Zhang 
34843c2a7987SHong Zhang   PetscFunctionBegin;
34853c2a7987SHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
34863c2a7987SHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
3487854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&rtmp);CHKERRQ(ierr);
34883c2a7987SHong Zhang   ics  = ic;
34893c2a7987SHong Zhang 
34903c2a7987SHong Zhang   for (i=0; i<n; i++) {
3491b78a477dSHong Zhang     /* initialize rtmp array */
3492b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i];       /* num of nozeros in L(i,:) */
34933c2a7987SHong Zhang     bjtmp = bj + bi[i];
3494b78a477dSHong Zhang     for  (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0;
3495b78a477dSHong Zhang     rtmp[i] = 0.0;
3496b78a477dSHong Zhang     nzu     = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */
3497b78a477dSHong Zhang     bjtmp   = bj + bdiag[i+1] + 1;
3498b78a477dSHong Zhang     for  (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0;
34993c2a7987SHong Zhang 
3500b78a477dSHong Zhang     /* load in initial unfactored row of A */
35016da515a1SHong Zhang     /* printf("row %d\n",i); */
35023c2a7987SHong Zhang     nz    = ai[r[i]+1] - ai[r[i]];
35033c2a7987SHong Zhang     ajtmp = aj + ai[r[i]];
35043c2a7987SHong Zhang     v     = aa + ai[r[i]];
35053c2a7987SHong Zhang     for (j=0; j<nz; j++) {
3506b78a477dSHong Zhang       rtmp[ics[*ajtmp++]] = v[j];
35076da515a1SHong Zhang       /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */
35083c2a7987SHong Zhang     }
35096da515a1SHong Zhang     /* printf("\n"); */
35103c2a7987SHong Zhang 
3511b78a477dSHong Zhang     /* numerical factorization */
3512b78a477dSHong Zhang     bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */
3513b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i]; /* num of entries in L(i,:) */
3514b78a477dSHong Zhang     k     = 0;
3515b78a477dSHong Zhang     while (k < nzl) {
35163c2a7987SHong Zhang       row = *bjtmp++;
35176da515a1SHong Zhang       /* printf("  prow %d\n",row); */
35183c2a7987SHong Zhang       pc         = rtmp + row;
3519b78a477dSHong Zhang       pv         = b->a + bdiag[row]; /* 1./(diag of the pivot row) */
3520b78a477dSHong Zhang       multiplier = (*pc) * (*pv);
35213c2a7987SHong Zhang       *pc        = multiplier;
3522b78a477dSHong Zhang       if (PetscAbsScalar(multiplier) > dt) {
3523b78a477dSHong Zhang         pj = bj + bdiag[row+1] + 1;         /* point to 1st entry of U(row,:) */
3524b78a477dSHong Zhang         pv = b->a + bdiag[row+1] + 1;
3525b78a477dSHong Zhang         nz = bdiag[row] - bdiag[row+1] - 1;         /* num of entries in U(row,:), excluding diagonal */
3526b78a477dSHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
352728b1a77fSLisandro Dalcin         ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
35283c2a7987SHong Zhang       }
3529b78a477dSHong Zhang       k++;
35303c2a7987SHong Zhang     }
35313c2a7987SHong Zhang 
3532b78a477dSHong Zhang     /* finished row so stick it into b->a */
3533b78a477dSHong Zhang     /* L-part */
3534b78a477dSHong Zhang     pv  = b->a + bi[i];
3535b78a477dSHong Zhang     pj  = bj + bi[i];
3536b78a477dSHong Zhang     nzl = bi[i+1] - bi[i];
3537b78a477dSHong Zhang     for (j=0; j<nzl; j++) {
3538b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
35396da515a1SHong Zhang       /* printf(" (%d,%g),",pj[j],pv[j]); */
35403c2a7987SHong Zhang     }
3541b78a477dSHong Zhang 
3542b78a477dSHong Zhang     /* diagonal: invert diagonal entries for simplier triangular solves */
3543b78a477dSHong Zhang     if (rtmp[i] == 0.0) rtmp[i] = dt+shift;
3544b78a477dSHong Zhang     b->a[bdiag[i]] = 1.0/rtmp[i];
35456da515a1SHong Zhang     /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */
3546b78a477dSHong Zhang 
3547b78a477dSHong Zhang     /* U-part */
3548b78a477dSHong Zhang     pv  = b->a + bdiag[i+1] + 1;
3549b78a477dSHong Zhang     pj  = bj + bdiag[i+1] + 1;
3550b78a477dSHong Zhang     nzu = bdiag[i] - bdiag[i+1] - 1;
3551b78a477dSHong Zhang     for (j=0; j<nzu; j++) {
3552b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
35536da515a1SHong Zhang       /* printf(" (%d,%g),",pj[j],pv[j]); */
3554b78a477dSHong Zhang     }
35556da515a1SHong Zhang     /* printf("\n"); */
3556b78a477dSHong Zhang   }
3557b78a477dSHong Zhang 
35583c2a7987SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
35593c2a7987SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
35603c2a7987SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3561b78a477dSHong Zhang 
35623c2a7987SHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
35633c2a7987SHong Zhang   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
35643c2a7987SHong Zhang   if (row_identity && col_identity) {
356535233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
35663c2a7987SHong Zhang   } else {
356735233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ;
35683c2a7987SHong Zhang   }
3569b78a477dSHong Zhang   C->ops->solveadd          = 0;
3570b78a477dSHong Zhang   C->ops->solvetranspose    = 0;
3571b78a477dSHong Zhang   C->ops->solvetransposeadd = 0;
3572b78a477dSHong Zhang   C->ops->matsolve          = 0;
35733c2a7987SHong Zhang   C->assembled              = PETSC_TRUE;
35743c2a7987SHong Zhang   C->preallocated           = PETSC_TRUE;
35752205254eSKarl Rupp 
35763c2a7987SHong Zhang   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
35773c2a7987SHong Zhang   PetscFunctionReturn(0);
35783c2a7987SHong Zhang }
3579