xref: /petsc/src/mat/impls/aij/seq/aijfact.c (revision 1683a169f336407cb9d0d8662b3b43f2a09d25dc)
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);
39ace3abfcSBarry Smith   ierr     = PetscMemzero(done,n*sizeof(PetscBool));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;
1642f18eb33SBarry Smith   ierr          = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space);CHKERRQ(ierr);
1651393bc97SHong Zhang   current_space = free_space;
166289bc588SBarry Smith 
167289bc588SBarry Smith   for (i=0; i<n; i++) {
1681393bc97SHong Zhang     /* copy previous fill into linked list */
169289bc588SBarry Smith     nzi = 0;
1701393bc97SHong Zhang     nnz = ai[r[i]+1] - ai[r[i]];
1711393bc97SHong Zhang     ajtmp = aj + ai[r[i]];
172afcc9904SHong Zhang     ierr  = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
1731393bc97SHong Zhang     nzi  += nlnk;
1741393bc97SHong Zhang 
1751393bc97SHong Zhang     /* add pivot rows into linked list */
1761393bc97SHong Zhang     row = lnk[n];
1771393bc97SHong Zhang     while (row < i) {
178d1170560SHong Zhang       nzbd  = bdiag[row] - bi[row] + 1;   /* num of entries in the row with column index <= row */
179d1170560SHong Zhang       ajtmp = bi_ptr[row] + nzbd;   /* points to the entry next to the diagonal */
180d1170560SHong Zhang       ierr  = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
1811393bc97SHong Zhang       nzi  += nlnk;
1821393bc97SHong Zhang       row   = lnk[row];
183289bc588SBarry Smith     }
1841393bc97SHong Zhang     bi[i+1] = bi[i] + nzi;
1851393bc97SHong Zhang     im[i]   = nzi;
1861393bc97SHong Zhang 
1871393bc97SHong Zhang     /* mark bdiag */
1881393bc97SHong Zhang     nzbd = 0;
1891393bc97SHong Zhang     nnz  = nzi;
1901393bc97SHong Zhang     k    = lnk[n];
1911393bc97SHong Zhang     while (nnz-- && k < i) {
1921393bc97SHong Zhang       nzbd++;
1931393bc97SHong Zhang       k = lnk[k];
1941393bc97SHong Zhang     }
1951393bc97SHong Zhang     bdiag[i] = bi[i] + nzbd;
1961393bc97SHong Zhang 
1971393bc97SHong Zhang     /* if free space is not available, make more free space */
1981393bc97SHong Zhang     if (current_space->local_remaining<nzi) {
199f91af8c7SBarry Smith       nnz  = PetscIntMultTruncate(n - i,nzi); /* estimated and max additional space needed */
200a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
2011393bc97SHong Zhang       reallocs++;
2021393bc97SHong Zhang     }
2031393bc97SHong Zhang 
2041393bc97SHong Zhang     /* copy data into free space, then initialize lnk */
2051393bc97SHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
2062205254eSKarl Rupp 
2071393bc97SHong Zhang     bi_ptr[i]                       = current_space->array;
2081393bc97SHong Zhang     current_space->array           += nzi;
2091393bc97SHong Zhang     current_space->local_used      += nzi;
2101393bc97SHong Zhang     current_space->local_remaining -= nzi;
211289bc588SBarry Smith   }
2126cf91177SBarry Smith #if defined(PETSC_USE_INFO)
213464e8d44SSatish Balay   if (ai[n] != 0) {
2141393bc97SHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
21557622a8eSBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)f,(double)af);CHKERRQ(ierr);
21657622a8eSBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
21757622a8eSBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g);\n",(double)af);CHKERRQ(ierr);
218ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
21905bf559cSSatish Balay   } else {
220ae15b995SBarry Smith     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
22105bf559cSSatish Balay   }
22263ba0a88SBarry Smith #endif
2232fb3ed76SBarry Smith 
224898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
225898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
2261987afe7SBarry Smith 
2271393bc97SHong Zhang   /* destroy list of free space and other temporary array(s) */
228854ce69bSBarry Smith   ierr = PetscMalloc1(bi[n]+1,&bj);CHKERRQ(ierr);
229a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr);
2301393bc97SHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
23135baf27eSSatish Balay   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
232289bc588SBarry Smith 
233289bc588SBarry Smith   /* put together the new matrix */
2340298fd71SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
2353bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)B,(PetscObject)isicol);CHKERRQ(ierr);
236719d5645SBarry Smith   b    = (Mat_SeqAIJ*)(B)->data;
2372205254eSKarl Rupp 
238e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
239e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
2407c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
2412205254eSKarl Rupp 
242854ce69bSBarry Smith   ierr    = PetscMalloc1(bi[n]+1,&b->a);CHKERRQ(ierr);
2431393bc97SHong Zhang   b->j    = bj;
2441393bc97SHong Zhang   b->i    = bi;
2451393bc97SHong Zhang   b->diag = bdiag;
246416022c9SBarry Smith   b->ilen = 0;
247416022c9SBarry Smith   b->imax = 0;
248416022c9SBarry Smith   b->row  = isrow;
249416022c9SBarry Smith   b->col  = iscol;
250c38d4ed2SBarry Smith   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
251c38d4ed2SBarry Smith   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
25282bf6240SBarry Smith   b->icol = isicol;
253854ce69bSBarry Smith   ierr    = PetscMalloc1(n+1,&b->solve_work);CHKERRQ(ierr);
2541393bc97SHong Zhang 
2551393bc97SHong Zhang   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
2563bb1ff40SBarry Smith   ierr     = PetscLogObjectMemory((PetscObject)B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
2571393bc97SHong Zhang   b->maxnz = b->nz = bi[n];
2587fd98bd6SLois Curfman McInnes 
259d5f3da31SBarry Smith   (B)->factortype            = MAT_FACTOR_LU;
260719d5645SBarry Smith   (B)->info.factor_mallocs   = reallocs;
261719d5645SBarry Smith   (B)->info.fill_ratio_given = f;
262703deb49SSatish Balay 
2639f7d0b68SHong Zhang   if (ai[n]) {
264719d5645SBarry Smith     (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
265eea2913aSSatish Balay   } else {
266719d5645SBarry Smith     (B)->info.fill_ratio_needed = 0.0;
267eea2913aSSatish Balay   }
268ad04f41aSHong Zhang   (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace;
269f2344125SBarry Smith   if (a->inode.size) {
270f2344125SBarry Smith     (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
271f2344125SBarry Smith   }
2723a40ed3dSBarry Smith   PetscFunctionReturn(0);
273289bc588SBarry Smith }
2741393bc97SHong Zhang 
27535233d99SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
276ce3d78c0SShri Abhyankar {
277ce3d78c0SShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
278ce3d78c0SShri Abhyankar   IS                 isicol;
279ce3d78c0SShri Abhyankar   PetscErrorCode     ierr;
2808758e1faSBarry Smith   const PetscInt     *r,*ic,*ai=a->i,*aj=a->j,*ajtmp;
2818758e1faSBarry Smith   PetscInt           i,n=A->rmap->n;
2828758e1faSBarry Smith   PetscInt           *bi,*bj;
283ce3d78c0SShri Abhyankar   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
284ce3d78c0SShri Abhyankar   PetscReal          f;
285ce3d78c0SShri Abhyankar   PetscInt           nlnk,*lnk,k,**bi_ptr;
2860298fd71SBarry Smith   PetscFreeSpaceList free_space=NULL,current_space=NULL;
287ce3d78c0SShri Abhyankar   PetscBT            lnkbt;
288ece542f9SHong Zhang   PetscBool          missing;
289ce3d78c0SShri Abhyankar 
290ce3d78c0SShri Abhyankar   PetscFunctionBegin;
291e32f2f54SBarry Smith   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square");
292ece542f9SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
293ece542f9SHong Zhang   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
294ece542f9SHong Zhang 
295ce3d78c0SShri Abhyankar   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
296ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
297ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
298ce3d78c0SShri Abhyankar 
2991bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
300854ce69bSBarry Smith   ierr  = PetscMalloc1(n+1,&bi);CHKERRQ(ierr);
301854ce69bSBarry Smith   ierr  = PetscMalloc1(n+1,&bdiag);CHKERRQ(ierr);
302a6df321fSShri Abhyankar   bi[0] = bdiag[0] = 0;
3039b48462bSShri Abhyankar 
3049b48462bSShri Abhyankar   /* linked list for storing column indices of the active row */
3059b48462bSShri Abhyankar   nlnk = n + 1;
3069b48462bSShri Abhyankar   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3079b48462bSShri Abhyankar 
308dcca6d9dSJed Brown   ierr = PetscMalloc2(n+1,&bi_ptr,n+1,&im);CHKERRQ(ierr);
3099b48462bSShri Abhyankar 
3109b48462bSShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
3119b48462bSShri Abhyankar   f             = info->fill;
3122f18eb33SBarry Smith   ierr          = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space);CHKERRQ(ierr);
3139b48462bSShri Abhyankar   current_space = free_space;
3149b48462bSShri Abhyankar 
3159b48462bSShri Abhyankar   for (i=0; i<n; i++) {
3169b48462bSShri Abhyankar     /* copy previous fill into linked list */
3179b48462bSShri Abhyankar     nzi = 0;
3189b48462bSShri Abhyankar     nnz = ai[r[i]+1] - ai[r[i]];
3199b48462bSShri Abhyankar     ajtmp = aj + ai[r[i]];
3209b48462bSShri Abhyankar     ierr  = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3219b48462bSShri Abhyankar     nzi  += nlnk;
3229b48462bSShri Abhyankar 
3239b48462bSShri Abhyankar     /* add pivot rows into linked list */
3249b48462bSShri Abhyankar     row = lnk[n];
3259b48462bSShri Abhyankar     while (row < i) {
3269b48462bSShri Abhyankar       nzbd  = bdiag[row] + 1; /* num of entries in the row with column index <= row */
3279b48462bSShri Abhyankar       ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
3289b48462bSShri Abhyankar       ierr  = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
3299b48462bSShri Abhyankar       nzi  += nlnk;
3309b48462bSShri Abhyankar       row   = lnk[row];
3319b48462bSShri Abhyankar     }
3329b48462bSShri Abhyankar     bi[i+1] = bi[i] + nzi;
3339b48462bSShri Abhyankar     im[i]   = nzi;
3349b48462bSShri Abhyankar 
3359b48462bSShri Abhyankar     /* mark bdiag */
3369b48462bSShri Abhyankar     nzbd = 0;
3379b48462bSShri Abhyankar     nnz  = nzi;
3389b48462bSShri Abhyankar     k    = lnk[n];
3399b48462bSShri Abhyankar     while (nnz-- && k < i) {
3409b48462bSShri Abhyankar       nzbd++;
3419b48462bSShri Abhyankar       k = lnk[k];
3429b48462bSShri Abhyankar     }
3439b48462bSShri Abhyankar     bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */
3449b48462bSShri Abhyankar 
3459b48462bSShri Abhyankar     /* if free space is not available, make more free space */
3469b48462bSShri Abhyankar     if (current_space->local_remaining<nzi) {
3472f18eb33SBarry Smith       /* estimated additional space needed */
3481df4278eSBarry Smith       nnz  = PetscIntMultTruncate(2,PetscIntMultTruncate(n-1,nzi));
3499b48462bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
3509b48462bSShri Abhyankar       reallocs++;
3519b48462bSShri Abhyankar     }
3529b48462bSShri Abhyankar 
3539b48462bSShri Abhyankar     /* copy data into free space, then initialize lnk */
3549b48462bSShri Abhyankar     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
3552205254eSKarl Rupp 
3569b48462bSShri Abhyankar     bi_ptr[i]                       = current_space->array;
3579b48462bSShri Abhyankar     current_space->array           += nzi;
3589b48462bSShri Abhyankar     current_space->local_used      += nzi;
3599b48462bSShri Abhyankar     current_space->local_remaining -= nzi;
3609b48462bSShri Abhyankar   }
3619b48462bSShri Abhyankar 
3629b48462bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3639b48462bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3649b48462bSShri Abhyankar 
3659263d837SHong Zhang   /*   copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */
366854ce69bSBarry Smith   ierr = PetscMalloc1(bi[n]+1,&bj);CHKERRQ(ierr);
3672ce24eb6SHong Zhang   ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
3689b48462bSShri Abhyankar   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
3699b48462bSShri Abhyankar   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
3709b48462bSShri Abhyankar 
3719b48462bSShri Abhyankar   /* put together the new matrix */
3720298fd71SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
3733bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)B,(PetscObject)isicol);CHKERRQ(ierr);
3749b48462bSShri Abhyankar   b    = (Mat_SeqAIJ*)(B)->data;
3752205254eSKarl Rupp 
3769b48462bSShri Abhyankar   b->free_a       = PETSC_TRUE;
3779b48462bSShri Abhyankar   b->free_ij      = PETSC_TRUE;
3789b48462bSShri Abhyankar   b->singlemalloc = PETSC_FALSE;
3792205254eSKarl Rupp 
380854ce69bSBarry Smith   ierr = PetscMalloc1(bdiag[0]+1,&b->a);CHKERRQ(ierr);
3812205254eSKarl Rupp 
3829b48462bSShri Abhyankar   b->j    = bj;
3839b48462bSShri Abhyankar   b->i    = bi;
3849b48462bSShri Abhyankar   b->diag = bdiag;
3859b48462bSShri Abhyankar   b->ilen = 0;
3869b48462bSShri Abhyankar   b->imax = 0;
3879b48462bSShri Abhyankar   b->row  = isrow;
3889b48462bSShri Abhyankar   b->col  = iscol;
3899b48462bSShri Abhyankar   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
3909b48462bSShri Abhyankar   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
3919b48462bSShri Abhyankar   b->icol = isicol;
392854ce69bSBarry Smith   ierr    = PetscMalloc1(n+1,&b->solve_work);CHKERRQ(ierr);
3939b48462bSShri Abhyankar 
3949b48462bSShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
3953bb1ff40SBarry Smith   ierr     = PetscLogObjectMemory((PetscObject)B,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
3969b48462bSShri Abhyankar   b->maxnz = b->nz = bdiag[0]+1;
3972205254eSKarl Rupp 
398d5f3da31SBarry Smith   B->factortype            = MAT_FACTOR_LU;
399f284f12aSHong Zhang   B->info.factor_mallocs   = reallocs;
400f284f12aSHong Zhang   B->info.fill_ratio_given = f;
4019b48462bSShri Abhyankar 
4029b48462bSShri Abhyankar   if (ai[n]) {
403f284f12aSHong Zhang     B->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
4049b48462bSShri Abhyankar   } else {
405f284f12aSHong Zhang     B->info.fill_ratio_needed = 0.0;
4069b48462bSShri Abhyankar   }
4079263d837SHong Zhang #if defined(PETSC_USE_INFO)
4089263d837SHong Zhang   if (ai[n] != 0) {
4099263d837SHong Zhang     PetscReal af = B->info.fill_ratio_needed;
41057622a8eSBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)f,(double)af);CHKERRQ(ierr);
41157622a8eSBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
41257622a8eSBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g);\n",(double)af);CHKERRQ(ierr);
4139263d837SHong Zhang     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
4149263d837SHong Zhang   } else {
4159263d837SHong Zhang     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
4169263d837SHong Zhang   }
4179263d837SHong Zhang #endif
41835233d99SShri Abhyankar   B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ;
419f2344125SBarry Smith   if (a->inode.size) {
420d3ac4fa3SBarry Smith     B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode;
421d3ac4fa3SBarry Smith   }
422a02bda8eSBarry Smith   ierr = MatSeqAIJCheckInode_FactorLU(B);CHKERRQ(ierr);
4239b48462bSShri Abhyankar   PetscFunctionReturn(0);
4249b48462bSShri Abhyankar }
4259b48462bSShri Abhyankar 
4265b5bc046SBarry Smith /*
4275b5bc046SBarry Smith     Trouble in factorization, should we dump the original matrix?
4285b5bc046SBarry Smith */
4295b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A)
4305b5bc046SBarry Smith {
4315b5bc046SBarry Smith   PetscErrorCode ierr;
432ace3abfcSBarry Smith   PetscBool      flg = PETSC_FALSE;
4335b5bc046SBarry Smith 
4345b5bc046SBarry Smith   PetscFunctionBegin;
435c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)A)->options,NULL,"-mat_factor_dump_on_error",&flg,NULL);CHKERRQ(ierr);
4365b5bc046SBarry Smith   if (flg) {
4375b5bc046SBarry Smith     PetscViewer viewer;
4385b5bc046SBarry Smith     char        filename[PETSC_MAX_PATH_LEN];
4395b5bc046SBarry Smith 
4405b5bc046SBarry Smith     ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr);
441ce94432eSBarry Smith     ierr = PetscViewerBinaryOpen(PetscObjectComm((PetscObject)A),filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr);
4425b5bc046SBarry Smith     ierr = MatView(A,viewer);CHKERRQ(ierr);
4436bf464f9SBarry Smith     ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr);
4445b5bc046SBarry Smith   }
4455b5bc046SBarry Smith   PetscFunctionReturn(0);
4465b5bc046SBarry Smith }
4475b5bc046SBarry Smith 
44835233d99SShri Abhyankar PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
449c9c72f8fSHong Zhang {
450c9c72f8fSHong Zhang   Mat             C     =B;
451c9c72f8fSHong Zhang   Mat_SeqAIJ      *a    =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data;
452c9c72f8fSHong Zhang   IS              isrow = b->row,isicol = b->icol;
453c9c72f8fSHong Zhang   PetscErrorCode  ierr;
454c9c72f8fSHong Zhang   const PetscInt  *r,*ic,*ics;
455bbd65245SShri Abhyankar   const PetscInt  n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bdiag=b->diag;
456bbd65245SShri Abhyankar   PetscInt        i,j,k,nz,nzL,row,*pj;
457bbd65245SShri Abhyankar   const PetscInt  *ajtmp,*bjtmp;
458bbd65245SShri Abhyankar   MatScalar       *rtmp,*pc,multiplier,*pv;
459bbd65245SShri Abhyankar   const MatScalar *aa=a->a,*v;
460ace3abfcSBarry Smith   PetscBool       row_identity,col_identity;
461d90ac83dSHong Zhang   FactorShiftCtx  sctx;
4624f81c4b7SBarry Smith   const PetscInt  *ddiag;
463d4ad06f7SHong Zhang   PetscReal       rs;
464d4ad06f7SHong Zhang   MatScalar       d;
465d4ad06f7SHong Zhang 
466c9c72f8fSHong Zhang   PetscFunctionBegin;
467d6e5152cSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
468d90ac83dSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
469d4ad06f7SHong Zhang 
470f4db908eSBarry Smith   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
471d4ad06f7SHong Zhang     ddiag          = a->diag;
472d4ad06f7SHong Zhang     sctx.shift_top = info->zeropivot;
473d4ad06f7SHong Zhang     for (i=0; i<n; i++) {
474d4ad06f7SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
475d4ad06f7SHong Zhang       d  = (aa)[ddiag[i]];
476d4ad06f7SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
477d4ad06f7SHong Zhang       v  = aa+ai[i];
478d4ad06f7SHong Zhang       nz = ai[i+1] - ai[i];
4792205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
480d4ad06f7SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
481d4ad06f7SHong Zhang     }
482d4ad06f7SHong Zhang     sctx.shift_top *= 1.1;
483d4ad06f7SHong Zhang     sctx.nshift_max = 5;
484d4ad06f7SHong Zhang     sctx.shift_lo   = 0.;
485d4ad06f7SHong Zhang     sctx.shift_hi   = 1.;
486d4ad06f7SHong Zhang   }
487d4ad06f7SHong Zhang 
488c9c72f8fSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
489c9c72f8fSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
490854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&rtmp);CHKERRQ(ierr);
491c9c72f8fSHong Zhang   ics  = ic;
492c9c72f8fSHong Zhang 
493d4ad06f7SHong Zhang   do {
49407b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
495c9c72f8fSHong Zhang     for (i=0; i<n; i++) {
496c9c72f8fSHong Zhang       /* zero rtmp */
497c9c72f8fSHong Zhang       /* L part */
498c9c72f8fSHong Zhang       nz    = bi[i+1] - bi[i];
499c9c72f8fSHong Zhang       bjtmp = bj + bi[i];
500c9c72f8fSHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
501c9c72f8fSHong Zhang 
502c9c72f8fSHong Zhang       /* U part */
50362fbd6afSShri Abhyankar       nz    = bdiag[i]-bdiag[i+1];
50462fbd6afSShri Abhyankar       bjtmp = bj + bdiag[i+1]+1;
505813a1f2bSShri Abhyankar       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
506813a1f2bSShri Abhyankar 
507813a1f2bSShri Abhyankar       /* load in initial (unfactored row) */
508813a1f2bSShri Abhyankar       nz    = ai[r[i]+1] - ai[r[i]];
509813a1f2bSShri Abhyankar       ajtmp = aj + ai[r[i]];
510813a1f2bSShri Abhyankar       v     = aa + ai[r[i]];
511813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
512813a1f2bSShri Abhyankar         rtmp[ics[ajtmp[j]]] = v[j];
513813a1f2bSShri Abhyankar       }
514813a1f2bSShri Abhyankar       /* ZeropivotApply() */
51598a93161SHong Zhang       rtmp[i] += sctx.shift_amount;  /* shift the diagonal of the matrix */
516813a1f2bSShri Abhyankar 
517813a1f2bSShri Abhyankar       /* elimination */
518813a1f2bSShri Abhyankar       bjtmp = bj + bi[i];
519813a1f2bSShri Abhyankar       row   = *bjtmp++;
520813a1f2bSShri Abhyankar       nzL   = bi[i+1] - bi[i];
521f268cba8SShri Abhyankar       for (k=0; k < nzL; k++) {
522813a1f2bSShri Abhyankar         pc = rtmp + row;
523813a1f2bSShri Abhyankar         if (*pc != 0.0) {
524813a1f2bSShri Abhyankar           pv         = b->a + bdiag[row];
525813a1f2bSShri Abhyankar           multiplier = *pc * (*pv);
526813a1f2bSShri Abhyankar           *pc        = multiplier;
5272205254eSKarl Rupp 
52862fbd6afSShri Abhyankar           pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */
52962fbd6afSShri Abhyankar           pv = b->a + bdiag[row+1]+1;
53062fbd6afSShri Abhyankar           nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
5312205254eSKarl Rupp 
532813a1f2bSShri Abhyankar           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
53328b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
534813a1f2bSShri Abhyankar         }
535f268cba8SShri Abhyankar         row = *bjtmp++;
536813a1f2bSShri Abhyankar       }
537813a1f2bSShri Abhyankar 
538813a1f2bSShri Abhyankar       /* finished row so stick it into b->a */
539813a1f2bSShri Abhyankar       rs = 0.0;
540813a1f2bSShri Abhyankar       /* L part */
541813a1f2bSShri Abhyankar       pv = b->a + bi[i];
542813a1f2bSShri Abhyankar       pj = b->j + bi[i];
543813a1f2bSShri Abhyankar       nz = bi[i+1] - bi[i];
544813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
545813a1f2bSShri Abhyankar         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
546813a1f2bSShri Abhyankar       }
547813a1f2bSShri Abhyankar 
548813a1f2bSShri Abhyankar       /* U part */
54962fbd6afSShri Abhyankar       pv = b->a + bdiag[i+1]+1;
55062fbd6afSShri Abhyankar       pj = b->j + bdiag[i+1]+1;
55162fbd6afSShri Abhyankar       nz = bdiag[i] - bdiag[i+1]-1;
552813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
553813a1f2bSShri Abhyankar         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
554813a1f2bSShri Abhyankar       }
555813a1f2bSShri Abhyankar 
556813a1f2bSShri Abhyankar       sctx.rs = rs;
557813a1f2bSShri Abhyankar       sctx.pv = rtmp[i];
5584cccfbddSHong Zhang       ierr    = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr);
55907b50cabSHong Zhang       if (sctx.newshift) break; /* break for-loop */
56007b50cabSHong Zhang       rtmp[i] = sctx.pv; /* sctx.pv might be updated in the case of MAT_SHIFT_INBLOCKS */
561813a1f2bSShri Abhyankar 
562813a1f2bSShri Abhyankar       /* Mark diagonal and invert diagonal for simplier triangular solves */
563813a1f2bSShri Abhyankar       pv  = b->a + bdiag[i];
564813a1f2bSShri Abhyankar       *pv = 1.0/rtmp[i];
565813a1f2bSShri Abhyankar 
566813a1f2bSShri Abhyankar     } /* endof for (i=0; i<n; i++) { */
567813a1f2bSShri Abhyankar 
5688ff23777SHong Zhang     /* MatPivotRefine() */
56907b50cabSHong Zhang     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
570813a1f2bSShri Abhyankar       /*
571813a1f2bSShri Abhyankar        * if no shift in this attempt & shifting & started shifting & can refine,
572813a1f2bSShri Abhyankar        * then try lower shift
573813a1f2bSShri Abhyankar        */
574813a1f2bSShri Abhyankar       sctx.shift_hi       = sctx.shift_fraction;
575813a1f2bSShri Abhyankar       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
576813a1f2bSShri Abhyankar       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
57707b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
578813a1f2bSShri Abhyankar       sctx.nshift++;
579813a1f2bSShri Abhyankar     }
58007b50cabSHong Zhang   } while (sctx.newshift);
581813a1f2bSShri Abhyankar 
582813a1f2bSShri Abhyankar   ierr = PetscFree(rtmp);CHKERRQ(ierr);
583813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
584813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
585d3ac4fa3SBarry Smith 
586813a1f2bSShri Abhyankar   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
587813a1f2bSShri Abhyankar   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
588abb87a52SBarry Smith   if (b->inode.size) {
589abb87a52SBarry Smith     C->ops->solve = MatSolve_SeqAIJ_Inode;
590abb87a52SBarry Smith   } else if (row_identity && col_identity) {
59135233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
592813a1f2bSShri Abhyankar   } else {
59335233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ;
594813a1f2bSShri Abhyankar   }
59535233d99SShri Abhyankar   C->ops->solveadd          = MatSolveAdd_SeqAIJ;
59635233d99SShri Abhyankar   C->ops->solvetranspose    = MatSolveTranspose_SeqAIJ;
59735233d99SShri Abhyankar   C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ;
59835233d99SShri Abhyankar   C->ops->matsolve          = MatMatSolve_SeqAIJ;
599813a1f2bSShri Abhyankar   C->assembled              = PETSC_TRUE;
600813a1f2bSShri Abhyankar   C->preallocated           = PETSC_TRUE;
6012205254eSKarl Rupp 
602813a1f2bSShri Abhyankar   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
603813a1f2bSShri Abhyankar 
60414d2772eSHong Zhang   /* MatShiftView(A,info,&sctx) */
605813a1f2bSShri Abhyankar   if (sctx.nshift) {
606f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
60757622a8eSBarry 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);
608f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
60957622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
610f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS) {
61157622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %g\n",sctx.nshift,(double)info->shiftamount);CHKERRQ(ierr);
612813a1f2bSShri Abhyankar     }
613813a1f2bSShri Abhyankar   }
614813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
615813a1f2bSShri Abhyankar }
616813a1f2bSShri Abhyankar 
617ad04f41aSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info)
618289bc588SBarry Smith {
619719d5645SBarry Smith   Mat             C     =B;
620416022c9SBarry Smith   Mat_SeqAIJ      *a    =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data;
62182bf6240SBarry Smith   IS              isrow = b->row,isicol = b->icol;
6226849ba73SBarry Smith   PetscErrorCode  ierr;
6235d0c19d7SBarry Smith   const PetscInt  *r,*ic,*ics;
624d278edc7SBarry Smith   PetscInt        nz,row,i,j,n=A->rmap->n,diag;
625d278edc7SBarry Smith   const PetscInt  *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
626d278edc7SBarry Smith   const PetscInt  *ajtmp,*bjtmp,*diag_offset = b->diag,*pj;
627d278edc7SBarry Smith   MatScalar       *pv,*rtmp,*pc,multiplier,d;
628d278edc7SBarry Smith   const MatScalar *v,*aa=a->a;
62933f9893dSHong Zhang   PetscReal       rs=0.0;
630d90ac83dSHong Zhang   FactorShiftCtx  sctx;
6314f81c4b7SBarry Smith   const PetscInt  *ddiag;
632ace3abfcSBarry Smith   PetscBool       row_identity, col_identity;
633289bc588SBarry Smith 
6343a40ed3dSBarry Smith   PetscFunctionBegin;
635504a3fadSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
636504a3fadSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
637ada7143aSSatish Balay 
638f4db908eSBarry Smith   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
63942898933SHong Zhang     ddiag          = a->diag;
6409f7d0b68SHong Zhang     sctx.shift_top = info->zeropivot;
6416cc28720Svictorle     for (i=0; i<n; i++) {
6429f95998fSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
6439f7d0b68SHong Zhang       d  = (aa)[ddiag[i]];
6441a890ab1SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
6459f7d0b68SHong Zhang       v  = aa+ai[i];
6469f7d0b68SHong Zhang       nz = ai[i+1] - ai[i];
6472205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
6481a890ab1SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
6496cc28720Svictorle     }
650118f5789SBarry Smith     sctx.shift_top *= 1.1;
651d2276718SHong Zhang     sctx.nshift_max = 5;
652d2276718SHong Zhang     sctx.shift_lo   = 0.;
653d2276718SHong Zhang     sctx.shift_hi   = 1.;
654a0a356efSHong Zhang   }
655a0a356efSHong Zhang 
656504a3fadSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
657504a3fadSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
658854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&rtmp);CHKERRQ(ierr);
659504a3fadSHong Zhang   ics  = ic;
660504a3fadSHong Zhang 
661085256b3SBarry Smith   do {
66207b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
663289bc588SBarry Smith     for (i=0; i<n; i++) {
664b3bf805bSHong Zhang       nz    = bi[i+1] - bi[i];
665b3bf805bSHong Zhang       bjtmp = bj + bi[i];
6669f7d0b68SHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
667289bc588SBarry Smith 
668289bc588SBarry Smith       /* load in initial (unfactored row) */
6699f7d0b68SHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
6709f7d0b68SHong Zhang       ajtmp = aj + ai[r[i]];
6719f7d0b68SHong Zhang       v     = aa + ai[r[i]];
672085256b3SBarry Smith       for (j=0; j<nz; j++) {
673b3bf805bSHong Zhang         rtmp[ics[ajtmp[j]]] = v[j];
674085256b3SBarry Smith       }
675d2276718SHong Zhang       rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */
676289bc588SBarry Smith 
677b3bf805bSHong Zhang       row = *bjtmp++;
678f2582111SSatish Balay       while  (row < i) {
6798c37ef55SBarry Smith         pc = rtmp + row;
6808c37ef55SBarry Smith         if (*pc != 0.0) {
681010a20caSHong Zhang           pv         = b->a + diag_offset[row];
682010a20caSHong Zhang           pj         = b->j + diag_offset[row] + 1;
68335aab85fSBarry Smith           multiplier = *pc / *pv++;
6848c37ef55SBarry Smith           *pc        = multiplier;
685b3bf805bSHong Zhang           nz         = bi[row+1] - diag_offset[row] - 1;
6869f7d0b68SHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
68728b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
688289bc588SBarry Smith         }
689b3bf805bSHong Zhang         row = *bjtmp++;
690289bc588SBarry Smith       }
691416022c9SBarry Smith       /* finished row so stick it into b->a */
692b3bf805bSHong Zhang       pv   = b->a + bi[i];
693b3bf805bSHong Zhang       pj   = b->j + bi[i];
694b3bf805bSHong Zhang       nz   = bi[i+1] - bi[i];
695b3bf805bSHong Zhang       diag = diag_offset[i] - bi[i];
696e57ff13aSBarry Smith       rs   = 0.0;
697d3d32019SBarry Smith       for (j=0; j<nz; j++) {
6989f7d0b68SHong Zhang         pv[j] = rtmp[pj[j]];
6999f7d0b68SHong Zhang         rs   += PetscAbsScalar(pv[j]);
700d3d32019SBarry Smith       }
701e57ff13aSBarry Smith       rs -= PetscAbsScalar(pv[diag]);
702d2276718SHong Zhang 
703d2276718SHong Zhang       sctx.rs = rs;
704d2276718SHong Zhang       sctx.pv = pv[diag];
7054cccfbddSHong Zhang       ierr    = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr);
70607b50cabSHong Zhang       if (sctx.newshift) break;
707504a3fadSHong Zhang       pv[diag] = sctx.pv;
70835aab85fSBarry Smith     }
709d2276718SHong Zhang 
71007b50cabSHong Zhang     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
7116cc28720Svictorle       /*
7126c037d1bSvictorle        * if no shift in this attempt & shifting & started shifting & can refine,
7136cc28720Svictorle        * then try lower shift
7146cc28720Svictorle        */
7150481f469SBarry Smith       sctx.shift_hi       = sctx.shift_fraction;
7160481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
7170481f469SBarry Smith       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
71807b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
719d2276718SHong Zhang       sctx.nshift++;
7206cc28720Svictorle     }
72107b50cabSHong Zhang   } while (sctx.newshift);
7220f11f9aeSSatish Balay 
72335aab85fSBarry Smith   /* invert diagonal entries for simplier triangular solves */
72435aab85fSBarry Smith   for (i=0; i<n; i++) {
725010a20caSHong Zhang     b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]];
72635aab85fSBarry Smith   }
727606d414cSSatish Balay   ierr = PetscFree(rtmp);CHKERRQ(ierr);
72878b31e54SBarry Smith   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
72978b31e54SBarry Smith   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
730d3ac4fa3SBarry Smith 
7318d8c7c43SBarry Smith   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
7328d8c7c43SBarry Smith   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
7338d8c7c43SBarry Smith   if (row_identity && col_identity) {
734ad04f41aSHong Zhang     C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_inplace;
7358d8c7c43SBarry Smith   } else {
736ad04f41aSHong Zhang     C->ops->solve = MatSolve_SeqAIJ_inplace;
737db4efbfdSBarry Smith   }
738ad04f41aSHong Zhang   C->ops->solveadd          = MatSolveAdd_SeqAIJ_inplace;
739ad04f41aSHong Zhang   C->ops->solvetranspose    = MatSolveTranspose_SeqAIJ_inplace;
740ad04f41aSHong Zhang   C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace;
741ad04f41aSHong Zhang   C->ops->matsolve          = MatMatSolve_SeqAIJ_inplace;
7422205254eSKarl Rupp 
743c456f294SBarry Smith   C->assembled    = PETSC_TRUE;
7445c9eb25fSBarry Smith   C->preallocated = PETSC_TRUE;
7452205254eSKarl Rupp 
746d0f46423SBarry Smith   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
747d2276718SHong Zhang   if (sctx.nshift) {
748f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
74957622a8eSBarry 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);
750f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
75157622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
7526cc28720Svictorle     }
7530697b6caSHong Zhang   }
754d3ac4fa3SBarry Smith   (C)->ops->solve          = MatSolve_SeqAIJ_inplace;
755d3ac4fa3SBarry Smith   (C)->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace;
7562205254eSKarl Rupp 
757a02bda8eSBarry Smith   ierr = MatSeqAIJCheckInode(C);CHKERRQ(ierr);
7583a40ed3dSBarry Smith   PetscFunctionReturn(0);
759289bc588SBarry Smith }
7600889b5dcSvictorle 
761137fb511SHong Zhang /*
762137fb511SHong Zhang    This routine implements inplace ILU(0) with row or/and column permutations.
763137fb511SHong Zhang    Input:
764137fb511SHong Zhang      A - original matrix
765137fb511SHong Zhang    Output;
766137fb511SHong Zhang      A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i]
767137fb511SHong Zhang          a->j (col index) is permuted by the inverse of colperm, then sorted
768137fb511SHong Zhang          a->a reordered accordingly with a->j
769137fb511SHong Zhang          a->diag (ptr to diagonal elements) is updated.
770137fb511SHong Zhang */
7710481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info)
772137fb511SHong Zhang {
773b051339dSHong Zhang   Mat_SeqAIJ      *a    =(Mat_SeqAIJ*)A->data;
774b051339dSHong Zhang   IS              isrow = a->row,isicol = a->icol;
775137fb511SHong Zhang   PetscErrorCode  ierr;
7765d0c19d7SBarry Smith   const PetscInt  *r,*ic,*ics;
7775d0c19d7SBarry Smith   PetscInt        i,j,n=A->rmap->n,*ai=a->i,*aj=a->j;
7785d0c19d7SBarry Smith   PetscInt        *ajtmp,nz,row;
779b051339dSHong Zhang   PetscInt        *diag = a->diag,nbdiag,*pj;
780a77337e4SBarry Smith   PetscScalar     *rtmp,*pc,multiplier,d;
781504a3fadSHong Zhang   MatScalar       *pv,*v;
782137fb511SHong Zhang   PetscReal       rs;
783d90ac83dSHong Zhang   FactorShiftCtx  sctx;
784504a3fadSHong Zhang   const MatScalar *aa=a->a,*vtmp;
785137fb511SHong Zhang 
786137fb511SHong Zhang   PetscFunctionBegin;
787e32f2f54SBarry Smith   if (A != B) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address");
788504a3fadSHong Zhang 
789504a3fadSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
790504a3fadSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
791504a3fadSHong Zhang 
792504a3fadSHong Zhang   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
793504a3fadSHong Zhang     const PetscInt *ddiag = a->diag;
794504a3fadSHong Zhang     sctx.shift_top = info->zeropivot;
795504a3fadSHong Zhang     for (i=0; i<n; i++) {
796504a3fadSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
797504a3fadSHong Zhang       d    = (aa)[ddiag[i]];
798504a3fadSHong Zhang       rs   = -PetscAbsScalar(d) - PetscRealPart(d);
799504a3fadSHong Zhang       vtmp = aa+ai[i];
800504a3fadSHong Zhang       nz   = ai[i+1] - ai[i];
8012205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(vtmp[j]);
802504a3fadSHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
803504a3fadSHong Zhang     }
804504a3fadSHong Zhang     sctx.shift_top *= 1.1;
805504a3fadSHong Zhang     sctx.nshift_max = 5;
806504a3fadSHong Zhang     sctx.shift_lo   = 0.;
807504a3fadSHong Zhang     sctx.shift_hi   = 1.;
808504a3fadSHong Zhang   }
809504a3fadSHong Zhang 
810137fb511SHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
811137fb511SHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
812854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&rtmp);CHKERRQ(ierr);
813137fb511SHong Zhang   ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr);
814b051339dSHong Zhang   ics  = ic;
815137fb511SHong Zhang 
816504a3fadSHong Zhang #if defined(MV)
81775567043SBarry Smith   sctx.shift_top      = 0.;
818e60cf9a0SBarry Smith   sctx.nshift_max     = 0;
81975567043SBarry Smith   sctx.shift_lo       = 0.;
82075567043SBarry Smith   sctx.shift_hi       = 0.;
82175567043SBarry Smith   sctx.shift_fraction = 0.;
822137fb511SHong Zhang 
823f4db908eSBarry Smith   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
82475567043SBarry Smith     sctx.shift_top = 0.;
825137fb511SHong Zhang     for (i=0; i<n; i++) {
826137fb511SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
827b051339dSHong Zhang       d  = (a->a)[diag[i]];
828137fb511SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
829b051339dSHong Zhang       v  = a->a+ai[i];
830b051339dSHong Zhang       nz = ai[i+1] - ai[i];
8312205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
832137fb511SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
833137fb511SHong Zhang     }
834137fb511SHong Zhang     if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot;
835137fb511SHong Zhang     sctx.shift_top *= 1.1;
836137fb511SHong Zhang     sctx.nshift_max = 5;
837137fb511SHong Zhang     sctx.shift_lo   = 0.;
838137fb511SHong Zhang     sctx.shift_hi   = 1.;
839137fb511SHong Zhang   }
840137fb511SHong Zhang 
84175567043SBarry Smith   sctx.shift_amount = 0.;
842137fb511SHong Zhang   sctx.nshift       = 0;
843504a3fadSHong Zhang #endif
844504a3fadSHong Zhang 
845137fb511SHong Zhang   do {
84607b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
847137fb511SHong Zhang     for (i=0; i<n; i++) {
848b051339dSHong Zhang       /* load in initial unfactored row */
849b051339dSHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
850b051339dSHong Zhang       ajtmp = aj + ai[r[i]];
851b051339dSHong Zhang       v     = a->a + ai[r[i]];
852137fb511SHong Zhang       /* sort permuted ajtmp and values v accordingly */
853b051339dSHong Zhang       for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]];
854137fb511SHong Zhang       ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr);
855137fb511SHong Zhang 
856b051339dSHong Zhang       diag[r[i]] = ai[r[i]];
857137fb511SHong Zhang       for (j=0; j<nz; j++) {
858137fb511SHong Zhang         rtmp[ajtmp[j]] = v[j];
859b051339dSHong Zhang         if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */
860137fb511SHong Zhang       }
861137fb511SHong Zhang       rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */
862137fb511SHong Zhang 
863b051339dSHong Zhang       row = *ajtmp++;
864137fb511SHong Zhang       while  (row < i) {
865137fb511SHong Zhang         pc = rtmp + row;
866137fb511SHong Zhang         if (*pc != 0.0) {
867b051339dSHong Zhang           pv = a->a + diag[r[row]];
868b051339dSHong Zhang           pj = aj + diag[r[row]] + 1;
869137fb511SHong Zhang 
870137fb511SHong Zhang           multiplier = *pc / *pv++;
871137fb511SHong Zhang           *pc        = multiplier;
872b051339dSHong Zhang           nz         = ai[r[row]+1] - diag[r[row]] - 1;
873b051339dSHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
87428b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
875137fb511SHong Zhang         }
876b051339dSHong Zhang         row = *ajtmp++;
877137fb511SHong Zhang       }
878b051339dSHong Zhang       /* finished row so overwrite it onto a->a */
879b051339dSHong Zhang       pv     = a->a + ai[r[i]];
880b051339dSHong Zhang       pj     = aj + ai[r[i]];
881b051339dSHong Zhang       nz     = ai[r[i]+1] - ai[r[i]];
882b051339dSHong Zhang       nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */
883137fb511SHong Zhang 
884137fb511SHong Zhang       rs = 0.0;
885137fb511SHong Zhang       for (j=0; j<nz; j++) {
886b051339dSHong Zhang         pv[j] = rtmp[pj[j]];
887b051339dSHong Zhang         if (j != nbdiag) rs += PetscAbsScalar(pv[j]);
888137fb511SHong Zhang       }
889137fb511SHong Zhang 
890137fb511SHong Zhang       sctx.rs = rs;
891b051339dSHong Zhang       sctx.pv = pv[nbdiag];
8924cccfbddSHong Zhang       ierr    = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr);
89307b50cabSHong Zhang       if (sctx.newshift) break;
894504a3fadSHong Zhang       pv[nbdiag] = sctx.pv;
895137fb511SHong Zhang     }
896137fb511SHong Zhang 
89707b50cabSHong Zhang     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
898137fb511SHong Zhang       /*
899137fb511SHong Zhang        * if no shift in this attempt & shifting & started shifting & can refine,
900137fb511SHong Zhang        * then try lower shift
901137fb511SHong Zhang        */
9020481f469SBarry Smith       sctx.shift_hi       = sctx.shift_fraction;
9030481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
9040481f469SBarry Smith       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
90507b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
906137fb511SHong Zhang       sctx.nshift++;
907137fb511SHong Zhang     }
90807b50cabSHong Zhang   } while (sctx.newshift);
909137fb511SHong Zhang 
910137fb511SHong Zhang   /* invert diagonal entries for simplier triangular solves */
911137fb511SHong Zhang   for (i=0; i<n; i++) {
912b051339dSHong Zhang     a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]];
913137fb511SHong Zhang   }
914137fb511SHong Zhang 
915137fb511SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
916137fb511SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
917137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
9182205254eSKarl Rupp 
919b051339dSHong Zhang   A->ops->solve             = MatSolve_SeqAIJ_InplaceWithPerm;
920ad04f41aSHong Zhang   A->ops->solveadd          = MatSolveAdd_SeqAIJ_inplace;
921ad04f41aSHong Zhang   A->ops->solvetranspose    = MatSolveTranspose_SeqAIJ_inplace;
922ad04f41aSHong Zhang   A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace;
9232205254eSKarl Rupp 
924b051339dSHong Zhang   A->assembled    = PETSC_TRUE;
9255c9eb25fSBarry Smith   A->preallocated = PETSC_TRUE;
9262205254eSKarl Rupp 
927d0f46423SBarry Smith   ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr);
928137fb511SHong Zhang   if (sctx.nshift) {
929f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
93057622a8eSBarry 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);
931f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
93257622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
933137fb511SHong Zhang     }
934137fb511SHong Zhang   }
935137fb511SHong Zhang   PetscFunctionReturn(0);
936137fb511SHong Zhang }
937137fb511SHong Zhang 
9380a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
9390481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info)
940da3a660dSBarry Smith {
941dfbe8321SBarry Smith   PetscErrorCode ierr;
942416022c9SBarry Smith   Mat            C;
943416022c9SBarry Smith 
9443a40ed3dSBarry Smith   PetscFunctionBegin;
9452692d6eeSBarry Smith   ierr = MatGetFactor(A,MATSOLVERPETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr);
946719d5645SBarry Smith   ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr);
947719d5645SBarry Smith   ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr);
9482205254eSKarl Rupp 
949db4efbfdSBarry Smith   A->ops->solve          = C->ops->solve;
950db4efbfdSBarry Smith   A->ops->solvetranspose = C->ops->solvetranspose;
9512205254eSKarl Rupp 
95228be2f97SBarry Smith   ierr = MatHeaderMerge(A,&C);CHKERRQ(ierr);
9533bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)A,(PetscObject)((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr);
9543a40ed3dSBarry Smith   PetscFunctionReturn(0);
955da3a660dSBarry Smith }
9560a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
9571d098868SHong Zhang 
958889a8dedSBarry Smith 
959ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_inplace(Mat A,Vec bb,Vec xx)
9608c37ef55SBarry Smith {
961416022c9SBarry Smith   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
962416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
9636849ba73SBarry Smith   PetscErrorCode    ierr;
9645d0c19d7SBarry Smith   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
9655d0c19d7SBarry Smith   PetscInt          nz;
9665d0c19d7SBarry Smith   const PetscInt    *rout,*cout,*r,*c;
967dd6ea824SBarry Smith   PetscScalar       *x,*tmp,*tmps,sum;
968d9fead3dSBarry Smith   const PetscScalar *b;
969dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
9708c37ef55SBarry Smith 
9713a40ed3dSBarry Smith   PetscFunctionBegin;
9723a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
97391bf9adeSBarry Smith 
9743649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
9751ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
976416022c9SBarry Smith   tmp  = a->solve_work;
9778c37ef55SBarry Smith 
9783b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
9793b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
9808c37ef55SBarry Smith 
9818c37ef55SBarry Smith   /* forward solve the lower triangular */
9828c37ef55SBarry Smith   tmp[0] = b[*r++];
983010a20caSHong Zhang   tmps   = tmp;
9848c37ef55SBarry Smith   for (i=1; i<n; i++) {
985010a20caSHong Zhang     v   = aa + ai[i];
986010a20caSHong Zhang     vi  = aj + ai[i];
98753e7425aSBarry Smith     nz  = a->diag[i] - ai[i];
98853e7425aSBarry Smith     sum = b[*r++];
989003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
9908c37ef55SBarry Smith     tmp[i] = sum;
9918c37ef55SBarry Smith   }
9928c37ef55SBarry Smith 
9938c37ef55SBarry Smith   /* backward solve the upper triangular */
9948c37ef55SBarry Smith   for (i=n-1; i>=0; i--) {
995010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
996010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
997416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
9988c37ef55SBarry Smith     sum = tmp[i];
999003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1000010a20caSHong Zhang     x[*c--] = tmp[i] = sum*aa[a->diag[i]];
10018c37ef55SBarry Smith   }
10028c37ef55SBarry Smith 
10033b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
10043b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
10053649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
10061ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1007dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
10083a40ed3dSBarry Smith   PetscFunctionReturn(0);
10098c37ef55SBarry Smith }
1010026e39d0SSatish Balay 
1011ad04f41aSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ_inplace(Mat A,Mat B,Mat X)
10122bebee5dSHong Zhang {
10132bebee5dSHong Zhang   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
10142bebee5dSHong Zhang   IS                iscol = a->col,isrow = a->row;
10152bebee5dSHong Zhang   PetscErrorCode    ierr;
10165d0c19d7SBarry Smith   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
10175d0c19d7SBarry Smith   PetscInt          nz,neq;
10185d0c19d7SBarry Smith   const PetscInt    *rout,*cout,*r,*c;
1019*1683a169SBarry Smith   PetscScalar       *x,*tmp,*tmps,sum;
1020*1683a169SBarry Smith   const PetscScalar *aa = a->a,*v;
1021*1683a169SBarry Smith   const PetscScalar *b;
1022ace3abfcSBarry Smith   PetscBool         bisdense,xisdense;
10232bebee5dSHong Zhang 
10242bebee5dSHong Zhang   PetscFunctionBegin;
10252bebee5dSHong Zhang   if (!n) PetscFunctionReturn(0);
10262bebee5dSHong Zhang 
1027251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr);
1028e32f2f54SBarry Smith   if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix");
1029251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr);
1030e32f2f54SBarry Smith   if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix");
1031db4efbfdSBarry Smith 
1032*1683a169SBarry Smith   ierr = MatDenseGetArrayRead(B,&b);CHKERRQ(ierr);
10338c778c55SBarry Smith   ierr = MatDenseGetArray(X,&x);CHKERRQ(ierr);
10342bebee5dSHong Zhang 
10352bebee5dSHong Zhang   tmp  = a->solve_work;
10362bebee5dSHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
10372bebee5dSHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
10382bebee5dSHong Zhang 
1039a36b22ccSSatish Balay   for (neq=0; neq<B->cmap->n; neq++) {
10402bebee5dSHong Zhang     /* forward solve the lower triangular */
10412bebee5dSHong Zhang     tmp[0] = b[r[0]];
10422bebee5dSHong Zhang     tmps   = tmp;
10432bebee5dSHong Zhang     for (i=1; i<n; i++) {
10442bebee5dSHong Zhang       v   = aa + ai[i];
10452bebee5dSHong Zhang       vi  = aj + ai[i];
10462bebee5dSHong Zhang       nz  = a->diag[i] - ai[i];
10472bebee5dSHong Zhang       sum = b[r[i]];
1048003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10492bebee5dSHong Zhang       tmp[i] = sum;
10502bebee5dSHong Zhang     }
10512bebee5dSHong Zhang     /* backward solve the upper triangular */
10522bebee5dSHong Zhang     for (i=n-1; i>=0; i--) {
10532bebee5dSHong Zhang       v   = aa + a->diag[i] + 1;
10542bebee5dSHong Zhang       vi  = aj + a->diag[i] + 1;
10552bebee5dSHong Zhang       nz  = ai[i+1] - a->diag[i] - 1;
10562bebee5dSHong Zhang       sum = tmp[i];
1057003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10582bebee5dSHong Zhang       x[c[i]] = tmp[i] = sum*aa[a->diag[i]];
10592bebee5dSHong Zhang     }
10602bebee5dSHong Zhang 
10612bebee5dSHong Zhang     b += n;
10622bebee5dSHong Zhang     x += n;
10632bebee5dSHong Zhang   }
10642bebee5dSHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
10652bebee5dSHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1066*1683a169SBarry Smith   ierr = MatDenseRestoreArrayRead(B,&b);CHKERRQ(ierr);
10678c778c55SBarry Smith   ierr = MatDenseRestoreArray(X,&x);CHKERRQ(ierr);
1068dc0b31edSSatish Balay   ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr);
10692bebee5dSHong Zhang   PetscFunctionReturn(0);
10702bebee5dSHong Zhang }
10712bebee5dSHong Zhang 
107235233d99SShri Abhyankar PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X)
10739bd0847aSShri Abhyankar {
10749bd0847aSShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
10759bd0847aSShri Abhyankar   IS                iscol = a->col,isrow = a->row;
10769bd0847aSShri Abhyankar   PetscErrorCode    ierr;
10779bd0847aSShri Abhyankar   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag;
10789bd0847aSShri Abhyankar   PetscInt          nz,neq;
10799bd0847aSShri Abhyankar   const PetscInt    *rout,*cout,*r,*c;
1080*1683a169SBarry Smith   PetscScalar       *x,*tmp,sum;
1081*1683a169SBarry Smith   const PetscScalar *b;
1082*1683a169SBarry Smith   const PetscScalar *aa = a->a,*v;
1083ace3abfcSBarry Smith   PetscBool         bisdense,xisdense;
10849bd0847aSShri Abhyankar 
10859bd0847aSShri Abhyankar   PetscFunctionBegin;
10869bd0847aSShri Abhyankar   if (!n) PetscFunctionReturn(0);
10879bd0847aSShri Abhyankar 
1088251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr);
1089e32f2f54SBarry Smith   if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix");
1090251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr);
1091e32f2f54SBarry Smith   if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix");
10929bd0847aSShri Abhyankar 
1093*1683a169SBarry Smith   ierr = MatDenseGetArrayRead(B,&b);CHKERRQ(ierr);
10948c778c55SBarry Smith   ierr = MatDenseGetArray(X,&x);CHKERRQ(ierr);
10959bd0847aSShri Abhyankar 
10969bd0847aSShri Abhyankar   tmp  = a->solve_work;
10979bd0847aSShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
10989bd0847aSShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
10999bd0847aSShri Abhyankar 
11009bd0847aSShri Abhyankar   for (neq=0; neq<B->cmap->n; neq++) {
11019bd0847aSShri Abhyankar     /* forward solve the lower triangular */
11029bd0847aSShri Abhyankar     tmp[0] = b[r[0]];
11039bd0847aSShri Abhyankar     v      = aa;
11049bd0847aSShri Abhyankar     vi     = aj;
11059bd0847aSShri Abhyankar     for (i=1; i<n; i++) {
11069bd0847aSShri Abhyankar       nz  = ai[i+1] - ai[i];
11079bd0847aSShri Abhyankar       sum = b[r[i]];
11089bd0847aSShri Abhyankar       PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
11099bd0847aSShri Abhyankar       tmp[i] = sum;
11109bd0847aSShri Abhyankar       v     += nz; vi += nz;
11119bd0847aSShri Abhyankar     }
11129bd0847aSShri Abhyankar 
11139bd0847aSShri Abhyankar     /* backward solve the upper triangular */
11149bd0847aSShri Abhyankar     for (i=n-1; i>=0; i--) {
11159bd0847aSShri Abhyankar       v   = aa + adiag[i+1]+1;
11169bd0847aSShri Abhyankar       vi  = aj + adiag[i+1]+1;
11179bd0847aSShri Abhyankar       nz  = adiag[i]-adiag[i+1]-1;
11189bd0847aSShri Abhyankar       sum = tmp[i];
11199bd0847aSShri Abhyankar       PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
11209bd0847aSShri Abhyankar       x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */
11219bd0847aSShri Abhyankar     }
11229bd0847aSShri Abhyankar 
11239bd0847aSShri Abhyankar     b += n;
11249bd0847aSShri Abhyankar     x += n;
11259bd0847aSShri Abhyankar   }
11269bd0847aSShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
11279bd0847aSShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1128*1683a169SBarry Smith   ierr = MatDenseRestoreArrayRead(B,&b);CHKERRQ(ierr);
11298c778c55SBarry Smith   ierr = MatDenseRestoreArray(X,&x);CHKERRQ(ierr);
11309bd0847aSShri Abhyankar   ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr);
11319bd0847aSShri Abhyankar   PetscFunctionReturn(0);
11329bd0847aSShri Abhyankar }
11339bd0847aSShri Abhyankar 
1134137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx)
1135137fb511SHong Zhang {
1136137fb511SHong Zhang   Mat_SeqAIJ      *a    = (Mat_SeqAIJ*)A->data;
1137137fb511SHong Zhang   IS              iscol = a->col,isrow = a->row;
1138137fb511SHong Zhang   PetscErrorCode  ierr;
11395d0c19d7SBarry Smith   const PetscInt  *r,*c,*rout,*cout;
11405d0c19d7SBarry Smith   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
11415d0c19d7SBarry Smith   PetscInt        nz,row;
1142dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,*tmps,sum;
1143dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
1144137fb511SHong Zhang 
1145137fb511SHong Zhang   PetscFunctionBegin;
1146137fb511SHong Zhang   if (!n) PetscFunctionReturn(0);
1147137fb511SHong Zhang 
1148137fb511SHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
1149137fb511SHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1150137fb511SHong Zhang   tmp  = a->solve_work;
1151137fb511SHong Zhang 
1152137fb511SHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1153137fb511SHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1154137fb511SHong Zhang 
1155137fb511SHong Zhang   /* forward solve the lower triangular */
1156137fb511SHong Zhang   tmp[0] = b[*r++];
1157137fb511SHong Zhang   tmps   = tmp;
1158137fb511SHong Zhang   for (row=1; row<n; row++) {
1159137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1160137fb511SHong Zhang     v   = aa + ai[i];
1161137fb511SHong Zhang     vi  = aj + ai[i];
1162137fb511SHong Zhang     nz  = a->diag[i] - ai[i];
1163137fb511SHong Zhang     sum = b[*r++];
1164003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1165137fb511SHong Zhang     tmp[row] = sum;
1166137fb511SHong Zhang   }
1167137fb511SHong Zhang 
1168137fb511SHong Zhang   /* backward solve the upper triangular */
1169137fb511SHong Zhang   for (row=n-1; row>=0; row--) {
1170137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1171137fb511SHong Zhang     v   = aa + a->diag[i] + 1;
1172137fb511SHong Zhang     vi  = aj + a->diag[i] + 1;
1173137fb511SHong Zhang     nz  = ai[i+1] - a->diag[i] - 1;
1174137fb511SHong Zhang     sum = tmp[row];
1175003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1176137fb511SHong Zhang     x[*c--] = tmp[row] = sum*aa[a->diag[i]];
1177137fb511SHong Zhang   }
1178137fb511SHong Zhang 
1179137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1180137fb511SHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1181137fb511SHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
1182137fb511SHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1183dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
1184137fb511SHong Zhang   PetscFunctionReturn(0);
1185137fb511SHong Zhang }
1186137fb511SHong Zhang 
1187930ae53cSBarry Smith /* ----------------------------------------------------------- */
1188c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/ftn-kernels/fsolve.h>
1189ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_inplace(Mat A,Vec bb,Vec xx)
1190930ae53cSBarry Smith {
1191930ae53cSBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
11926849ba73SBarry Smith   PetscErrorCode    ierr;
1193d0f46423SBarry Smith   PetscInt          n   = A->rmap->n;
1194b377110cSBarry Smith   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag;
1195356650c2SBarry Smith   PetscScalar       *x;
1196356650c2SBarry Smith   const PetscScalar *b;
1197dd6ea824SBarry Smith   const MatScalar   *aa = a->a;
1198aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
1199356650c2SBarry Smith   PetscInt        adiag_i,i,nz,ai_i;
1200b377110cSBarry Smith   const PetscInt  *vi;
1201dd6ea824SBarry Smith   const MatScalar *v;
1202dd6ea824SBarry Smith   PetscScalar     sum;
1203d85afc42SSatish Balay #endif
1204930ae53cSBarry Smith 
12053a40ed3dSBarry Smith   PetscFunctionBegin;
12063a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
1207930ae53cSBarry Smith 
12083649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
12091ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1210930ae53cSBarry Smith 
1211aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
12121c4feecaSSatish Balay   fortransolveaij_(&n,x,ai,aj,adiag,aa,b);
12131c4feecaSSatish Balay #else
1214930ae53cSBarry Smith   /* forward solve the lower triangular */
1215930ae53cSBarry Smith   x[0] = b[0];
1216930ae53cSBarry Smith   for (i=1; i<n; i++) {
1217930ae53cSBarry Smith     ai_i = ai[i];
1218930ae53cSBarry Smith     v    = aa + ai_i;
1219930ae53cSBarry Smith     vi   = aj + ai_i;
1220930ae53cSBarry Smith     nz   = adiag[i] - ai_i;
1221930ae53cSBarry Smith     sum  = b[i];
1222003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1223930ae53cSBarry Smith     x[i] = sum;
1224930ae53cSBarry Smith   }
1225930ae53cSBarry Smith 
1226930ae53cSBarry Smith   /* backward solve the upper triangular */
1227930ae53cSBarry Smith   for (i=n-1; i>=0; i--) {
1228930ae53cSBarry Smith     adiag_i = adiag[i];
1229d708749eSBarry Smith     v       = aa + adiag_i + 1;
1230d708749eSBarry Smith     vi      = aj + adiag_i + 1;
1231930ae53cSBarry Smith     nz      = ai[i+1] - adiag_i - 1;
1232930ae53cSBarry Smith     sum     = x[i];
1233003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1234930ae53cSBarry Smith     x[i] = sum*aa[adiag_i];
1235930ae53cSBarry Smith   }
12361c4feecaSSatish Balay #endif
1237dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
12383649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
12391ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
12403a40ed3dSBarry Smith   PetscFunctionReturn(0);
1241930ae53cSBarry Smith }
1242930ae53cSBarry Smith 
1243ad04f41aSHong Zhang PetscErrorCode MatSolveAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec yy,Vec xx)
1244da3a660dSBarry Smith {
1245416022c9SBarry Smith   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
1246416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
12476849ba73SBarry Smith   PetscErrorCode    ierr;
124804fbf559SBarry Smith   PetscInt          i, n = A->rmap->n,j;
12495d0c19d7SBarry Smith   PetscInt          nz;
125004fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j;
125104fbf559SBarry Smith   PetscScalar       *x,*tmp,sum;
125204fbf559SBarry Smith   const PetscScalar *b;
1253dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
1254da3a660dSBarry Smith 
12553a40ed3dSBarry Smith   PetscFunctionBegin;
125678b31e54SBarry Smith   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
1257da3a660dSBarry Smith 
12583649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
12591ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1260416022c9SBarry Smith   tmp  = a->solve_work;
1261da3a660dSBarry Smith 
12623b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
12633b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1264da3a660dSBarry Smith 
1265da3a660dSBarry Smith   /* forward solve the lower triangular */
1266da3a660dSBarry Smith   tmp[0] = b[*r++];
1267da3a660dSBarry Smith   for (i=1; i<n; i++) {
1268010a20caSHong Zhang     v   = aa + ai[i];
1269010a20caSHong Zhang     vi  = aj + ai[i];
1270416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
1271da3a660dSBarry Smith     sum = b[*r++];
127204fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1273da3a660dSBarry Smith     tmp[i] = sum;
1274da3a660dSBarry Smith   }
1275da3a660dSBarry Smith 
1276da3a660dSBarry Smith   /* backward solve the upper triangular */
1277da3a660dSBarry Smith   for (i=n-1; i>=0; i--) {
1278010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
1279010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
1280416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
1281da3a660dSBarry Smith     sum = tmp[i];
128204fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1283010a20caSHong Zhang     tmp[i]   = sum*aa[a->diag[i]];
1284da3a660dSBarry Smith     x[*c--] += tmp[i];
1285da3a660dSBarry Smith   }
1286da3a660dSBarry Smith 
12873b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
12883b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
12893649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
12901ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1291dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
12923a40ed3dSBarry Smith   PetscFunctionReturn(0);
1293da3a660dSBarry Smith }
129404fbf559SBarry Smith 
129535233d99SShri Abhyankar PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx)
12963c0119dfSShri Abhyankar {
12973c0119dfSShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
12983c0119dfSShri Abhyankar   IS                iscol = a->col,isrow = a->row;
12993c0119dfSShri Abhyankar   PetscErrorCode    ierr;
13003c0119dfSShri Abhyankar   PetscInt          i, n = A->rmap->n,j;
13013c0119dfSShri Abhyankar   PetscInt          nz;
13023c0119dfSShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag;
13033c0119dfSShri Abhyankar   PetscScalar       *x,*tmp,sum;
13043c0119dfSShri Abhyankar   const PetscScalar *b;
13053c0119dfSShri Abhyankar   const MatScalar   *aa = a->a,*v;
13063c0119dfSShri Abhyankar 
13073c0119dfSShri Abhyankar   PetscFunctionBegin;
13083c0119dfSShri Abhyankar   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
13093c0119dfSShri Abhyankar 
13103649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
13113c0119dfSShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
13123c0119dfSShri Abhyankar   tmp  = a->solve_work;
13133c0119dfSShri Abhyankar 
13143c0119dfSShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
13153c0119dfSShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
13163c0119dfSShri Abhyankar 
13173c0119dfSShri Abhyankar   /* forward solve the lower triangular */
13183c0119dfSShri Abhyankar   tmp[0] = b[r[0]];
13193c0119dfSShri Abhyankar   v      = aa;
13203c0119dfSShri Abhyankar   vi     = aj;
13213c0119dfSShri Abhyankar   for (i=1; i<n; i++) {
13223c0119dfSShri Abhyankar     nz  = ai[i+1] - ai[i];
13233c0119dfSShri Abhyankar     sum = b[r[i]];
13243c0119dfSShri Abhyankar     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
13253c0119dfSShri Abhyankar     tmp[i] = sum;
13262205254eSKarl Rupp     v     += nz;
13272205254eSKarl Rupp     vi    += nz;
13283c0119dfSShri Abhyankar   }
13293c0119dfSShri Abhyankar 
13303c0119dfSShri Abhyankar   /* backward solve the upper triangular */
13313c0119dfSShri Abhyankar   v  = aa + adiag[n-1];
13323c0119dfSShri Abhyankar   vi = aj + adiag[n-1];
13333c0119dfSShri Abhyankar   for (i=n-1; i>=0; i--) {
13343c0119dfSShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
13353c0119dfSShri Abhyankar     sum = tmp[i];
13363c0119dfSShri Abhyankar     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
13373c0119dfSShri Abhyankar     tmp[i]   = sum*v[nz];
13383c0119dfSShri Abhyankar     x[c[i]] += tmp[i];
13393c0119dfSShri Abhyankar     v       += nz+1; vi += nz+1;
13403c0119dfSShri Abhyankar   }
13413c0119dfSShri Abhyankar 
13423c0119dfSShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
13433c0119dfSShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
13443649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
13453c0119dfSShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
13463c0119dfSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
13473c0119dfSShri Abhyankar   PetscFunctionReturn(0);
13483c0119dfSShri Abhyankar }
13493c0119dfSShri Abhyankar 
1350ad04f41aSHong Zhang PetscErrorCode MatSolveTranspose_SeqAIJ_inplace(Mat A,Vec bb,Vec xx)
1351da3a660dSBarry Smith {
1352416022c9SBarry Smith   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
13532235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
13546849ba73SBarry Smith   PetscErrorCode    ierr;
135504fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
135604fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
135704fbf559SBarry Smith   PetscInt          nz;
135804fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1359dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
136004fbf559SBarry Smith   const PetscScalar *b;
1361da3a660dSBarry Smith 
13623a40ed3dSBarry Smith   PetscFunctionBegin;
13633649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
13641ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1365416022c9SBarry Smith   tmp  = a->solve_work;
1366da3a660dSBarry Smith 
13672235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
13682235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1369da3a660dSBarry Smith 
1370da3a660dSBarry Smith   /* copy the b into temp work space according to permutation */
13712235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1372da3a660dSBarry Smith 
1373da3a660dSBarry Smith   /* forward solve the U^T */
1374da3a660dSBarry Smith   for (i=0; i<n; i++) {
1375010a20caSHong Zhang     v   = aa + diag[i];
1376010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1377f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
1378c38d4ed2SBarry Smith     s1  = tmp[i];
1379ef66eb69SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
138004fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1381c38d4ed2SBarry Smith     tmp[i] = s1;
1382da3a660dSBarry Smith   }
1383da3a660dSBarry Smith 
1384da3a660dSBarry Smith   /* backward solve the L^T */
1385da3a660dSBarry Smith   for (i=n-1; i>=0; i--) {
1386010a20caSHong Zhang     v  = aa + diag[i] - 1;
1387010a20caSHong Zhang     vi = aj + diag[i] - 1;
1388f1af5d2fSBarry Smith     nz = diag[i] - ai[i];
1389f1af5d2fSBarry Smith     s1 = tmp[i];
139004fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
1391da3a660dSBarry Smith   }
1392da3a660dSBarry Smith 
1393da3a660dSBarry Smith   /* copy tmp into x according to permutation */
1394591294e4SBarry Smith   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1395da3a660dSBarry Smith 
13962235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
13972235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
13983649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
13991ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1400da3a660dSBarry Smith 
1401dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
14023a40ed3dSBarry Smith   PetscFunctionReturn(0);
1403da3a660dSBarry Smith }
1404da3a660dSBarry Smith 
140535233d99SShri Abhyankar PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx)
1406d1fa9404SShri Abhyankar {
1407d1fa9404SShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
1408d1fa9404SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
1409d1fa9404SShri Abhyankar   PetscErrorCode    ierr;
1410d1fa9404SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi;
1411d1fa9404SShri Abhyankar   PetscInt          i,n = A->rmap->n,j;
1412d1fa9404SShri Abhyankar   PetscInt          nz;
1413d1fa9404SShri Abhyankar   PetscScalar       *x,*tmp,s1;
1414d1fa9404SShri Abhyankar   const MatScalar   *aa = a->a,*v;
1415d1fa9404SShri Abhyankar   const PetscScalar *b;
1416d1fa9404SShri Abhyankar 
1417d1fa9404SShri Abhyankar   PetscFunctionBegin;
14183649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
1419d1fa9404SShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1420d1fa9404SShri Abhyankar   tmp  = a->solve_work;
1421d1fa9404SShri Abhyankar 
1422d1fa9404SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1423d1fa9404SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1424d1fa9404SShri Abhyankar 
1425d1fa9404SShri Abhyankar   /* copy the b into temp work space according to permutation */
1426d1fa9404SShri Abhyankar   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1427d1fa9404SShri Abhyankar 
1428d1fa9404SShri Abhyankar   /* forward solve the U^T */
1429d1fa9404SShri Abhyankar   for (i=0; i<n; i++) {
1430d1fa9404SShri Abhyankar     v   = aa + adiag[i+1] + 1;
1431d1fa9404SShri Abhyankar     vi  = aj + adiag[i+1] + 1;
1432d1fa9404SShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
1433d1fa9404SShri Abhyankar     s1  = tmp[i];
1434d1fa9404SShri Abhyankar     s1 *= v[nz];  /* multiply by inverse of diagonal entry */
1435d1fa9404SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1436d1fa9404SShri Abhyankar     tmp[i] = s1;
1437d1fa9404SShri Abhyankar   }
1438d1fa9404SShri Abhyankar 
1439d1fa9404SShri Abhyankar   /* backward solve the L^T */
1440d1fa9404SShri Abhyankar   for (i=n-1; i>=0; i--) {
1441d1fa9404SShri Abhyankar     v  = aa + ai[i];
1442d1fa9404SShri Abhyankar     vi = aj + ai[i];
1443d1fa9404SShri Abhyankar     nz = ai[i+1] - ai[i];
1444d1fa9404SShri Abhyankar     s1 = tmp[i];
1445d1fa9404SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1446d1fa9404SShri Abhyankar   }
1447d1fa9404SShri Abhyankar 
1448d1fa9404SShri Abhyankar   /* copy tmp into x according to permutation */
1449d1fa9404SShri Abhyankar   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1450d1fa9404SShri Abhyankar 
1451d1fa9404SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1452d1fa9404SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
14533649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1454d1fa9404SShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1455d1fa9404SShri Abhyankar 
1456d1fa9404SShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
1457d1fa9404SShri Abhyankar   PetscFunctionReturn(0);
1458d1fa9404SShri Abhyankar }
1459d1fa9404SShri Abhyankar 
1460ad04f41aSHong Zhang PetscErrorCode MatSolveTransposeAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec zz,Vec xx)
1461da3a660dSBarry Smith {
1462416022c9SBarry Smith   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
14632235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
14646849ba73SBarry Smith   PetscErrorCode    ierr;
146504fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
146604fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
146704fbf559SBarry Smith   PetscInt          nz;
146804fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1469dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
147004fbf559SBarry Smith   const PetscScalar *b;
14716abc6512SBarry Smith 
14723a40ed3dSBarry Smith   PetscFunctionBegin;
147323bc6035SMatthew Knepley   if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);}
14743649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
14751ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1476416022c9SBarry Smith   tmp  = a->solve_work;
14776abc6512SBarry Smith 
14782235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
14792235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
14806abc6512SBarry Smith 
14816abc6512SBarry Smith   /* copy the b into temp work space according to permutation */
14822235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
14836abc6512SBarry Smith 
14846abc6512SBarry Smith   /* forward solve the U^T */
14856abc6512SBarry Smith   for (i=0; i<n; i++) {
1486010a20caSHong Zhang     v   = aa + diag[i];
1487010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1488f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
148904fbf559SBarry Smith     s1  = tmp[i];
149004fbf559SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
149104fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
149204fbf559SBarry Smith     tmp[i] = s1;
14936abc6512SBarry Smith   }
14946abc6512SBarry Smith 
14956abc6512SBarry Smith   /* backward solve the L^T */
14966abc6512SBarry Smith   for (i=n-1; i>=0; i--) {
1497010a20caSHong Zhang     v  = aa + diag[i] - 1;
1498010a20caSHong Zhang     vi = aj + diag[i] - 1;
1499f1af5d2fSBarry Smith     nz = diag[i] - ai[i];
150004fbf559SBarry Smith     s1 = tmp[i];
150104fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
15026abc6512SBarry Smith   }
15036abc6512SBarry Smith 
15046abc6512SBarry Smith   /* copy tmp into x according to permutation */
15056abc6512SBarry Smith   for (i=0; i<n; i++) x[r[i]] += tmp[i];
15066abc6512SBarry Smith 
15072235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
15082235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
15093649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
15101ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
15116abc6512SBarry Smith 
151204fbf559SBarry Smith   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
15133a40ed3dSBarry Smith   PetscFunctionReturn(0);
1514da3a660dSBarry Smith }
151504fbf559SBarry Smith 
151635233d99SShri Abhyankar PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx)
1517533e6011SShri Abhyankar {
1518533e6011SShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
1519533e6011SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
1520533e6011SShri Abhyankar   PetscErrorCode    ierr;
1521533e6011SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi;
1522533e6011SShri Abhyankar   PetscInt          i,n = A->rmap->n,j;
1523533e6011SShri Abhyankar   PetscInt          nz;
1524533e6011SShri Abhyankar   PetscScalar       *x,*tmp,s1;
1525533e6011SShri Abhyankar   const MatScalar   *aa = a->a,*v;
1526533e6011SShri Abhyankar   const PetscScalar *b;
1527533e6011SShri Abhyankar 
1528533e6011SShri Abhyankar   PetscFunctionBegin;
1529533e6011SShri Abhyankar   if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);}
15303649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
1531533e6011SShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1532533e6011SShri Abhyankar   tmp  = a->solve_work;
1533533e6011SShri Abhyankar 
1534533e6011SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1535533e6011SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1536533e6011SShri Abhyankar 
1537533e6011SShri Abhyankar   /* copy the b into temp work space according to permutation */
1538533e6011SShri Abhyankar   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1539533e6011SShri Abhyankar 
1540533e6011SShri Abhyankar   /* forward solve the U^T */
1541533e6011SShri Abhyankar   for (i=0; i<n; i++) {
1542533e6011SShri Abhyankar     v   = aa + adiag[i+1] + 1;
1543533e6011SShri Abhyankar     vi  = aj + adiag[i+1] + 1;
1544533e6011SShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
1545533e6011SShri Abhyankar     s1  = tmp[i];
1546533e6011SShri Abhyankar     s1 *= v[nz];  /* multiply by inverse of diagonal entry */
1547533e6011SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1548533e6011SShri Abhyankar     tmp[i] = s1;
1549533e6011SShri Abhyankar   }
1550533e6011SShri Abhyankar 
1551533e6011SShri Abhyankar 
1552533e6011SShri Abhyankar   /* backward solve the L^T */
1553533e6011SShri Abhyankar   for (i=n-1; i>=0; i--) {
1554533e6011SShri Abhyankar     v  = aa + ai[i];
1555533e6011SShri Abhyankar     vi = aj + ai[i];
1556533e6011SShri Abhyankar     nz = ai[i+1] - ai[i];
1557533e6011SShri Abhyankar     s1 = tmp[i];
1558c5e3b2a3SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1559533e6011SShri Abhyankar   }
1560533e6011SShri Abhyankar 
1561533e6011SShri Abhyankar   /* copy tmp into x according to permutation */
1562533e6011SShri Abhyankar   for (i=0; i<n; i++) x[r[i]] += tmp[i];
1563533e6011SShri Abhyankar 
1564533e6011SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1565533e6011SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
15663649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1567533e6011SShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1568533e6011SShri Abhyankar 
1569533e6011SShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
1570533e6011SShri Abhyankar   PetscFunctionReturn(0);
1571533e6011SShri Abhyankar }
1572533e6011SShri Abhyankar 
15739e25ed09SBarry Smith /* ----------------------------------------------------------------*/
1574b47f5a65SHong Zhang 
15758fc3a347SHong Zhang /*
15768a76ed4fSHong Zhang    ilu() under revised new data structure.
1577813a1f2bSShri Abhyankar    Factored arrays bj and ba are stored as
1578813a1f2bSShri Abhyankar      L(0,:), L(1,:), ...,L(n-1,:),  U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:)
1579813a1f2bSShri Abhyankar 
1580813a1f2bSShri Abhyankar    bi=fact->i is an array of size n+1, in which
1581813a1f2bSShri Abhyankar    bi+
1582baabb860SHong Zhang      bi[i]:  points to 1st entry of L(i,:),i=0,...,n-1
1583baabb860SHong Zhang      bi[n]:  points to L(n-1,n-1)+1
1584baabb860SHong Zhang 
1585813a1f2bSShri Abhyankar   bdiag=fact->diag is an array of size n+1,in which
1586baabb860SHong Zhang      bdiag[i]: points to diagonal of U(i,:), i=0,...,n-1
1587a24f213cSHong Zhang      bdiag[n]: points to entry of U(n-1,0)-1
1588813a1f2bSShri Abhyankar 
1589813a1f2bSShri Abhyankar    U(i,:) contains bdiag[i] as its last entry, i.e.,
1590813a1f2bSShri Abhyankar     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
1591813a1f2bSShri Abhyankar */
159235233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
1593813a1f2bSShri Abhyankar {
1594813a1f2bSShri Abhyankar   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data,*b;
1595813a1f2bSShri Abhyankar   PetscErrorCode ierr;
1596bbd65245SShri Abhyankar   const PetscInt n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag;
1597bbd65245SShri Abhyankar   PetscInt       i,j,k=0,nz,*bi,*bj,*bdiag;
15981df811f5SHong Zhang   IS             isicol;
1599813a1f2bSShri Abhyankar 
1600813a1f2bSShri Abhyankar   PetscFunctionBegin;
16011df811f5SHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
1602813a1f2bSShri Abhyankar   ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr);
1603813a1f2bSShri Abhyankar   b    = (Mat_SeqAIJ*)(fact)->data;
1604813a1f2bSShri Abhyankar 
1605813a1f2bSShri Abhyankar   /* allocate matrix arrays for new data structure */
1606dcca6d9dSJed Brown   ierr = PetscMalloc3(ai[n]+1,&b->a,ai[n]+1,&b->j,n+1,&b->i);CHKERRQ(ierr);
16073bb1ff40SBarry Smith   ierr = PetscLogObjectMemory((PetscObject)fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(n+1)*sizeof(PetscInt));CHKERRQ(ierr);
16082205254eSKarl Rupp 
1609813a1f2bSShri Abhyankar   b->singlemalloc = PETSC_TRUE;
1610813a1f2bSShri Abhyankar   if (!b->diag) {
1611854ce69bSBarry Smith     ierr = PetscMalloc1(n+1,&b->diag);CHKERRQ(ierr);
16123bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)fact,(n+1)*sizeof(PetscInt));CHKERRQ(ierr);
1613813a1f2bSShri Abhyankar   }
1614813a1f2bSShri Abhyankar   bdiag = b->diag;
1615813a1f2bSShri Abhyankar 
1616813a1f2bSShri Abhyankar   if (n > 0) {
1617813a1f2bSShri Abhyankar     ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr);
1618813a1f2bSShri Abhyankar   }
1619813a1f2bSShri Abhyankar 
1620813a1f2bSShri Abhyankar   /* set bi and bj with new data structure */
1621813a1f2bSShri Abhyankar   bi = b->i;
1622813a1f2bSShri Abhyankar   bj = b->j;
1623813a1f2bSShri Abhyankar 
1624813a1f2bSShri Abhyankar   /* L part */
1625e60cf9a0SBarry Smith   bi[0] = 0;
1626813a1f2bSShri Abhyankar   for (i=0; i<n; i++) {
1627813a1f2bSShri Abhyankar     nz      = adiag[i] - ai[i];
1628813a1f2bSShri Abhyankar     bi[i+1] = bi[i] + nz;
1629813a1f2bSShri Abhyankar     aj      = a->j + ai[i];
1630813a1f2bSShri Abhyankar     for (j=0; j<nz; j++) {
1631bbd65245SShri Abhyankar       /*   *bj = aj[j]; bj++; */
1632bbd65245SShri Abhyankar       bj[k++] = aj[j];
1633813a1f2bSShri Abhyankar     }
1634813a1f2bSShri Abhyankar   }
1635813a1f2bSShri Abhyankar 
1636813a1f2bSShri Abhyankar   /* U part */
163762fbd6afSShri Abhyankar   bdiag[n] = bi[n]-1;
1638813a1f2bSShri Abhyankar   for (i=n-1; i>=0; i--) {
1639813a1f2bSShri Abhyankar     nz = ai[i+1] - adiag[i] - 1;
1640813a1f2bSShri Abhyankar     aj = a->j + adiag[i] + 1;
1641813a1f2bSShri Abhyankar     for (j=0; j<nz; j++) {
1642bbd65245SShri Abhyankar       /*      *bj = aj[j]; bj++; */
1643bbd65245SShri Abhyankar       bj[k++] = aj[j];
1644813a1f2bSShri Abhyankar     }
1645813a1f2bSShri Abhyankar     /* diag[i] */
1646bbd65245SShri Abhyankar     /*    *bj = i; bj++; */
1647bbd65245SShri Abhyankar     bj[k++]  = i;
1648a24f213cSHong Zhang     bdiag[i] = bdiag[i+1] + nz + 1;
1649813a1f2bSShri Abhyankar   }
16501df811f5SHong Zhang 
1651d5f3da31SBarry Smith   fact->factortype             = MAT_FACTOR_ILU;
16521df811f5SHong Zhang   fact->info.factor_mallocs    = 0;
16531df811f5SHong Zhang   fact->info.fill_ratio_given  = info->fill;
16541df811f5SHong Zhang   fact->info.fill_ratio_needed = 1.0;
165535233d99SShri Abhyankar   fact->ops->lufactornumeric   = MatLUFactorNumeric_SeqAIJ;
1656a02bda8eSBarry Smith   ierr = MatSeqAIJCheckInode_FactorLU(fact);CHKERRQ(ierr);
16571df811f5SHong Zhang 
16581df811f5SHong Zhang   b       = (Mat_SeqAIJ*)(fact)->data;
16591df811f5SHong Zhang   b->row  = isrow;
16601df811f5SHong Zhang   b->col  = iscol;
16611df811f5SHong Zhang   b->icol = isicol;
1662854ce69bSBarry Smith   ierr    = PetscMalloc1(fact->rmap->n+1,&b->solve_work);CHKERRQ(ierr);
16631df811f5SHong Zhang   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
16641df811f5SHong Zhang   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
1665813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
1666813a1f2bSShri Abhyankar }
1667813a1f2bSShri Abhyankar 
166835233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
1669813a1f2bSShri Abhyankar {
1670813a1f2bSShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
1671813a1f2bSShri Abhyankar   IS                 isicol;
1672813a1f2bSShri Abhyankar   PetscErrorCode     ierr;
1673813a1f2bSShri Abhyankar   const PetscInt     *r,*ic;
16741df811f5SHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j;
1675813a1f2bSShri Abhyankar   PetscInt           *bi,*cols,nnz,*cols_lvl;
1676813a1f2bSShri Abhyankar   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
1677813a1f2bSShri Abhyankar   PetscInt           i,levels,diagonal_fill;
1678035f4963SHong Zhang   PetscBool          col_identity,row_identity,missing;
1679813a1f2bSShri Abhyankar   PetscReal          f;
16800298fd71SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=NULL;
1681813a1f2bSShri Abhyankar   PetscBT            lnkbt;
1682813a1f2bSShri Abhyankar   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
16830298fd71SBarry Smith   PetscFreeSpaceList free_space    =NULL,current_space=NULL;
16840298fd71SBarry Smith   PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL;
1685813a1f2bSShri Abhyankar 
1686813a1f2bSShri Abhyankar   PetscFunctionBegin;
1687035f4963SHong 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);
1688035f4963SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
1689035f4963SHong Zhang   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
1690ad04f41aSHong Zhang 
1691813a1f2bSShri Abhyankar   levels = (PetscInt)info->levels;
1692813a1f2bSShri Abhyankar   ierr   = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
1693813a1f2bSShri Abhyankar   ierr   = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
1694813a1f2bSShri Abhyankar   if (!levels && row_identity && col_identity) {
1695813a1f2bSShri Abhyankar     /* special case: ilu(0) with natural ordering */
169635233d99SShri Abhyankar     ierr = MatILUFactorSymbolic_SeqAIJ_ilu0(fact,A,isrow,iscol,info);CHKERRQ(ierr);
1697124a3351SHong Zhang     if (a->inode.size) {
1698124a3351SHong Zhang       fact->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode;
1699124a3351SHong Zhang     }
1700813a1f2bSShri Abhyankar     PetscFunctionReturn(0);
1701813a1f2bSShri Abhyankar   }
1702813a1f2bSShri Abhyankar 
17031df811f5SHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
1704813a1f2bSShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1705813a1f2bSShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
1706813a1f2bSShri Abhyankar 
17071bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
1708854ce69bSBarry Smith   ierr  = PetscMalloc1(n+1,&bi);CHKERRQ(ierr);
1709854ce69bSBarry Smith   ierr  = PetscMalloc1(n+1,&bdiag);CHKERRQ(ierr);
1710e60cf9a0SBarry Smith   bi[0] = bdiag[0] = 0;
1711dcca6d9dSJed Brown   ierr  = PetscMalloc2(n,&bj_ptr,n,&bjlvl_ptr);CHKERRQ(ierr);
1712813a1f2bSShri Abhyankar 
1713813a1f2bSShri Abhyankar   /* create a linked list for storing column indices of the active row */
1714813a1f2bSShri Abhyankar   nlnk = n + 1;
1715813a1f2bSShri Abhyankar   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
1716813a1f2bSShri Abhyankar 
1717813a1f2bSShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
17181df811f5SHong Zhang   f                 = info->fill;
17191df811f5SHong Zhang   diagonal_fill     = (PetscInt)info->diagonal_fill;
17202f18eb33SBarry Smith   ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space);CHKERRQ(ierr);
1721813a1f2bSShri Abhyankar   current_space     = free_space;
17222f18eb33SBarry Smith   ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space_lvl);CHKERRQ(ierr);
1723813a1f2bSShri Abhyankar   current_space_lvl = free_space_lvl;
1724813a1f2bSShri Abhyankar   for (i=0; i<n; i++) {
1725813a1f2bSShri Abhyankar     nzi = 0;
1726813a1f2bSShri Abhyankar     /* copy current row into linked list */
1727813a1f2bSShri Abhyankar     nnz = ai[r[i]+1] - ai[r[i]];
1728e32f2f54SBarry 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);
1729813a1f2bSShri Abhyankar     cols   = aj + ai[r[i]];
1730813a1f2bSShri Abhyankar     lnk[i] = -1; /* marker to indicate if diagonal exists */
1731813a1f2bSShri Abhyankar     ierr   = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
1732813a1f2bSShri Abhyankar     nzi   += nlnk;
1733813a1f2bSShri Abhyankar 
1734813a1f2bSShri Abhyankar     /* make sure diagonal entry is included */
1735813a1f2bSShri Abhyankar     if (diagonal_fill && lnk[i] == -1) {
1736813a1f2bSShri Abhyankar       fm = n;
1737813a1f2bSShri Abhyankar       while (lnk[fm] < i) fm = lnk[fm];
1738813a1f2bSShri Abhyankar       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
1739813a1f2bSShri Abhyankar       lnk[fm]    = i;
1740813a1f2bSShri Abhyankar       lnk_lvl[i] = 0;
1741813a1f2bSShri Abhyankar       nzi++; dcount++;
1742813a1f2bSShri Abhyankar     }
1743813a1f2bSShri Abhyankar 
1744813a1f2bSShri Abhyankar     /* add pivot rows into the active row */
1745813a1f2bSShri Abhyankar     nzbd = 0;
1746813a1f2bSShri Abhyankar     prow = lnk[n];
1747813a1f2bSShri Abhyankar     while (prow < i) {
1748813a1f2bSShri Abhyankar       nnz      = bdiag[prow];
1749813a1f2bSShri Abhyankar       cols     = bj_ptr[prow] + nnz + 1;
1750813a1f2bSShri Abhyankar       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
1751813a1f2bSShri Abhyankar       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
1752813a1f2bSShri Abhyankar       ierr     = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
1753813a1f2bSShri Abhyankar       nzi     += nlnk;
1754813a1f2bSShri Abhyankar       prow     = lnk[prow];
1755813a1f2bSShri Abhyankar       nzbd++;
1756813a1f2bSShri Abhyankar     }
1757813a1f2bSShri Abhyankar     bdiag[i] = nzbd;
1758813a1f2bSShri Abhyankar     bi[i+1]  = bi[i] + nzi;
1759813a1f2bSShri Abhyankar     /* if free space is not available, make more free space */
1760813a1f2bSShri Abhyankar     if (current_space->local_remaining<nzi) {
1761f91af8c7SBarry Smith       nnz  = PetscIntMultTruncate(2,PetscIntMultTruncate(nzi,n - i)); /* estimated and max additional space needed */
1762813a1f2bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
1763813a1f2bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
1764813a1f2bSShri Abhyankar       reallocs++;
1765813a1f2bSShri Abhyankar     }
1766813a1f2bSShri Abhyankar 
1767813a1f2bSShri Abhyankar     /* copy data into free_space and free_space_lvl, then initialize lnk */
1768813a1f2bSShri Abhyankar     ierr         = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
1769813a1f2bSShri Abhyankar     bj_ptr[i]    = current_space->array;
1770813a1f2bSShri Abhyankar     bjlvl_ptr[i] = current_space_lvl->array;
1771813a1f2bSShri Abhyankar 
1772813a1f2bSShri Abhyankar     /* make sure the active row i has diagonal entry */
177365e19b50SBarry 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);
1774813a1f2bSShri Abhyankar 
1775813a1f2bSShri Abhyankar     current_space->array               += nzi;
1776813a1f2bSShri Abhyankar     current_space->local_used          += nzi;
1777813a1f2bSShri Abhyankar     current_space->local_remaining     -= nzi;
1778813a1f2bSShri Abhyankar     current_space_lvl->array           += nzi;
1779813a1f2bSShri Abhyankar     current_space_lvl->local_used      += nzi;
1780813a1f2bSShri Abhyankar     current_space_lvl->local_remaining -= nzi;
1781813a1f2bSShri Abhyankar   }
1782813a1f2bSShri Abhyankar 
1783813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
1784813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
1785813a1f2bSShri Abhyankar   /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */
1786854ce69bSBarry Smith   ierr = PetscMalloc1(bi[n]+1,&bj);CHKERRQ(ierr);
17872ce24eb6SHong Zhang   ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
1788813a1f2bSShri Abhyankar 
1789813a1f2bSShri Abhyankar   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
1790813a1f2bSShri Abhyankar   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
17910e83c824SBarry Smith   ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr);
1792813a1f2bSShri Abhyankar 
1793813a1f2bSShri Abhyankar #if defined(PETSC_USE_INFO)
1794813a1f2bSShri Abhyankar   {
1795aef85c9fSShri Abhyankar     PetscReal af = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
179657622a8eSBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)f,(double)af);CHKERRQ(ierr);
179757622a8eSBarry Smith     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
179857622a8eSBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%g);\n",(double)af);CHKERRQ(ierr);
1799813a1f2bSShri Abhyankar     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
1800813a1f2bSShri Abhyankar     if (diagonal_fill) {
1801955c1f14SBarry Smith       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals\n",dcount);CHKERRQ(ierr);
1802813a1f2bSShri Abhyankar     }
1803813a1f2bSShri Abhyankar   }
1804813a1f2bSShri Abhyankar #endif
1805813a1f2bSShri Abhyankar   /* put together the new matrix */
18060298fd71SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
18073bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)fact,(PetscObject)isicol);CHKERRQ(ierr);
1808813a1f2bSShri Abhyankar   b    = (Mat_SeqAIJ*)(fact)->data;
18092205254eSKarl Rupp 
1810813a1f2bSShri Abhyankar   b->free_a       = PETSC_TRUE;
1811813a1f2bSShri Abhyankar   b->free_ij      = PETSC_TRUE;
1812813a1f2bSShri Abhyankar   b->singlemalloc = PETSC_FALSE;
18132205254eSKarl Rupp 
1814854ce69bSBarry Smith   ierr = PetscMalloc1(bdiag[0]+1,&b->a);CHKERRQ(ierr);
18152205254eSKarl Rupp 
1816813a1f2bSShri Abhyankar   b->j    = bj;
1817813a1f2bSShri Abhyankar   b->i    = bi;
1818813a1f2bSShri Abhyankar   b->diag = bdiag;
1819813a1f2bSShri Abhyankar   b->ilen = 0;
1820813a1f2bSShri Abhyankar   b->imax = 0;
1821813a1f2bSShri Abhyankar   b->row  = isrow;
1822813a1f2bSShri Abhyankar   b->col  = iscol;
1823813a1f2bSShri Abhyankar   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1824813a1f2bSShri Abhyankar   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
1825813a1f2bSShri Abhyankar   b->icol = isicol;
18262205254eSKarl Rupp 
1827854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&b->solve_work);CHKERRQ(ierr);
1828813a1f2bSShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.
1829813a1f2bSShri Abhyankar      Allocate bdiag, solve_work, new a, new j */
18303bb1ff40SBarry Smith   ierr     = PetscLogObjectMemory((PetscObject)fact,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
1831f268cba8SShri Abhyankar   b->maxnz = b->nz = bdiag[0]+1;
18322205254eSKarl Rupp 
1833813a1f2bSShri Abhyankar   (fact)->info.factor_mallocs    = reallocs;
1834813a1f2bSShri Abhyankar   (fact)->info.fill_ratio_given  = f;
1835f268cba8SShri Abhyankar   (fact)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
183635233d99SShri Abhyankar   (fact)->ops->lufactornumeric   = MatLUFactorNumeric_SeqAIJ;
1837f2344125SBarry Smith   if (a->inode.size) {
1838f2344125SBarry Smith     (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode;
1839f2344125SBarry Smith   }
1840a02bda8eSBarry Smith   ierr = MatSeqAIJCheckInode_FactorLU(fact);CHKERRQ(ierr);
1841813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
1842813a1f2bSShri Abhyankar }
1843813a1f2bSShri Abhyankar 
1844ad04f41aSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
18456516239fSHong Zhang {
18466516239fSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
18476516239fSHong Zhang   IS                 isicol;
18486516239fSHong Zhang   PetscErrorCode     ierr;
18496516239fSHong Zhang   const PetscInt     *r,*ic;
1850ece542f9SHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j;
18516516239fSHong Zhang   PetscInt           *bi,*cols,nnz,*cols_lvl;
18526516239fSHong Zhang   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
18536516239fSHong Zhang   PetscInt           i,levels,diagonal_fill;
1854ace3abfcSBarry Smith   PetscBool          col_identity,row_identity;
18556516239fSHong Zhang   PetscReal          f;
18560298fd71SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=NULL;
18576516239fSHong Zhang   PetscBT            lnkbt;
18586516239fSHong Zhang   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
18590298fd71SBarry Smith   PetscFreeSpaceList free_space    =NULL,current_space=NULL;
18600298fd71SBarry Smith   PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL;
1861ace3abfcSBarry Smith   PetscBool          missing;
18626516239fSHong Zhang 
18636516239fSHong Zhang   PetscFunctionBegin;
1864e32f2f54SBarry 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);
1865ece542f9SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
1866ece542f9SHong Zhang   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
1867ece542f9SHong Zhang 
18686516239fSHong Zhang   f             = info->fill;
18696516239fSHong Zhang   levels        = (PetscInt)info->levels;
18706516239fSHong Zhang   diagonal_fill = (PetscInt)info->diagonal_fill;
18712205254eSKarl Rupp 
18726516239fSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
18736516239fSHong Zhang 
18746516239fSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
18756516239fSHong Zhang   ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
18766516239fSHong Zhang   if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */
1877f77e22a1SHong Zhang     ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr);
18782205254eSKarl Rupp 
1879ad04f41aSHong Zhang     (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ_inplace;
1880f2344125SBarry Smith     if (a->inode.size) {
1881f2344125SBarry Smith       (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
1882f2344125SBarry Smith     }
1883d5f3da31SBarry Smith     fact->factortype               = MAT_FACTOR_ILU;
1884719d5645SBarry Smith     (fact)->info.factor_mallocs    = 0;
1885719d5645SBarry Smith     (fact)->info.fill_ratio_given  = info->fill;
1886719d5645SBarry Smith     (fact)->info.fill_ratio_needed = 1.0;
18872205254eSKarl Rupp 
1888719d5645SBarry Smith     b    = (Mat_SeqAIJ*)(fact)->data;
188908480c60SBarry Smith     b->row  = isrow;
189008480c60SBarry Smith     b->col  = iscol;
189182bf6240SBarry Smith     b->icol = isicol;
1892854ce69bSBarry Smith     ierr    = PetscMalloc1((fact)->rmap->n+1,&b->solve_work);CHKERRQ(ierr);
1893c38d4ed2SBarry Smith     ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1894c38d4ed2SBarry Smith     ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
18953a40ed3dSBarry Smith     PetscFunctionReturn(0);
189608480c60SBarry Smith   }
189708480c60SBarry Smith 
1898898183ecSLois Curfman McInnes   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1899898183ecSLois Curfman McInnes   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
19009e25ed09SBarry Smith 
19011bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
1902854ce69bSBarry Smith   ierr  = PetscMalloc1(n+1,&bi);CHKERRQ(ierr);
1903854ce69bSBarry Smith   ierr  = PetscMalloc1(n+1,&bdiag);CHKERRQ(ierr);
1904a6df321fSShri Abhyankar   bi[0] = bdiag[0] = 0;
19056cf997b0SBarry Smith 
1906dcca6d9dSJed Brown   ierr = PetscMalloc2(n,&bj_ptr,n,&bjlvl_ptr);CHKERRQ(ierr);
19075a8e39fbSHong Zhang 
19085a8e39fbSHong Zhang   /* create a linked list for storing column indices of the active row */
19095a8e39fbSHong Zhang   nlnk = n + 1;
19105a8e39fbSHong Zhang   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
19115a8e39fbSHong Zhang 
19125a8e39fbSHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
19132f18eb33SBarry Smith   ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space);CHKERRQ(ierr);
19145a8e39fbSHong Zhang   current_space     = free_space;
19152f18eb33SBarry Smith   ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space_lvl);CHKERRQ(ierr);
19165a8e39fbSHong Zhang   current_space_lvl = free_space_lvl;
19175a8e39fbSHong Zhang 
19185a8e39fbSHong Zhang   for (i=0; i<n; i++) {
19195a8e39fbSHong Zhang     nzi = 0;
19206cf997b0SBarry Smith     /* copy current row into linked list */
19215a8e39fbSHong Zhang     nnz = ai[r[i]+1] - ai[r[i]];
1922e32f2f54SBarry 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);
19235a8e39fbSHong Zhang     cols   = aj + ai[r[i]];
19245a8e39fbSHong Zhang     lnk[i] = -1; /* marker to indicate if diagonal exists */
19255a8e39fbSHong Zhang     ierr   = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
19265a8e39fbSHong Zhang     nzi   += nlnk;
19276cf997b0SBarry Smith 
19286cf997b0SBarry Smith     /* make sure diagonal entry is included */
19295a8e39fbSHong Zhang     if (diagonal_fill && lnk[i] == -1) {
19306cf997b0SBarry Smith       fm = n;
19315a8e39fbSHong Zhang       while (lnk[fm] < i) fm = lnk[fm];
19325a8e39fbSHong Zhang       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
19335a8e39fbSHong Zhang       lnk[fm]    = i;
19345a8e39fbSHong Zhang       lnk_lvl[i] = 0;
19355a8e39fbSHong Zhang       nzi++; dcount++;
19366cf997b0SBarry Smith     }
19376cf997b0SBarry Smith 
19385a8e39fbSHong Zhang     /* add pivot rows into the active row */
19395a8e39fbSHong Zhang     nzbd = 0;
19405a8e39fbSHong Zhang     prow = lnk[n];
19415a8e39fbSHong Zhang     while (prow < i) {
19425a8e39fbSHong Zhang       nnz      = bdiag[prow];
19435a8e39fbSHong Zhang       cols     = bj_ptr[prow] + nnz + 1;
19445a8e39fbSHong Zhang       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
19455a8e39fbSHong Zhang       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
19465a8e39fbSHong Zhang       ierr     = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
19475a8e39fbSHong Zhang       nzi     += nlnk;
19485a8e39fbSHong Zhang       prow     = lnk[prow];
19495a8e39fbSHong Zhang       nzbd++;
195056beaf04SBarry Smith     }
19515a8e39fbSHong Zhang     bdiag[i] = nzbd;
19525a8e39fbSHong Zhang     bi[i+1]  = bi[i] + nzi;
1953ecf371e4SBarry Smith 
19545a8e39fbSHong Zhang     /* if free space is not available, make more free space */
19555a8e39fbSHong Zhang     if (current_space->local_remaining<nzi) {
1956f91af8c7SBarry Smith       nnz  = PetscIntMultTruncate(nzi,n - i); /* estimated and max additional space needed */
1957a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
1958a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
19595a8e39fbSHong Zhang       reallocs++;
19605a8e39fbSHong Zhang     }
1961ecf371e4SBarry Smith 
19625a8e39fbSHong Zhang     /* copy data into free_space and free_space_lvl, then initialize lnk */
19635a8e39fbSHong Zhang     ierr         = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
19645a8e39fbSHong Zhang     bj_ptr[i]    = current_space->array;
19655a8e39fbSHong Zhang     bjlvl_ptr[i] = current_space_lvl->array;
19665a8e39fbSHong Zhang 
19675a8e39fbSHong Zhang     /* make sure the active row i has diagonal entry */
196865e19b50SBarry 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);
19695a8e39fbSHong Zhang 
19705a8e39fbSHong Zhang     current_space->array               += nzi;
19715a8e39fbSHong Zhang     current_space->local_used          += nzi;
19725a8e39fbSHong Zhang     current_space->local_remaining     -= nzi;
19735a8e39fbSHong Zhang     current_space_lvl->array           += nzi;
19745a8e39fbSHong Zhang     current_space_lvl->local_used      += nzi;
19755a8e39fbSHong Zhang     current_space_lvl->local_remaining -= nzi;
19769e25ed09SBarry Smith   }
19775a8e39fbSHong Zhang 
1978898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
1979898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
19805a8e39fbSHong Zhang 
19815a8e39fbSHong Zhang   /* destroy list of free space and other temporary arrays */
1982854ce69bSBarry Smith   ierr = PetscMalloc1(bi[n]+1,&bj);CHKERRQ(ierr);
19836516239fSHong Zhang   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */
19845a8e39fbSHong Zhang   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
1985a1a86e44SBarry Smith   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
19860e83c824SBarry Smith   ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr);
19879e25ed09SBarry Smith 
19886cf91177SBarry Smith #if defined(PETSC_USE_INFO)
1989f64065bbSBarry Smith   {
19905a8e39fbSHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
199157622a8eSBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)f,(double)af);CHKERRQ(ierr);
199257622a8eSBarry Smith     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
199357622a8eSBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%g);\n",(double)af);CHKERRQ(ierr);
1994ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
1995335d9088SBarry Smith     if (diagonal_fill) {
1996955c1f14SBarry Smith       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals\n",dcount);CHKERRQ(ierr);
1997335d9088SBarry Smith     }
1998f64065bbSBarry Smith   }
199963ba0a88SBarry Smith #endif
2000bbb6d6a8SBarry Smith 
20019e25ed09SBarry Smith   /* put together the new matrix */
20020298fd71SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
20033bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)fact,(PetscObject)isicol);CHKERRQ(ierr);
2004719d5645SBarry Smith   b    = (Mat_SeqAIJ*)(fact)->data;
20052205254eSKarl Rupp 
2006e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
2007e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
20087c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
20092205254eSKarl Rupp 
2010785e854fSJed Brown   ierr = PetscMalloc1(bi[n],&b->a);CHKERRQ(ierr);
20115a8e39fbSHong Zhang   b->j = bj;
20125a8e39fbSHong Zhang   b->i = bi;
20135a8e39fbSHong Zhang   for (i=0; i<n; i++) bdiag[i] += bi[i];
20145a8e39fbSHong Zhang   b->diag = bdiag;
2015416022c9SBarry Smith   b->ilen = 0;
2016416022c9SBarry Smith   b->imax = 0;
2017416022c9SBarry Smith   b->row  = isrow;
2018416022c9SBarry Smith   b->col  = iscol;
2019c38d4ed2SBarry Smith   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
2020c38d4ed2SBarry Smith   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
202182bf6240SBarry Smith   b->icol = isicol;
2022854ce69bSBarry Smith   ierr    = PetscMalloc1(n+1,&b->solve_work);CHKERRQ(ierr);
2023416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
20245a8e39fbSHong Zhang      Allocate bdiag, solve_work, new a, new j */
20253bb1ff40SBarry Smith   ierr     = PetscLogObjectMemory((PetscObject)fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
20265a8e39fbSHong Zhang   b->maxnz = b->nz = bi[n];
20272205254eSKarl Rupp 
2028719d5645SBarry Smith   (fact)->info.factor_mallocs    = reallocs;
2029719d5645SBarry Smith   (fact)->info.fill_ratio_given  = f;
2030719d5645SBarry Smith   (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
2031ad04f41aSHong Zhang   (fact)->ops->lufactornumeric   =  MatLUFactorNumeric_SeqAIJ_inplace;
2032f2344125SBarry Smith   if (a->inode.size) {
2033f2344125SBarry Smith     (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
2034d3ac4fa3SBarry Smith   }
20353a40ed3dSBarry Smith   PetscFunctionReturn(0);
20369e25ed09SBarry Smith }
2037d5d45c9bSBarry Smith 
203835233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
20395f44c854SHong Zhang {
20405f44c854SHong Zhang   Mat            C = B;
20415f44c854SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
20425f44c854SHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
20435f44c854SHong Zhang   IS             ip=b->row,iip = b->icol;
20445f44c854SHong Zhang   PetscErrorCode ierr;
20455f44c854SHong Zhang   const PetscInt *rip,*riip;
20465f44c854SHong Zhang   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp;
20475f44c854SHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
20485f44c854SHong Zhang   PetscInt       k,jmin,jmax,*c2r,*il,col,nexti,ili,nz;
20495f44c854SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
2050ace3abfcSBarry Smith   PetscBool      perm_identity;
2051d90ac83dSHong Zhang   FactorShiftCtx sctx;
205298a93161SHong Zhang   PetscReal      rs;
205398a93161SHong Zhang   MatScalar      d,*v;
205498a93161SHong Zhang 
20555f44c854SHong Zhang   PetscFunctionBegin;
205698a93161SHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
2057d90ac83dSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
205898a93161SHong Zhang 
2059f4db908eSBarry Smith   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
206098a93161SHong Zhang     sctx.shift_top = info->zeropivot;
206198a93161SHong Zhang     for (i=0; i<mbs; i++) {
206298a93161SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
206398a93161SHong Zhang       d  = (aa)[a->diag[i]];
206498a93161SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
206598a93161SHong Zhang       v  = aa+ai[i];
206698a93161SHong Zhang       nz = ai[i+1] - ai[i];
20672205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
206898a93161SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
206998a93161SHong Zhang     }
207098a93161SHong Zhang     sctx.shift_top *= 1.1;
207198a93161SHong Zhang     sctx.nshift_max = 5;
207298a93161SHong Zhang     sctx.shift_lo   = 0.;
207398a93161SHong Zhang     sctx.shift_hi   = 1.;
207498a93161SHong Zhang   }
20755f44c854SHong Zhang 
20765f44c854SHong Zhang   ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr);
20775f44c854SHong Zhang   ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr);
20785f44c854SHong Zhang 
20795f44c854SHong Zhang   /* allocate working arrays
20805f44c854SHong Zhang      c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col
20815f44c854SHong 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
20825f44c854SHong Zhang   */
2083dcca6d9dSJed Brown   ierr = PetscMalloc3(mbs,&rtmp,mbs,&il,mbs,&c2r);CHKERRQ(ierr);
20845f44c854SHong Zhang 
208598a93161SHong Zhang   do {
208607b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
208798a93161SHong Zhang 
2088c5546cabSHong Zhang     for (i=0; i<mbs; i++) c2r[i] = mbs;
20892e987568SSatish Balay     if (mbs) il[0] = 0;
20905f44c854SHong Zhang 
20915f44c854SHong Zhang     for (k = 0; k<mbs; k++) {
20925f44c854SHong Zhang       /* zero rtmp */
20935f44c854SHong Zhang       nz    = bi[k+1] - bi[k];
20945f44c854SHong Zhang       bjtmp = bj + bi[k];
20955f44c854SHong Zhang       for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
20965f44c854SHong Zhang 
20975f44c854SHong Zhang       /* load in initial unfactored row */
20985f44c854SHong Zhang       bval = ba + bi[k];
20995f44c854SHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
21005f44c854SHong Zhang       for (j = jmin; j < jmax; j++) {
21015f44c854SHong Zhang         col = riip[aj[j]];
21025f44c854SHong Zhang         if (col >= k) { /* only take upper triangular entry */
21035f44c854SHong Zhang           rtmp[col] = aa[j];
21045f44c854SHong Zhang           *bval++   = 0.0; /* for in-place factorization */
21055f44c854SHong Zhang         }
21065f44c854SHong Zhang       }
210798a93161SHong Zhang       /* shift the diagonal of the matrix: ZeropivotApply() */
210898a93161SHong Zhang       rtmp[k] += sctx.shift_amount;  /* shift the diagonal of the matrix */
21095f44c854SHong Zhang 
21105f44c854SHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
21115f44c854SHong Zhang       dk = rtmp[k];
21125f44c854SHong Zhang       i  = c2r[k]; /* first row to be added to k_th row  */
21135f44c854SHong Zhang 
21145f44c854SHong Zhang       while (i < k) {
21155f44c854SHong Zhang         nexti = c2r[i]; /* next row to be added to k_th row */
21165f44c854SHong Zhang 
21175f44c854SHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
21185f44c854SHong Zhang         ili     = il[i]; /* index of first nonzero element in U(i,k:bms-1) */
21195f44c854SHong Zhang         uikdi   = -ba[ili]*ba[bdiag[i]]; /* diagonal(k) */
21205f44c854SHong Zhang         dk     += uikdi*ba[ili]; /* update diag[k] */
21215f44c854SHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
21225f44c854SHong Zhang 
21235f44c854SHong Zhang         /* add multiple of row i to k-th row */
21245f44c854SHong Zhang         jmin = ili + 1; jmax = bi[i+1];
21255f44c854SHong Zhang         if (jmin < jmax) {
21265f44c854SHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
21275f44c854SHong Zhang           /* update il and c2r for row i */
21285f44c854SHong Zhang           il[i] = jmin;
21295f44c854SHong Zhang           j     = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i;
21305f44c854SHong Zhang         }
21315f44c854SHong Zhang         i = nexti;
21325f44c854SHong Zhang       }
21335f44c854SHong Zhang 
21345f44c854SHong Zhang       /* copy data into U(k,:) */
213598a93161SHong Zhang       rs   = 0.0;
21365f44c854SHong Zhang       jmin = bi[k]; jmax = bi[k+1]-1;
21375f44c854SHong Zhang       if (jmin < jmax) {
21385f44c854SHong Zhang         for (j=jmin; j<jmax; j++) {
213998a93161SHong Zhang           col = bj[j]; ba[j] = rtmp[col]; rs += PetscAbsScalar(ba[j]);
21405f44c854SHong Zhang         }
21415f44c854SHong Zhang         /* add the k-th row into il and c2r */
21425f44c854SHong Zhang         il[k] = jmin;
21435f44c854SHong Zhang         i     = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k;
21445f44c854SHong Zhang       }
214598a93161SHong Zhang 
214698a93161SHong Zhang       /* MatPivotCheck() */
214798a93161SHong Zhang       sctx.rs = rs;
214898a93161SHong Zhang       sctx.pv = dk;
21494cccfbddSHong Zhang       ierr    = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr);
215007b50cabSHong Zhang       if (sctx.newshift) break;
215198a93161SHong Zhang       dk = sctx.pv;
215298a93161SHong Zhang 
215398a93161SHong Zhang       ba[bdiag[k]] = 1.0/dk; /* U(k,k) */
215498a93161SHong Zhang     }
215507b50cabSHong Zhang   } while (sctx.newshift);
21565f44c854SHong Zhang 
21570e83c824SBarry Smith   ierr = PetscFree3(rtmp,il,c2r);CHKERRQ(ierr);
21585f44c854SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
21595f44c854SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
21605f44c854SHong Zhang 
21615f44c854SHong Zhang   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
21625f44c854SHong Zhang   if (perm_identity) {
216335233d99SShri Abhyankar     B->ops->solve          = MatSolve_SeqSBAIJ_1_NaturalOrdering;
216435233d99SShri Abhyankar     B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering;
21652169352eSHong Zhang     B->ops->forwardsolve   = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering;
21662169352eSHong Zhang     B->ops->backwardsolve  = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering;
21675f44c854SHong Zhang   } else {
216835233d99SShri Abhyankar     B->ops->solve          = MatSolve_SeqSBAIJ_1;
216935233d99SShri Abhyankar     B->ops->solvetranspose = MatSolve_SeqSBAIJ_1;
217035233d99SShri Abhyankar     B->ops->forwardsolve   = MatForwardSolve_SeqSBAIJ_1;
217135233d99SShri Abhyankar     B->ops->backwardsolve  = MatBackwardSolve_SeqSBAIJ_1;
21725f44c854SHong Zhang   }
21735f44c854SHong Zhang 
21745f44c854SHong Zhang   C->assembled    = PETSC_TRUE;
21755f44c854SHong Zhang   C->preallocated = PETSC_TRUE;
21762205254eSKarl Rupp 
21775f44c854SHong Zhang   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
217898a93161SHong Zhang 
217998a93161SHong Zhang   /* MatPivotView() */
218098a93161SHong Zhang   if (sctx.nshift) {
2181f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
218257622a8eSBarry 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);
2183f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
218457622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
2185f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS) {
218657622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %g\n",sctx.nshift,(double)info->shiftamount);CHKERRQ(ierr);
218798a93161SHong Zhang     }
218898a93161SHong Zhang   }
21895f44c854SHong Zhang   PetscFunctionReturn(0);
21905f44c854SHong Zhang }
21915f44c854SHong Zhang 
21920a3c351aSHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info)
2193a6175056SHong Zhang {
2194719d5645SBarry Smith   Mat            C = B;
21950968510aSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
21962ed133dbSHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
21979bfd6278SHong Zhang   IS             ip=b->row,iip = b->icol;
21982ed133dbSHong Zhang   PetscErrorCode ierr;
21995d0c19d7SBarry Smith   const PetscInt *rip,*riip;
2200c5546cabSHong Zhang   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol,*bjtmp;
22012ed133dbSHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
22022ed133dbSHong Zhang   PetscInt       k,jmin,jmax,*jl,*il,col,nexti,ili,nz;
2203622e2028SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
2204ace3abfcSBarry Smith   PetscBool      perm_identity;
22050e95ead3SHong Zhang   FactorShiftCtx sctx;
22060e95ead3SHong Zhang   PetscReal      rs;
22070e95ead3SHong Zhang   MatScalar      d,*v;
2208d5d45c9bSBarry Smith 
2209a6175056SHong Zhang   PetscFunctionBegin;
22100e95ead3SHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
22110e95ead3SHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
22120e95ead3SHong Zhang 
22130e95ead3SHong Zhang   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
22140e95ead3SHong Zhang     sctx.shift_top = info->zeropivot;
22150e95ead3SHong Zhang     for (i=0; i<mbs; i++) {
22160e95ead3SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
22170e95ead3SHong Zhang       d  = (aa)[a->diag[i]];
22180e95ead3SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
22190e95ead3SHong Zhang       v  = aa+ai[i];
22200e95ead3SHong Zhang       nz = ai[i+1] - ai[i];
22212205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
22220e95ead3SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
22230e95ead3SHong Zhang     }
22240e95ead3SHong Zhang     sctx.shift_top *= 1.1;
22250e95ead3SHong Zhang     sctx.nshift_max = 5;
22260e95ead3SHong Zhang     sctx.shift_lo   = 0.;
22270e95ead3SHong Zhang     sctx.shift_hi   = 1.;
22280e95ead3SHong Zhang   }
2229ee45ca4aSHong Zhang 
22302ed133dbSHong Zhang   ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr);
22319bfd6278SHong Zhang   ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr);
22322ed133dbSHong Zhang 
22332ed133dbSHong Zhang   /* initialization */
2234dcca6d9dSJed Brown   ierr = PetscMalloc3(mbs,&rtmp,mbs,&il,mbs,&jl);CHKERRQ(ierr);
22350e95ead3SHong Zhang 
22362ed133dbSHong Zhang   do {
223707b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
22380e95ead3SHong Zhang 
2239c5546cabSHong Zhang     for (i=0; i<mbs; i++) jl[i] = mbs;
2240c5546cabSHong Zhang     il[0] = 0;
22412ed133dbSHong Zhang 
22422ed133dbSHong Zhang     for (k = 0; k<mbs; k++) {
2243c5546cabSHong Zhang       /* zero rtmp */
2244c5546cabSHong Zhang       nz    = bi[k+1] - bi[k];
2245c5546cabSHong Zhang       bjtmp = bj + bi[k];
2246c5546cabSHong Zhang       for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
2247c5546cabSHong Zhang 
2248c05c3958SHong Zhang       bval = ba + bi[k];
22492ed133dbSHong Zhang       /* initialize k-th row by the perm[k]-th row of A */
22502ed133dbSHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
22512ed133dbSHong Zhang       for (j = jmin; j < jmax; j++) {
22529bfd6278SHong Zhang         col = riip[aj[j]];
22532ed133dbSHong Zhang         if (col >= k) { /* only take upper triangular entry */
22542ed133dbSHong Zhang           rtmp[col] = aa[j];
2255c05c3958SHong Zhang           *bval++   = 0.0; /* for in-place factorization */
22562ed133dbSHong Zhang         }
22572ed133dbSHong Zhang       }
225839e3d630SHong Zhang       /* shift the diagonal of the matrix */
2259540703acSHong Zhang       if (sctx.nshift) rtmp[k] += sctx.shift_amount;
22602ed133dbSHong Zhang 
22612ed133dbSHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
22622ed133dbSHong Zhang       dk = rtmp[k];
22632ed133dbSHong Zhang       i  = jl[k]; /* first row to be added to k_th row  */
22642ed133dbSHong Zhang 
22652ed133dbSHong Zhang       while (i < k) {
22662ed133dbSHong Zhang         nexti = jl[i]; /* next row to be added to k_th row */
22672ed133dbSHong Zhang 
22682ed133dbSHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
22692ed133dbSHong Zhang         ili     = il[i]; /* index of first nonzero element in U(i,k:bms-1) */
22702ed133dbSHong Zhang         uikdi   = -ba[ili]*ba[bi[i]]; /* diagonal(k) */
22712ed133dbSHong Zhang         dk     += uikdi*ba[ili];
22722ed133dbSHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
22732ed133dbSHong Zhang 
22742ed133dbSHong Zhang         /* add multiple of row i to k-th row */
22752ed133dbSHong Zhang         jmin = ili + 1; jmax = bi[i+1];
22762ed133dbSHong Zhang         if (jmin < jmax) {
22772ed133dbSHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
22782ed133dbSHong Zhang           /* update il and jl for row i */
22792ed133dbSHong Zhang           il[i] = jmin;
22802ed133dbSHong Zhang           j     = bj[jmin]; jl[i] = jl[j]; jl[j] = i;
22812ed133dbSHong Zhang         }
22822ed133dbSHong Zhang         i = nexti;
22832ed133dbSHong Zhang       }
22842ed133dbSHong Zhang 
22850697b6caSHong Zhang       /* shift the diagonals when zero pivot is detected */
22860697b6caSHong Zhang       /* compute rs=sum of abs(off-diagonal) */
22870697b6caSHong Zhang       rs   = 0.0;
22883c9e8598SHong Zhang       jmin = bi[k]+1;
22893c9e8598SHong Zhang       nz   = bi[k+1] - jmin;
22903c9e8598SHong Zhang       bcol = bj + jmin;
229104fbf559SBarry Smith       for (j=0; j<nz; j++) {
229204fbf559SBarry Smith         rs += PetscAbsScalar(rtmp[bcol[j]]);
22933c9e8598SHong Zhang       }
2294540703acSHong Zhang 
2295540703acSHong Zhang       sctx.rs = rs;
2296540703acSHong Zhang       sctx.pv = dk;
22974cccfbddSHong Zhang       ierr    = MatPivotCheck(B,A,info,&sctx,k);CHKERRQ(ierr);
229807b50cabSHong Zhang       if (sctx.newshift) break;
22990e95ead3SHong Zhang       dk = sctx.pv;
23003c9e8598SHong Zhang 
23013c9e8598SHong Zhang       /* copy data into U(k,:) */
230239e3d630SHong Zhang       ba[bi[k]] = 1.0/dk; /* U(k,k) */
230339e3d630SHong Zhang       jmin      = bi[k]+1; jmax = bi[k+1];
230439e3d630SHong Zhang       if (jmin < jmax) {
230539e3d630SHong Zhang         for (j=jmin; j<jmax; j++) {
2306c6ae9bfcSHong Zhang           col = bj[j]; ba[j] = rtmp[col];
23073c9e8598SHong Zhang         }
230839e3d630SHong Zhang         /* add the k-th row into il and jl */
23093c9e8598SHong Zhang         il[k] = jmin;
23103c9e8598SHong Zhang         i     = bj[jmin]; jl[k] = jl[i]; jl[i] = k;
23113c9e8598SHong Zhang       }
23123c9e8598SHong Zhang     }
231307b50cabSHong Zhang   } while (sctx.newshift);
23140e95ead3SHong Zhang 
23150e83c824SBarry Smith   ierr = PetscFree3(rtmp,il,jl);CHKERRQ(ierr);
231639e3d630SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
23179bfd6278SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
2318db4efbfdSBarry Smith 
2319db4efbfdSBarry Smith   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
2320db4efbfdSBarry Smith   if (perm_identity) {
23210a3c351aSHong Zhang     B->ops->solve          = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23220a3c351aSHong Zhang     B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23230a3c351aSHong Zhang     B->ops->forwardsolve   = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23240a3c351aSHong Zhang     B->ops->backwardsolve  = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
2325db4efbfdSBarry Smith   } else {
23260a3c351aSHong Zhang     B->ops->solve          = MatSolve_SeqSBAIJ_1_inplace;
23270a3c351aSHong Zhang     B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_inplace;
23280a3c351aSHong Zhang     B->ops->forwardsolve   = MatForwardSolve_SeqSBAIJ_1_inplace;
23290a3c351aSHong Zhang     B->ops->backwardsolve  = MatBackwardSolve_SeqSBAIJ_1_inplace;
2330db4efbfdSBarry Smith   }
2331db4efbfdSBarry Smith 
23323c9e8598SHong Zhang   C->assembled    = PETSC_TRUE;
23333c9e8598SHong Zhang   C->preallocated = PETSC_TRUE;
23342205254eSKarl Rupp 
2335d0f46423SBarry Smith   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
2336540703acSHong Zhang   if (sctx.nshift) {
2337f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
233857622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
2339f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
234057622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
23410697b6caSHong Zhang     }
23423c9e8598SHong Zhang   }
23433c9e8598SHong Zhang   PetscFunctionReturn(0);
23443c9e8598SHong Zhang }
2345a6175056SHong Zhang 
23468a76ed4fSHong Zhang /*
23478a76ed4fSHong Zhang    icc() under revised new data structure.
23488a76ed4fSHong Zhang    Factored arrays bj and ba are stored as
23498a76ed4fSHong Zhang      U(0,:),...,U(i,:),U(n-1,:)
23508a76ed4fSHong Zhang 
23518a76ed4fSHong Zhang    ui=fact->i is an array of size n+1, in which
23528a76ed4fSHong Zhang    ui+
23538a76ed4fSHong Zhang      ui[i]:  points to 1st entry of U(i,:),i=0,...,n-1
23548a76ed4fSHong Zhang      ui[n]:  points to U(n-1,n-1)+1
23558a76ed4fSHong Zhang 
23568a76ed4fSHong Zhang   udiag=fact->diag is an array of size n,in which
23578a76ed4fSHong Zhang      udiag[i]: points to diagonal of U(i,:), i=0,...,n-1
23588a76ed4fSHong Zhang 
23598a76ed4fSHong Zhang    U(i,:) contains udiag[i] as its last entry, i.e.,
23608a76ed4fSHong Zhang     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
23618a76ed4fSHong Zhang */
23628a76ed4fSHong Zhang 
236335233d99SShri Abhyankar PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2364a6175056SHong Zhang {
23650968510aSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
2366ed59401aSHong Zhang   Mat_SeqSBAIJ       *b;
2367dfbe8321SBarry Smith   PetscErrorCode     ierr;
2368ace3abfcSBarry Smith   PetscBool          perm_identity,missing;
23695f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
23705d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
2371ed59401aSHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
23720298fd71SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=NULL,d;
23735a8e39fbSHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
2374ed59401aSHong Zhang   PetscReal          fill          =info->fill,levels=info->levels;
23750298fd71SBarry Smith   PetscFreeSpaceList free_space    =NULL,current_space=NULL;
23760298fd71SBarry Smith   PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL;
23777a48dd6fSHong Zhang   PetscBT            lnkbt;
2378b635d36bSHong Zhang   IS                 iperm;
2379a6175056SHong Zhang 
2380a6175056SHong Zhang   PetscFunctionBegin;
2381e32f2f54SBarry 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);
238258ebbce7SBarry Smith   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
2383e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
2384653a6975SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
2385b635d36bSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
23867a48dd6fSHong Zhang 
2387854ce69bSBarry Smith   ierr  = PetscMalloc1(am+1,&ui);CHKERRQ(ierr);
2388854ce69bSBarry Smith   ierr  = PetscMalloc1(am+1,&udiag);CHKERRQ(ierr);
238939e3d630SHong Zhang   ui[0] = 0;
239039e3d630SHong Zhang 
23918a76ed4fSHong Zhang   /* ICC(0) without matrix ordering: simply rearrange column indices */
2392622e2028SHong Zhang   if (!levels && perm_identity) {
23935f44c854SHong Zhang     for (i=0; i<am; i++) {
2394c5546cabSHong Zhang       ncols    = ai[i+1] - a->diag[i];
2395c5546cabSHong Zhang       ui[i+1]  = ui[i] + ncols;
2396c5546cabSHong Zhang       udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */
2397dc1e2a3fSHong Zhang     }
2398854ce69bSBarry Smith     ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr);
2399dc1e2a3fSHong Zhang     cols = uj;
2400dc1e2a3fSHong Zhang     for (i=0; i<am; i++) {
24015f44c854SHong Zhang       aj    = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */
2402dc1e2a3fSHong Zhang       ncols = ai[i+1] - a->diag[i] -1;
24035f44c854SHong Zhang       for (j=0; j<ncols; j++) *cols++ = aj[j];
24045f44c854SHong Zhang       *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */
24055f44c854SHong Zhang     }
24065f44c854SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
24075f44c854SHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
24085f44c854SHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
24095f44c854SHong Zhang 
24105f44c854SHong Zhang     /* initialization */
2411854ce69bSBarry Smith     ierr = PetscMalloc1(am+1,&ajtmp);CHKERRQ(ierr);
24125f44c854SHong Zhang 
24135f44c854SHong Zhang     /* jl: linked list for storing indices of the pivot rows
24145f44c854SHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
2415dcca6d9dSJed Brown     ierr = PetscMalloc4(am,&uj_ptr,am,&uj_lvl_ptr,am,&jl,am,&il);CHKERRQ(ierr);
24165f44c854SHong Zhang     for (i=0; i<am; i++) {
24175f44c854SHong Zhang       jl[i] = am; il[i] = 0;
24185f44c854SHong Zhang     }
24195f44c854SHong Zhang 
24205f44c854SHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
24215f44c854SHong Zhang     nlnk = am + 1;
24225f44c854SHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
24235f44c854SHong Zhang 
242495b5ac22SHong Zhang     /* initial FreeSpace size is fill*(ai[am]+am)/2 */
24252f18eb33SBarry Smith     ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,(ai[am]+am)/2),&free_space);CHKERRQ(ierr);
24265f44c854SHong Zhang     current_space     = free_space;
24272f18eb33SBarry Smith     ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,(ai[am]+am)/2),&free_space_lvl);CHKERRQ(ierr);
24285f44c854SHong Zhang     current_space_lvl = free_space_lvl;
24295f44c854SHong Zhang 
24305f44c854SHong Zhang     for (k=0; k<am; k++) {  /* for each active row k */
24315f44c854SHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
24325f44c854SHong Zhang       nzk   = 0;
24335f44c854SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
2434e32f2f54SBarry 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);
24355f44c854SHong Zhang       ncols_upper = 0;
24365f44c854SHong Zhang       for (j=0; j<ncols; j++) {
24375f44c854SHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
24385f44c854SHong Zhang         if (riip[i] >= k) { /* only take upper triangular entry */
24395f44c854SHong Zhang           ajtmp[ncols_upper] = i;
24405f44c854SHong Zhang           ncols_upper++;
24415f44c854SHong Zhang         }
24425f44c854SHong Zhang       }
24435f44c854SHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
24445f44c854SHong Zhang       nzk += nlnk;
24455f44c854SHong Zhang 
24465f44c854SHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
24475f44c854SHong Zhang       prow = jl[k]; /* 1st pivot row */
24485f44c854SHong Zhang 
24495f44c854SHong Zhang       while (prow < k) {
24505f44c854SHong Zhang         nextprow = jl[prow];
24515f44c854SHong Zhang 
24525f44c854SHong Zhang         /* merge prow into k-th row */
24535f44c854SHong Zhang         jmin  = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */
24545f44c854SHong Zhang         jmax  = ui[prow+1];
24555f44c854SHong Zhang         ncols = jmax-jmin;
24565f44c854SHong Zhang         i     = jmin - ui[prow];
24575f44c854SHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
24585f44c854SHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
24595f44c854SHong Zhang         j     = *(uj - 1);
24605f44c854SHong Zhang         ierr  = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
24615f44c854SHong Zhang         nzk  += nlnk;
24625f44c854SHong Zhang 
24635f44c854SHong Zhang         /* update il and jl for prow */
24645f44c854SHong Zhang         if (jmin < jmax) {
24655f44c854SHong Zhang           il[prow] = jmin;
24665f44c854SHong Zhang           j        = *cols; jl[prow] = jl[j]; jl[j] = prow;
24675f44c854SHong Zhang         }
24685f44c854SHong Zhang         prow = nextprow;
24695f44c854SHong Zhang       }
24705f44c854SHong Zhang 
24715f44c854SHong Zhang       /* if free space is not available, make more free space */
24725f44c854SHong Zhang       if (current_space->local_remaining<nzk) {
24735f44c854SHong Zhang         i    = am - k + 1; /* num of unfactored rows */
2474f91af8c7SBarry Smith         i    = PetscIntMultTruncate(i,PetscMin(nzk, i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
24755f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
24765f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
24775f44c854SHong Zhang         reallocs++;
24785f44c854SHong Zhang       }
24795f44c854SHong Zhang 
24805f44c854SHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
2481e32f2f54SBarry Smith       if (nzk == 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
24825f44c854SHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
24835f44c854SHong Zhang 
24845f44c854SHong Zhang       /* add the k-th row into il and jl */
24855f44c854SHong Zhang       if (nzk > 1) {
24865f44c854SHong Zhang         i     = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
24875f44c854SHong Zhang         jl[k] = jl[i]; jl[i] = k;
24885f44c854SHong Zhang         il[k] = ui[k] + 1;
24895f44c854SHong Zhang       }
24905f44c854SHong Zhang       uj_ptr[k]     = current_space->array;
24915f44c854SHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
24925f44c854SHong Zhang 
24935f44c854SHong Zhang       current_space->array           += nzk;
24945f44c854SHong Zhang       current_space->local_used      += nzk;
24955f44c854SHong Zhang       current_space->local_remaining -= nzk;
24965f44c854SHong Zhang 
24975f44c854SHong Zhang       current_space_lvl->array           += nzk;
24985f44c854SHong Zhang       current_space_lvl->local_used      += nzk;
24995f44c854SHong Zhang       current_space_lvl->local_remaining -= nzk;
25005f44c854SHong Zhang 
25015f44c854SHong Zhang       ui[k+1] = ui[k] + nzk;
25025f44c854SHong Zhang     }
25035f44c854SHong Zhang 
25045f44c854SHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
25055f44c854SHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
25060e83c824SBarry Smith     ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr);
25075f44c854SHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
25085f44c854SHong Zhang 
25099263d837SHong Zhang     /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */
2510854ce69bSBarry Smith     ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr);
251135233d99SShri Abhyankar     ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor  */
25125f44c854SHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
25135f44c854SHong Zhang     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
25145f44c854SHong Zhang 
25155f44c854SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
25165f44c854SHong Zhang 
25175f44c854SHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
25185f44c854SHong Zhang   b               = (Mat_SeqSBAIJ*)(fact)->data;
25195f44c854SHong Zhang   b->singlemalloc = PETSC_FALSE;
25202205254eSKarl Rupp 
2521854ce69bSBarry Smith   ierr = PetscMalloc1(ui[am]+1,&b->a);CHKERRQ(ierr);
25222205254eSKarl Rupp 
25235f44c854SHong Zhang   b->j             = uj;
25245f44c854SHong Zhang   b->i             = ui;
25255f44c854SHong Zhang   b->diag          = udiag;
25267f53bb6cSHong Zhang   b->free_diag     = PETSC_TRUE;
25275f44c854SHong Zhang   b->ilen          = 0;
25285f44c854SHong Zhang   b->imax          = 0;
25295f44c854SHong Zhang   b->row           = perm;
25305f44c854SHong Zhang   b->col           = perm;
25315f44c854SHong Zhang   ierr             = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
25325f44c854SHong Zhang   ierr             = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
25335f44c854SHong Zhang   b->icol          = iperm;
25345f44c854SHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
25352205254eSKarl Rupp 
2536854ce69bSBarry Smith   ierr = PetscMalloc1(am+1,&b->solve_work);CHKERRQ(ierr);
25373bb1ff40SBarry Smith   ierr = PetscLogObjectMemory((PetscObject)fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
25382205254eSKarl Rupp 
25395f44c854SHong Zhang   b->maxnz   = b->nz = ui[am];
25405f44c854SHong Zhang   b->free_a  = PETSC_TRUE;
25415f44c854SHong Zhang   b->free_ij = PETSC_TRUE;
25425f44c854SHong Zhang 
2543f284f12aSHong Zhang   fact->info.factor_mallocs   = reallocs;
2544f284f12aSHong Zhang   fact->info.fill_ratio_given = fill;
25455f44c854SHong Zhang   if (ai[am] != 0) {
25466a69fef8SHong Zhang     /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */
254795b5ac22SHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am);
25485f44c854SHong Zhang   } else {
2549f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
25505f44c854SHong Zhang   }
25519263d837SHong Zhang #if defined(PETSC_USE_INFO)
25529263d837SHong Zhang   if (ai[am] != 0) {
25539263d837SHong Zhang     PetscReal af = fact->info.fill_ratio_needed;
255457622a8eSBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)fill,(double)af);CHKERRQ(ierr);
255557622a8eSBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
255657622a8eSBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g) for best performance.\n",(double)af);CHKERRQ(ierr);
25579263d837SHong Zhang   } else {
25589263d837SHong Zhang     ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
25599263d837SHong Zhang   }
25609263d837SHong Zhang #endif
256135233d99SShri Abhyankar   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
25625f44c854SHong Zhang   PetscFunctionReturn(0);
25635f44c854SHong Zhang }
25645f44c854SHong Zhang 
2565ad04f41aSHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
25665f44c854SHong Zhang {
25675f44c854SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
25685f44c854SHong Zhang   Mat_SeqSBAIJ       *b;
25695f44c854SHong Zhang   PetscErrorCode     ierr;
2570ace3abfcSBarry Smith   PetscBool          perm_identity,missing;
25715f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
25725f44c854SHong Zhang   const PetscInt     *rip,*riip;
25735f44c854SHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
25740298fd71SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=NULL,d;
25755f44c854SHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
25765f44c854SHong Zhang   PetscReal          fill          =info->fill,levels=info->levels;
25770298fd71SBarry Smith   PetscFreeSpaceList free_space    =NULL,current_space=NULL;
25780298fd71SBarry Smith   PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL;
25795f44c854SHong Zhang   PetscBT            lnkbt;
25805f44c854SHong Zhang   IS                 iperm;
25815f44c854SHong Zhang 
25825f44c854SHong Zhang   PetscFunctionBegin;
2583e32f2f54SBarry 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);
25845f44c854SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
2585e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
25865f44c854SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
25875f44c854SHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
25885f44c854SHong Zhang 
2589854ce69bSBarry Smith   ierr  = PetscMalloc1(am+1,&ui);CHKERRQ(ierr);
2590854ce69bSBarry Smith   ierr  = PetscMalloc1(am+1,&udiag);CHKERRQ(ierr);
25915f44c854SHong Zhang   ui[0] = 0;
25925f44c854SHong Zhang 
25935f44c854SHong Zhang   /* ICC(0) without matrix ordering: simply copies fill pattern */
25945f44c854SHong Zhang   if (!levels && perm_identity) {
25955f44c854SHong Zhang 
25965f44c854SHong Zhang     for (i=0; i<am; i++) {
25975f44c854SHong Zhang       ui[i+1]  = ui[i] + ai[i+1] - a->diag[i];
25985f44c854SHong Zhang       udiag[i] = ui[i];
2599ed59401aSHong Zhang     }
2600854ce69bSBarry Smith     ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr);
260139e3d630SHong Zhang     cols = uj;
2602ed59401aSHong Zhang     for (i=0; i<am; i++) {
2603ed59401aSHong Zhang       aj    = a->j + a->diag[i];
260439e3d630SHong Zhang       ncols = ui[i+1] - ui[i];
260539e3d630SHong Zhang       for (j=0; j<ncols; j++) *cols++ = *aj++;
2606ed59401aSHong Zhang     }
260739e3d630SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
2608b635d36bSHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
2609b635d36bSHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
2610b635d36bSHong Zhang 
26117a48dd6fSHong Zhang     /* initialization */
2612854ce69bSBarry Smith     ierr = PetscMalloc1(am+1,&ajtmp);CHKERRQ(ierr);
26137a48dd6fSHong Zhang 
26147a48dd6fSHong Zhang     /* jl: linked list for storing indices of the pivot rows
26157a48dd6fSHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
2616dcca6d9dSJed Brown     ierr = PetscMalloc4(am,&uj_ptr,am,&uj_lvl_ptr,am,&jl,am,&il);CHKERRQ(ierr);
26177a48dd6fSHong Zhang     for (i=0; i<am; i++) {
26187a48dd6fSHong Zhang       jl[i] = am; il[i] = 0;
26197a48dd6fSHong Zhang     }
26207a48dd6fSHong Zhang 
26217a48dd6fSHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
26227a48dd6fSHong Zhang     nlnk = am + 1;
26232ed133dbSHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
26247a48dd6fSHong Zhang 
26257a48dd6fSHong Zhang     /* initial FreeSpace size is fill*(ai[am]+1) */
26262f18eb33SBarry Smith     ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,ai[am]+1),&free_space);CHKERRQ(ierr);
26277a48dd6fSHong Zhang     current_space     = free_space;
26282f18eb33SBarry Smith     ierr              = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,ai[am]+1),&free_space_lvl);CHKERRQ(ierr);
26297a48dd6fSHong Zhang     current_space_lvl = free_space_lvl;
26307a48dd6fSHong Zhang 
26317a48dd6fSHong Zhang     for (k=0; k<am; k++) {  /* for each active row k */
26327a48dd6fSHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
26337a48dd6fSHong Zhang       nzk   = 0;
2634622e2028SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
2635e32f2f54SBarry 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);
2636622e2028SHong Zhang       ncols_upper = 0;
2637622e2028SHong Zhang       for (j=0; j<ncols; j++) {
2638b635d36bSHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
2639b635d36bSHong Zhang         if (riip[i] >= k) { /* only take upper triangular entry */
26405a8e39fbSHong Zhang           ajtmp[ncols_upper] = i;
2641622e2028SHong Zhang           ncols_upper++;
2642622e2028SHong Zhang         }
2643622e2028SHong Zhang       }
2644b635d36bSHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
26457a48dd6fSHong Zhang       nzk += nlnk;
26467a48dd6fSHong Zhang 
26477a48dd6fSHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
26487a48dd6fSHong Zhang       prow = jl[k]; /* 1st pivot row */
26497a48dd6fSHong Zhang 
26507a48dd6fSHong Zhang       while (prow < k) {
26517a48dd6fSHong Zhang         nextprow = jl[prow];
26527a48dd6fSHong Zhang 
26537a48dd6fSHong Zhang         /* merge prow into k-th row */
26547a48dd6fSHong Zhang         jmin  = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */
26557a48dd6fSHong Zhang         jmax  = ui[prow+1];
26567a48dd6fSHong Zhang         ncols = jmax-jmin;
2657ed59401aSHong Zhang         i     = jmin - ui[prow];
2658ed59401aSHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
26595a8e39fbSHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
26605a8e39fbSHong Zhang         j     = *(uj - 1);
26615a8e39fbSHong Zhang         ierr  = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
26627a48dd6fSHong Zhang         nzk  += nlnk;
26637a48dd6fSHong Zhang 
26647a48dd6fSHong Zhang         /* update il and jl for prow */
26657a48dd6fSHong Zhang         if (jmin < jmax) {
26667a48dd6fSHong Zhang           il[prow] = jmin;
26677a48dd6fSHong Zhang           j        = *cols; jl[prow] = jl[j]; jl[j] = prow;
26687a48dd6fSHong Zhang         }
26697a48dd6fSHong Zhang         prow = nextprow;
26707a48dd6fSHong Zhang       }
26717a48dd6fSHong Zhang 
26727a48dd6fSHong Zhang       /* if free space is not available, make more free space */
26737a48dd6fSHong Zhang       if (current_space->local_remaining<nzk) {
26747a48dd6fSHong Zhang         i    = am - k + 1; /* num of unfactored rows */
2675f91af8c7SBarry Smith         i    = PetscIntMultTruncate(i,PetscMin(nzk, i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
2676a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
2677a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
26787a48dd6fSHong Zhang         reallocs++;
26797a48dd6fSHong Zhang       }
26807a48dd6fSHong Zhang 
26817a48dd6fSHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
268265e19b50SBarry Smith       if (!nzk) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
26832ed133dbSHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
26847a48dd6fSHong Zhang 
26857a48dd6fSHong Zhang       /* add the k-th row into il and jl */
268639e3d630SHong Zhang       if (nzk > 1) {
26877a48dd6fSHong Zhang         i     = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
26887a48dd6fSHong Zhang         jl[k] = jl[i]; jl[i] = k;
26897a48dd6fSHong Zhang         il[k] = ui[k] + 1;
26907a48dd6fSHong Zhang       }
26917a48dd6fSHong Zhang       uj_ptr[k]     = current_space->array;
26927a48dd6fSHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
26937a48dd6fSHong Zhang 
26947a48dd6fSHong Zhang       current_space->array           += nzk;
26957a48dd6fSHong Zhang       current_space->local_used      += nzk;
26967a48dd6fSHong Zhang       current_space->local_remaining -= nzk;
26977a48dd6fSHong Zhang 
26987a48dd6fSHong Zhang       current_space_lvl->array           += nzk;
26997a48dd6fSHong Zhang       current_space_lvl->local_used      += nzk;
27007a48dd6fSHong Zhang       current_space_lvl->local_remaining -= nzk;
27017a48dd6fSHong Zhang 
27027a48dd6fSHong Zhang       ui[k+1] = ui[k] + nzk;
27037a48dd6fSHong Zhang     }
27047a48dd6fSHong Zhang 
27056cf91177SBarry Smith #if defined(PETSC_USE_INFO)
27067a48dd6fSHong Zhang     if (ai[am] != 0) {
270739e3d630SHong Zhang       PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]);
270857622a8eSBarry Smith       ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)fill,(double)af);CHKERRQ(ierr);
270957622a8eSBarry Smith       ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
271057622a8eSBarry Smith       ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g) for best performance.\n",(double)af);CHKERRQ(ierr);
27117a48dd6fSHong Zhang     } else {
2712ae15b995SBarry Smith       ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
27137a48dd6fSHong Zhang     }
271463ba0a88SBarry Smith #endif
27157a48dd6fSHong Zhang 
27167a48dd6fSHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
2717b635d36bSHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
27180e83c824SBarry Smith     ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr);
27195a8e39fbSHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
27207a48dd6fSHong Zhang 
27217a48dd6fSHong Zhang     /* destroy list of free space and other temporary array(s) */
2722854ce69bSBarry Smith     ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr);
2723a1a86e44SBarry Smith     ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
27242ed133dbSHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
2725a1a86e44SBarry Smith     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
27267a48dd6fSHong Zhang 
272739e3d630SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
272839e3d630SHong Zhang 
27297a48dd6fSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
27307a48dd6fSHong Zhang 
2731f284f12aSHong Zhang   b               = (Mat_SeqSBAIJ*)fact->data;
27327a48dd6fSHong Zhang   b->singlemalloc = PETSC_FALSE;
27332205254eSKarl Rupp 
2734854ce69bSBarry Smith   ierr = PetscMalloc1(ui[am]+1,&b->a);CHKERRQ(ierr);
27352205254eSKarl Rupp 
27367a48dd6fSHong Zhang   b->j         = uj;
27377a48dd6fSHong Zhang   b->i         = ui;
27385f44c854SHong Zhang   b->diag      = udiag;
27397f53bb6cSHong Zhang   b->free_diag = PETSC_TRUE;
27407a48dd6fSHong Zhang   b->ilen      = 0;
27417a48dd6fSHong Zhang   b->imax      = 0;
27427a48dd6fSHong Zhang   b->row       = perm;
2743b635d36bSHong Zhang   b->col       = perm;
27442205254eSKarl Rupp 
2745b635d36bSHong Zhang   ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
2746b635d36bSHong Zhang   ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
27472205254eSKarl Rupp 
2748b635d36bSHong Zhang   b->icol          = iperm;
27497a48dd6fSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
2750854ce69bSBarry Smith   ierr             = PetscMalloc1(am+1,&b->solve_work);CHKERRQ(ierr);
27513bb1ff40SBarry Smith   ierr             = PetscLogObjectMemory((PetscObject)fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
27527a48dd6fSHong Zhang   b->maxnz         = b->nz = ui[am];
2753e6b907acSBarry Smith   b->free_a        = PETSC_TRUE;
2754e6b907acSBarry Smith   b->free_ij       = PETSC_TRUE;
27557a48dd6fSHong Zhang 
2756f284f12aSHong Zhang   fact->info.factor_mallocs   = reallocs;
2757f284f12aSHong Zhang   fact->info.fill_ratio_given = fill;
27587a48dd6fSHong Zhang   if (ai[am] != 0) {
2759f284f12aSHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
27607a48dd6fSHong Zhang   } else {
2761f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
27627a48dd6fSHong Zhang   }
27630a3c351aSHong Zhang   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace;
2764a6175056SHong Zhang   PetscFunctionReturn(0);
2765a6175056SHong Zhang }
2766d5d45c9bSBarry Smith 
276735233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
27680be760fbSHong Zhang {
27690be760fbSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
27700be760fbSHong Zhang   Mat_SeqSBAIJ       *b;
27710be760fbSHong Zhang   PetscErrorCode     ierr;
27729186b105SHong Zhang   PetscBool          perm_identity,missing;
27730be760fbSHong Zhang   PetscReal          fill = info->fill;
27740be760fbSHong Zhang   const PetscInt     *rip,*riip;
27750be760fbSHong Zhang   PetscInt           i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
27760be760fbSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
27770be760fbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr,*udiag;
27780298fd71SBarry Smith   PetscFreeSpaceList free_space=NULL,current_space=NULL;
27790be760fbSHong Zhang   PetscBT            lnkbt;
27800be760fbSHong Zhang   IS                 iperm;
27810be760fbSHong Zhang 
27820be760fbSHong Zhang   PetscFunctionBegin;
2783e32f2f54SBarry 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);
27849186b105SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
27859186b105SHong Zhang   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
27869186b105SHong Zhang 
27870be760fbSHong Zhang   /* check whether perm is the identity mapping */
27880be760fbSHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
27890be760fbSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
27900be760fbSHong Zhang   ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
27910be760fbSHong Zhang   ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
27920be760fbSHong Zhang 
27930be760fbSHong Zhang   /* initialization */
2794854ce69bSBarry Smith   ierr  = PetscMalloc1(am+1,&ui);CHKERRQ(ierr);
2795854ce69bSBarry Smith   ierr  = PetscMalloc1(am+1,&udiag);CHKERRQ(ierr);
27960be760fbSHong Zhang   ui[0] = 0;
27970be760fbSHong Zhang 
27980be760fbSHong Zhang   /* jl: linked list for storing indices of the pivot rows
27990be760fbSHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
2800dcca6d9dSJed Brown   ierr = PetscMalloc4(am,&ui_ptr,am,&jl,am,&il,am,&cols);CHKERRQ(ierr);
28010be760fbSHong Zhang   for (i=0; i<am; i++) {
28020be760fbSHong Zhang     jl[i] = am; il[i] = 0;
28030be760fbSHong Zhang   }
28040be760fbSHong Zhang 
28050be760fbSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
28060be760fbSHong Zhang   nlnk = am + 1;
28070be760fbSHong Zhang   ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28080be760fbSHong Zhang 
280995b5ac22SHong Zhang   /* initial FreeSpace size is fill*(ai[am]+am)/2 */
28102f18eb33SBarry Smith   ierr          = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,(ai[am]+am)/2),&free_space);CHKERRQ(ierr);
28110be760fbSHong Zhang   current_space = free_space;
28120be760fbSHong Zhang 
28130be760fbSHong Zhang   for (k=0; k<am; k++) {  /* for each active row k */
28140be760fbSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
28150be760fbSHong Zhang     nzk   = 0;
28160be760fbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
2817e32f2f54SBarry 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);
28180be760fbSHong Zhang     ncols_upper = 0;
28190be760fbSHong Zhang     for (j=0; j<ncols; j++) {
28200be760fbSHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
28210be760fbSHong Zhang       if (i >= k) { /* only take upper triangular entry */
28220be760fbSHong Zhang         cols[ncols_upper] = i;
28230be760fbSHong Zhang         ncols_upper++;
28240be760fbSHong Zhang       }
28250be760fbSHong Zhang     }
28260be760fbSHong Zhang     ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28270be760fbSHong Zhang     nzk += nlnk;
28280be760fbSHong Zhang 
28290be760fbSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
28300be760fbSHong Zhang     prow = jl[k]; /* 1st pivot row */
28310be760fbSHong Zhang 
28320be760fbSHong Zhang     while (prow < k) {
28330be760fbSHong Zhang       nextprow = jl[prow];
28340be760fbSHong Zhang       /* merge prow into k-th row */
28350be760fbSHong Zhang       jmin   = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */
28360be760fbSHong Zhang       jmax   = ui[prow+1];
28370be760fbSHong Zhang       ncols  = jmax-jmin;
28380be760fbSHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
28390be760fbSHong Zhang       ierr   = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28400be760fbSHong Zhang       nzk   += nlnk;
28410be760fbSHong Zhang 
28420be760fbSHong Zhang       /* update il and jl for prow */
28430be760fbSHong Zhang       if (jmin < jmax) {
28440be760fbSHong Zhang         il[prow] = jmin;
28452205254eSKarl Rupp         j        = *uj_ptr;
28462205254eSKarl Rupp         jl[prow] = jl[j];
28472205254eSKarl Rupp         jl[j]    = prow;
28480be760fbSHong Zhang       }
28490be760fbSHong Zhang       prow = nextprow;
28500be760fbSHong Zhang     }
28510be760fbSHong Zhang 
28520be760fbSHong Zhang     /* if free space is not available, make more free space */
28530be760fbSHong Zhang     if (current_space->local_remaining<nzk) {
28540be760fbSHong Zhang       i    = am - k + 1; /* num of unfactored rows */
2855f91af8c7SBarry Smith       i    = PetscIntMultTruncate(i,PetscMin(nzk,i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
28560be760fbSHong Zhang       ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
28570be760fbSHong Zhang       reallocs++;
28580be760fbSHong Zhang     }
28590be760fbSHong Zhang 
28600be760fbSHong Zhang     /* copy data into free space, then initialize lnk */
28610be760fbSHong Zhang     ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
28620be760fbSHong Zhang 
28630be760fbSHong Zhang     /* add the k-th row into il and jl */
28647b056e98SHong Zhang     if (nzk > 1) {
28650be760fbSHong Zhang       i     = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
28660be760fbSHong Zhang       jl[k] = jl[i]; jl[i] = k;
28670be760fbSHong Zhang       il[k] = ui[k] + 1;
28680be760fbSHong Zhang     }
28690be760fbSHong Zhang     ui_ptr[k] = current_space->array;
28702205254eSKarl Rupp 
28710be760fbSHong Zhang     current_space->array           += nzk;
28720be760fbSHong Zhang     current_space->local_used      += nzk;
28730be760fbSHong Zhang     current_space->local_remaining -= nzk;
28740be760fbSHong Zhang 
28750be760fbSHong Zhang     ui[k+1] = ui[k] + nzk;
28760be760fbSHong Zhang   }
28770be760fbSHong Zhang 
28780be760fbSHong Zhang   ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
28790be760fbSHong Zhang   ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
28800be760fbSHong Zhang   ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr);
28810be760fbSHong Zhang 
28829263d837SHong Zhang   /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */
2883854ce69bSBarry Smith   ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr);
288435233d99SShri Abhyankar   ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor */
28850be760fbSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
28860be760fbSHong Zhang 
28870be760fbSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
28880be760fbSHong Zhang 
2889f284f12aSHong Zhang   b               = (Mat_SeqSBAIJ*)fact->data;
28900be760fbSHong Zhang   b->singlemalloc = PETSC_FALSE;
28910be760fbSHong Zhang   b->free_a       = PETSC_TRUE;
28920be760fbSHong Zhang   b->free_ij      = PETSC_TRUE;
28932205254eSKarl Rupp 
2894854ce69bSBarry Smith   ierr = PetscMalloc1(ui[am]+1,&b->a);CHKERRQ(ierr);
28952205254eSKarl Rupp 
28960be760fbSHong Zhang   b->j         = uj;
28970be760fbSHong Zhang   b->i         = ui;
28980be760fbSHong Zhang   b->diag      = udiag;
28990be760fbSHong Zhang   b->free_diag = PETSC_TRUE;
29000be760fbSHong Zhang   b->ilen      = 0;
29010be760fbSHong Zhang   b->imax      = 0;
29020be760fbSHong Zhang   b->row       = perm;
29030be760fbSHong Zhang   b->col       = perm;
290426fbe8dcSKarl Rupp 
29050be760fbSHong Zhang   ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
29060be760fbSHong Zhang   ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
29072205254eSKarl Rupp 
29080be760fbSHong Zhang   b->icol          = iperm;
29090be760fbSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
291026fbe8dcSKarl Rupp 
2911854ce69bSBarry Smith   ierr = PetscMalloc1(am+1,&b->solve_work);CHKERRQ(ierr);
29123bb1ff40SBarry Smith   ierr = PetscLogObjectMemory((PetscObject)fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
29132205254eSKarl Rupp 
29140be760fbSHong Zhang   b->maxnz = b->nz = ui[am];
29150be760fbSHong Zhang 
2916f284f12aSHong Zhang   fact->info.factor_mallocs   = reallocs;
2917f284f12aSHong Zhang   fact->info.fill_ratio_given = fill;
29180be760fbSHong Zhang   if (ai[am] != 0) {
29196a69fef8SHong Zhang     /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */
292095b5ac22SHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am);
29210be760fbSHong Zhang   } else {
2922f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
29230be760fbSHong Zhang   }
29249263d837SHong Zhang #if defined(PETSC_USE_INFO)
29259263d837SHong Zhang   if (ai[am] != 0) {
29269263d837SHong Zhang     PetscReal af = fact->info.fill_ratio_needed;
292757622a8eSBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)fill,(double)af);CHKERRQ(ierr);
292857622a8eSBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
292957622a8eSBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g) for best performance.\n",(double)af);CHKERRQ(ierr);
29309263d837SHong Zhang   } else {
29319263d837SHong Zhang     ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
29329263d837SHong Zhang   }
29339263d837SHong Zhang #endif
293435233d99SShri Abhyankar   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
29350be760fbSHong Zhang   PetscFunctionReturn(0);
29360be760fbSHong Zhang }
29370be760fbSHong Zhang 
2938ad04f41aSHong Zhang PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2939f76d2b81SHong Zhang {
2940f76d2b81SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
294110c27e3dSHong Zhang   Mat_SeqSBAIJ       *b;
2942dfbe8321SBarry Smith   PetscErrorCode     ierr;
29439186b105SHong Zhang   PetscBool          perm_identity,missing;
294410c27e3dSHong Zhang   PetscReal          fill = info->fill;
29455d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
29465d0c19d7SBarry Smith   PetscInt           i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
294710c27e3dSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
29482ed133dbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr;
29490298fd71SBarry Smith   PetscFreeSpaceList free_space=NULL,current_space=NULL;
295010c27e3dSHong Zhang   PetscBT            lnkbt;
29512ed133dbSHong Zhang   IS                 iperm;
2952f76d2b81SHong Zhang 
2953f76d2b81SHong Zhang   PetscFunctionBegin;
2954e32f2f54SBarry 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);
29559186b105SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
29569186b105SHong Zhang   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
29579186b105SHong Zhang 
29582ed133dbSHong Zhang   /* check whether perm is the identity mapping */
2959f76d2b81SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
29602ed133dbSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
29612ed133dbSHong Zhang   ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
29629bfd6278SHong Zhang   ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
296310c27e3dSHong Zhang 
296410c27e3dSHong Zhang   /* initialization */
2965854ce69bSBarry Smith   ierr  = PetscMalloc1(am+1,&ui);CHKERRQ(ierr);
296610c27e3dSHong Zhang   ui[0] = 0;
296710c27e3dSHong Zhang 
296810c27e3dSHong Zhang   /* jl: linked list for storing indices of the pivot rows
29691393bc97SHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
2970dcca6d9dSJed Brown   ierr = PetscMalloc4(am,&ui_ptr,am,&jl,am,&il,am,&cols);CHKERRQ(ierr);
29711393bc97SHong Zhang   for (i=0; i<am; i++) {
29721393bc97SHong Zhang     jl[i] = am; il[i] = 0;
2973f76d2b81SHong Zhang   }
297410c27e3dSHong Zhang 
297510c27e3dSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
29761393bc97SHong Zhang   nlnk = am + 1;
29771393bc97SHong Zhang   ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
297810c27e3dSHong Zhang 
29791393bc97SHong Zhang   /* initial FreeSpace size is fill*(ai[am]+1) */
29802f18eb33SBarry Smith   ierr          = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,ai[am]+1),&free_space);CHKERRQ(ierr);
298110c27e3dSHong Zhang   current_space = free_space;
298210c27e3dSHong Zhang 
29831393bc97SHong Zhang   for (k=0; k<am; k++) {  /* for each active row k */
298410c27e3dSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
298510c27e3dSHong Zhang     nzk   = 0;
29862ed133dbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
2987e32f2f54SBarry 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);
29882ed133dbSHong Zhang     ncols_upper = 0;
29892ed133dbSHong Zhang     for (j=0; j<ncols; j++) {
29909bfd6278SHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
29912ed133dbSHong Zhang       if (i >= k) { /* only take upper triangular entry */
29922ed133dbSHong Zhang         cols[ncols_upper] = i;
29932ed133dbSHong Zhang         ncols_upper++;
29942ed133dbSHong Zhang       }
29952ed133dbSHong Zhang     }
29961393bc97SHong Zhang     ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
299710c27e3dSHong Zhang     nzk += nlnk;
299810c27e3dSHong Zhang 
299910c27e3dSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
300010c27e3dSHong Zhang     prow = jl[k]; /* 1st pivot row */
300110c27e3dSHong Zhang 
300210c27e3dSHong Zhang     while (prow < k) {
300310c27e3dSHong Zhang       nextprow = jl[prow];
300410c27e3dSHong Zhang       /* merge prow into k-th row */
30051393bc97SHong Zhang       jmin   = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */
300610c27e3dSHong Zhang       jmax   = ui[prow+1];
300710c27e3dSHong Zhang       ncols  = jmax-jmin;
30081393bc97SHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
30095a8e39fbSHong Zhang       ierr   = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
301010c27e3dSHong Zhang       nzk   += nlnk;
301110c27e3dSHong Zhang 
301210c27e3dSHong Zhang       /* update il and jl for prow */
301310c27e3dSHong Zhang       if (jmin < jmax) {
301410c27e3dSHong Zhang         il[prow] = jmin;
30152ed133dbSHong Zhang         j        = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow;
301610c27e3dSHong Zhang       }
301710c27e3dSHong Zhang       prow = nextprow;
301810c27e3dSHong Zhang     }
301910c27e3dSHong Zhang 
302010c27e3dSHong Zhang     /* if free space is not available, make more free space */
302110c27e3dSHong Zhang     if (current_space->local_remaining<nzk) {
30221393bc97SHong Zhang       i    = am - k + 1; /* num of unfactored rows */
302310c27e3dSHong Zhang       i    = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
3024a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
302510c27e3dSHong Zhang       reallocs++;
302610c27e3dSHong Zhang     }
302710c27e3dSHong Zhang 
302810c27e3dSHong Zhang     /* copy data into free space, then initialize lnk */
30291393bc97SHong Zhang     ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
303010c27e3dSHong Zhang 
303110c27e3dSHong Zhang     /* add the k-th row into il and jl */
303210c27e3dSHong Zhang     if (nzk-1 > 0) {
30331393bc97SHong Zhang       i     = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
303410c27e3dSHong Zhang       jl[k] = jl[i]; jl[i] = k;
303510c27e3dSHong Zhang       il[k] = ui[k] + 1;
303610c27e3dSHong Zhang     }
30372ed133dbSHong Zhang     ui_ptr[k] = current_space->array;
30382205254eSKarl Rupp 
303910c27e3dSHong Zhang     current_space->array           += nzk;
304010c27e3dSHong Zhang     current_space->local_used      += nzk;
304110c27e3dSHong Zhang     current_space->local_remaining -= nzk;
304210c27e3dSHong Zhang 
304310c27e3dSHong Zhang     ui[k+1] = ui[k] + nzk;
304410c27e3dSHong Zhang   }
304510c27e3dSHong Zhang 
30466cf91177SBarry Smith #if defined(PETSC_USE_INFO)
30471393bc97SHong Zhang   if (ai[am] != 0) {
30481393bc97SHong Zhang     PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]);
304957622a8eSBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)fill,(double)af);CHKERRQ(ierr);
305057622a8eSBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr);
305157622a8eSBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g) for best performance.\n",(double)af);CHKERRQ(ierr);
305210c27e3dSHong Zhang   } else {
3053ae15b995SBarry Smith     ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
305410c27e3dSHong Zhang   }
305563ba0a88SBarry Smith #endif
305610c27e3dSHong Zhang 
305710c27e3dSHong Zhang   ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
30589bfd6278SHong Zhang   ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
30590e83c824SBarry Smith   ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr);
306010c27e3dSHong Zhang 
306110c27e3dSHong Zhang   /* destroy list of free space and other temporary array(s) */
3062854ce69bSBarry Smith   ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr);
3063a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
306410c27e3dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
306510c27e3dSHong Zhang 
306610c27e3dSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
306710c27e3dSHong Zhang 
3068f284f12aSHong Zhang   b               = (Mat_SeqSBAIJ*)fact->data;
306910c27e3dSHong Zhang   b->singlemalloc = PETSC_FALSE;
3070e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
3071e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
30722205254eSKarl Rupp 
3073854ce69bSBarry Smith   ierr = PetscMalloc1(ui[am]+1,&b->a);CHKERRQ(ierr);
30742205254eSKarl Rupp 
307510c27e3dSHong Zhang   b->j    = uj;
307610c27e3dSHong Zhang   b->i    = ui;
307710c27e3dSHong Zhang   b->diag = 0;
307810c27e3dSHong Zhang   b->ilen = 0;
307910c27e3dSHong Zhang   b->imax = 0;
308010c27e3dSHong Zhang   b->row  = perm;
30819bfd6278SHong Zhang   b->col  = perm;
30822205254eSKarl Rupp 
30839bfd6278SHong Zhang   ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
30849bfd6278SHong Zhang   ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
30852205254eSKarl Rupp 
30869bfd6278SHong Zhang   b->icol          = iperm;
308710c27e3dSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
30882205254eSKarl Rupp 
3089854ce69bSBarry Smith   ierr     = PetscMalloc1(am+1,&b->solve_work);CHKERRQ(ierr);
30903bb1ff40SBarry Smith   ierr     = PetscLogObjectMemory((PetscObject)fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
30911393bc97SHong Zhang   b->maxnz = b->nz = ui[am];
309210c27e3dSHong Zhang 
3093f284f12aSHong Zhang   fact->info.factor_mallocs   = reallocs;
3094f284f12aSHong Zhang   fact->info.fill_ratio_given = fill;
30951393bc97SHong Zhang   if (ai[am] != 0) {
3096f284f12aSHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
309710c27e3dSHong Zhang   } else {
3098f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
309910c27e3dSHong Zhang   }
31000a3c351aSHong Zhang   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace;
3101f76d2b81SHong Zhang   PetscFunctionReturn(0);
3102f76d2b81SHong Zhang }
3103599ef60dSHong Zhang 
310435233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx)
3105813a1f2bSShri Abhyankar {
3106813a1f2bSShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
3107813a1f2bSShri Abhyankar   PetscErrorCode    ierr;
3108813a1f2bSShri Abhyankar   PetscInt          n   = A->rmap->n;
3109813a1f2bSShri Abhyankar   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag,*vi;
3110813a1f2bSShri Abhyankar   PetscScalar       *x,sum;
3111813a1f2bSShri Abhyankar   const PetscScalar *b;
3112813a1f2bSShri Abhyankar   const MatScalar   *aa = a->a,*v;
3113813a1f2bSShri Abhyankar   PetscInt          i,nz;
3114813a1f2bSShri Abhyankar 
3115813a1f2bSShri Abhyankar   PetscFunctionBegin;
3116813a1f2bSShri Abhyankar   if (!n) PetscFunctionReturn(0);
3117813a1f2bSShri Abhyankar 
31183649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
3119813a1f2bSShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
3120813a1f2bSShri Abhyankar 
3121813a1f2bSShri Abhyankar   /* forward solve the lower triangular */
3122813a1f2bSShri Abhyankar   x[0] = b[0];
3123813a1f2bSShri Abhyankar   v    = aa;
3124813a1f2bSShri Abhyankar   vi   = aj;
3125813a1f2bSShri Abhyankar   for (i=1; i<n; i++) {
3126813a1f2bSShri Abhyankar     nz  = ai[i+1] - ai[i];
3127813a1f2bSShri Abhyankar     sum = b[i];
3128813a1f2bSShri Abhyankar     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
3129813a1f2bSShri Abhyankar     v   += nz;
3130813a1f2bSShri Abhyankar     vi  += nz;
3131813a1f2bSShri Abhyankar     x[i] = sum;
3132813a1f2bSShri Abhyankar   }
3133813a1f2bSShri Abhyankar 
3134813a1f2bSShri Abhyankar   /* backward solve the upper triangular */
313562fbd6afSShri Abhyankar   for (i=n-1; i>=0; i--) {
31363c0119dfSShri Abhyankar     v   = aa + adiag[i+1] + 1;
31373c0119dfSShri Abhyankar     vi  = aj + adiag[i+1] + 1;
3138813a1f2bSShri Abhyankar     nz  = adiag[i] - adiag[i+1]-1;
3139813a1f2bSShri Abhyankar     sum = x[i];
3140813a1f2bSShri Abhyankar     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
31413c0119dfSShri Abhyankar     x[i] = sum*v[nz]; /* x[i]=aa[adiag[i]]*sum; v++; */
3142813a1f2bSShri Abhyankar   }
3143813a1f2bSShri Abhyankar 
3144813a1f2bSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
31453649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
3146813a1f2bSShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3147813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
3148813a1f2bSShri Abhyankar }
3149813a1f2bSShri Abhyankar 
315035233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx)
3151f268cba8SShri Abhyankar {
3152f268cba8SShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
3153f268cba8SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
3154f268cba8SShri Abhyankar   PetscErrorCode    ierr;
3155f268cba8SShri Abhyankar   PetscInt          i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,*adiag = a->diag,nz;
3156f268cba8SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c;
3157f268cba8SShri Abhyankar   PetscScalar       *x,*tmp,sum;
3158f268cba8SShri Abhyankar   const PetscScalar *b;
3159f268cba8SShri Abhyankar   const MatScalar   *aa = a->a,*v;
3160f268cba8SShri Abhyankar 
3161f268cba8SShri Abhyankar   PetscFunctionBegin;
3162f268cba8SShri Abhyankar   if (!n) PetscFunctionReturn(0);
3163f268cba8SShri Abhyankar 
31643649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
3165f268cba8SShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
3166f268cba8SShri Abhyankar   tmp  = a->solve_work;
3167f268cba8SShri Abhyankar 
3168f268cba8SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
3169f268cba8SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
3170f268cba8SShri Abhyankar 
3171f268cba8SShri Abhyankar   /* forward solve the lower triangular */
3172f268cba8SShri Abhyankar   tmp[0] = b[r[0]];
3173f268cba8SShri Abhyankar   v      = aa;
3174f268cba8SShri Abhyankar   vi     = aj;
3175f268cba8SShri Abhyankar   for (i=1; i<n; i++) {
3176f268cba8SShri Abhyankar     nz  = ai[i+1] - ai[i];
3177f268cba8SShri Abhyankar     sum = b[r[i]];
3178f268cba8SShri Abhyankar     PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
3179f268cba8SShri Abhyankar     tmp[i] = sum;
3180f268cba8SShri Abhyankar     v     += nz; vi += nz;
3181f268cba8SShri Abhyankar   }
3182f268cba8SShri Abhyankar 
3183f268cba8SShri Abhyankar   /* backward solve the upper triangular */
3184f268cba8SShri Abhyankar   for (i=n-1; i>=0; i--) {
31853c0119dfSShri Abhyankar     v   = aa + adiag[i+1]+1;
31863c0119dfSShri Abhyankar     vi  = aj + adiag[i+1]+1;
3187f268cba8SShri Abhyankar     nz  = adiag[i]-adiag[i+1]-1;
3188f268cba8SShri Abhyankar     sum = tmp[i];
3189f268cba8SShri Abhyankar     PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
3190f268cba8SShri Abhyankar     x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */
3191f268cba8SShri Abhyankar   }
3192f268cba8SShri Abhyankar 
3193f268cba8SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
3194f268cba8SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
31953649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
3196f268cba8SShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3197f268cba8SShri Abhyankar   ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr);
3198f268cba8SShri Abhyankar   PetscFunctionReturn(0);
3199f268cba8SShri Abhyankar }
3200f268cba8SShri Abhyankar 
3201fe97e370SBarry Smith /*
3202bebe2cf6SSatish 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
3203fe97e370SBarry Smith */
32041d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact)
32051d098868SHong Zhang {
32061d098868SHong Zhang   Mat            B = *fact;
3207599ef60dSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*b;
3208599ef60dSHong Zhang   IS             isicol;
3209599ef60dSHong Zhang   PetscErrorCode ierr;
32101d098868SHong Zhang   const PetscInt *r,*ic;
32111d098868SHong Zhang   PetscInt       i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag;
3212dc3a2fd3SHong Zhang   PetscInt       *bi,*bj,*bdiag,*bdiag_rev;
3213f61a948aSLisandro Dalcin   PetscInt       row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au;
32141d098868SHong Zhang   PetscInt       nlnk,*lnk;
32151d098868SHong Zhang   PetscBT        lnkbt;
3216ace3abfcSBarry Smith   PetscBool      row_identity,icol_identity;
32178aee2decSHong Zhang   MatScalar      *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp;
32181d098868SHong Zhang   const PetscInt *ics;
32191d098868SHong Zhang   PetscInt       j,nz,*pj,*bjtmp,k,ncut,*jtmp;
3220a2ea699eSBarry Smith   PetscReal      dt     =info->dt,shift=info->shiftamount;
3221f61a948aSLisandro Dalcin   PetscInt       dtcount=(PetscInt)info->dtcount,nnz_max;
3222ace3abfcSBarry Smith   PetscBool      missing;
3223599ef60dSHong Zhang 
3224599ef60dSHong Zhang   PetscFunctionBegin;
3225f61a948aSLisandro Dalcin   if (dt      == PETSC_DEFAULT) dt = 0.005;
3226f61a948aSLisandro Dalcin   if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax);
3227f61a948aSLisandro Dalcin 
32281d098868SHong Zhang   /* ------- symbolic factorization, can be reused ---------*/
32291d098868SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
3230e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
32311d098868SHong Zhang   adiag=a->diag;
3232599ef60dSHong Zhang 
3233599ef60dSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
3234599ef60dSHong Zhang 
3235599ef60dSHong Zhang   /* bdiag is location of diagonal in factor */
3236854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&bdiag);CHKERRQ(ierr);     /* becomes b->diag */
3237854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&bdiag_rev);CHKERRQ(ierr); /* temporary */
3238599ef60dSHong Zhang 
32391d098868SHong Zhang   /* allocate row pointers bi */
3240854ce69bSBarry Smith   ierr = PetscMalloc1(2*n+2,&bi);CHKERRQ(ierr);
32411d098868SHong Zhang 
32421d098868SHong Zhang   /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */
3243393d3a68SHong Zhang   if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */
32441d098868SHong Zhang   nnz_max = ai[n]+2*n*dtcount+2;
32458fc3a347SHong Zhang 
3246854ce69bSBarry Smith   ierr = PetscMalloc1(nnz_max+1,&bj);CHKERRQ(ierr);
3247854ce69bSBarry Smith   ierr = PetscMalloc1(nnz_max+1,&ba);CHKERRQ(ierr);
3248599ef60dSHong Zhang 
3249599ef60dSHong Zhang   /* put together the new matrix */
32500298fd71SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
32513bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)B,(PetscObject)isicol);CHKERRQ(ierr);
3252f284f12aSHong Zhang   b    = (Mat_SeqAIJ*)B->data;
32532205254eSKarl Rupp 
3254599ef60dSHong Zhang   b->free_a       = PETSC_TRUE;
3255599ef60dSHong Zhang   b->free_ij      = PETSC_TRUE;
3256599ef60dSHong Zhang   b->singlemalloc = PETSC_FALSE;
32572205254eSKarl Rupp 
3258599ef60dSHong Zhang   b->a    = ba;
3259599ef60dSHong Zhang   b->j    = bj;
3260599ef60dSHong Zhang   b->i    = bi;
3261599ef60dSHong Zhang   b->diag = bdiag;
3262599ef60dSHong Zhang   b->ilen = 0;
3263599ef60dSHong Zhang   b->imax = 0;
3264599ef60dSHong Zhang   b->row  = isrow;
3265599ef60dSHong Zhang   b->col  = iscol;
3266599ef60dSHong Zhang   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
3267599ef60dSHong Zhang   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
3268599ef60dSHong Zhang   b->icol = isicol;
3269599ef60dSHong Zhang 
3270854ce69bSBarry Smith   ierr     = PetscMalloc1(n+1,&b->solve_work);CHKERRQ(ierr);
32713bb1ff40SBarry Smith   ierr     = PetscLogObjectMemory((PetscObject)B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
32721d098868SHong Zhang   b->maxnz = nnz_max;
3273599ef60dSHong Zhang 
3274d5f3da31SBarry Smith   B->factortype            = MAT_FACTOR_ILUDT;
3275f284f12aSHong Zhang   B->info.factor_mallocs   = 0;
3276f284f12aSHong Zhang   B->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]);
32771d098868SHong Zhang   /* ------- end of symbolic factorization ---------*/
3278599ef60dSHong Zhang 
3279599ef60dSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
3280599ef60dSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
3281599ef60dSHong Zhang   ics  = ic;
3282599ef60dSHong Zhang 
3283599ef60dSHong Zhang   /* linked list for storing column indices of the active row */
3284599ef60dSHong Zhang   nlnk = n + 1;
3285599ef60dSHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
32861d098868SHong Zhang 
32871d098868SHong Zhang   /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */
3288dcca6d9dSJed Brown   ierr = PetscMalloc2(n,&im,n,&jtmp);CHKERRQ(ierr);
32891d098868SHong Zhang   /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */
3290dcca6d9dSJed Brown   ierr = PetscMalloc2(n,&rtmp,n,&vtmp);CHKERRQ(ierr);
32910e83c824SBarry Smith   ierr = PetscMemzero(rtmp,n*sizeof(MatScalar));CHKERRQ(ierr);
3292599ef60dSHong Zhang 
3293599ef60dSHong Zhang   bi[0]        = 0;
32948fc3a347SHong Zhang   bdiag[0]     = nnz_max-1; /* location of diag[0] in factor B */
3295dc3a2fd3SHong Zhang   bdiag_rev[n] = bdiag[0];
32968fc3a347SHong Zhang   bi[2*n+1]    = bdiag[0]+1; /* endof bj and ba array */
3297599ef60dSHong Zhang   for (i=0; i<n; i++) {
3298599ef60dSHong Zhang     /* copy initial fill into linked list */
32998369b28dSHong Zhang     nzi = ai[r[i]+1] - ai[r[i]];
3300e32f2f54SBarry 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);
33018369b28dSHong Zhang     nzi_al = adiag[r[i]] - ai[r[i]];
33028369b28dSHong Zhang     nzi_au = ai[r[i]+1] - adiag[r[i]] -1;
3303599ef60dSHong Zhang     ajtmp  = aj + ai[r[i]];
33048369b28dSHong Zhang     ierr   = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3305599ef60dSHong Zhang 
3306599ef60dSHong Zhang     /* load in initial (unfactored row) */
33071d098868SHong Zhang     aatmp = a->a + ai[r[i]];
33088369b28dSHong Zhang     for (j=0; j<nzi; j++) {
33091d098868SHong Zhang       rtmp[ics[*ajtmp++]] = *aatmp++;
3310599ef60dSHong Zhang     }
3311599ef60dSHong Zhang 
3312599ef60dSHong Zhang     /* add pivot rows into linked list */
3313599ef60dSHong Zhang     row = lnk[n];
3314599ef60dSHong Zhang     while (row < i) {
33151d098868SHong Zhang       nzi_bl = bi[row+1] - bi[row] + 1;
33161d098868SHong Zhang       bjtmp  = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */
33171d098868SHong Zhang       ierr   = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr);
3318599ef60dSHong Zhang       nzi   += nlnk;
3319599ef60dSHong Zhang       row    = lnk[row];
3320599ef60dSHong Zhang     }
3321599ef60dSHong Zhang 
33228369b28dSHong Zhang     /* copy data from lnk into jtmp, then initialize lnk */
33238369b28dSHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr);
3324599ef60dSHong Zhang 
3325599ef60dSHong Zhang     /* numerical factorization */
33268369b28dSHong Zhang     bjtmp = jtmp;
3327599ef60dSHong Zhang     row   = *bjtmp++; /* 1st pivot row */
3328599ef60dSHong Zhang     while (row < i) {
3329599ef60dSHong Zhang       pc         = rtmp + row;
33303c2a7987SHong Zhang       pv         = ba + bdiag[row]; /* 1./(diag of the pivot row) */
33313c2a7987SHong Zhang       multiplier = (*pc) * (*pv);
3332599ef60dSHong Zhang       *pc        = multiplier;
33333c2a7987SHong Zhang       if (PetscAbsScalar(*pc) > dt) { /* apply tolerance dropping rule */
33341d098868SHong Zhang         pj = bj + bdiag[row+1] + 1;         /* point to 1st entry of U(row,:) */
33351d098868SHong Zhang         pv = ba + bdiag[row+1] + 1;
33361d098868SHong Zhang         /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */
33371d098868SHong Zhang         nz = bdiag[row] - bdiag[row+1] - 1;         /* num of entries in U(row,:), excluding diagonal */
33381d098868SHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
333928b1a77fSLisandro Dalcin         ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
3340599ef60dSHong Zhang       }
3341599ef60dSHong Zhang       row = *bjtmp++;
3342599ef60dSHong Zhang     }
3343599ef60dSHong Zhang 
33448369b28dSHong Zhang     /* copy sparse rtmp into contiguous vtmp; separate L and U part */
33458aee2decSHong Zhang     diag_tmp = rtmp[i];  /* save diagonal value - may not needed?? */
33468369b28dSHong Zhang     nzi_bl   = 0; j = 0;
33478369b28dSHong Zhang     while (jtmp[j] < i) { /* Note: jtmp is sorted */
33488aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
33498369b28dSHong Zhang       nzi_bl++; j++;
33508369b28dSHong Zhang     }
33518369b28dSHong Zhang     nzi_bu = nzi - nzi_bl -1;
33528369b28dSHong Zhang     while (j < nzi) {
33538aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
33548369b28dSHong Zhang       j++;
33558369b28dSHong Zhang     }
33568369b28dSHong Zhang 
3357599ef60dSHong Zhang     bjtmp = bj + bi[i];
3358599ef60dSHong Zhang     batmp = ba + bi[i];
33598369b28dSHong Zhang     /* apply level dropping rule to L part */
33608369b28dSHong Zhang     ncut = nzi_al + dtcount;
33618369b28dSHong Zhang     if (ncut < nzi_bl) {
33628369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr);
33638369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr);
3364599ef60dSHong Zhang     } else {
33658369b28dSHong Zhang       ncut = nzi_bl;
33668369b28dSHong Zhang     }
33678369b28dSHong Zhang     for (j=0; j<ncut; j++) {
33688369b28dSHong Zhang       bjtmp[j] = jtmp[j];
33698369b28dSHong Zhang       batmp[j] = vtmp[j];
33706da515a1SHong Zhang       /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */
33718369b28dSHong Zhang     }
33721d098868SHong Zhang     bi[i+1] = bi[i] + ncut;
33738369b28dSHong Zhang     nzi     = ncut + 1;
33748369b28dSHong Zhang 
33758369b28dSHong Zhang     /* apply level dropping rule to U part */
33768369b28dSHong Zhang     ncut = nzi_au + dtcount;
33778369b28dSHong Zhang     if (ncut < nzi_bu) {
33788369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr);
33798369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr);
33808369b28dSHong Zhang     } else {
33818369b28dSHong Zhang       ncut = nzi_bu;
33828369b28dSHong Zhang     }
33838369b28dSHong Zhang     nzi += ncut;
33841d098868SHong Zhang 
33851d098868SHong Zhang     /* mark bdiagonal */
33861d098868SHong Zhang     bdiag[i+1]       = bdiag[i] - (ncut + 1);
3387dc3a2fd3SHong Zhang     bdiag_rev[n-i-1] = bdiag[i+1];
33888fc3a347SHong Zhang     bi[2*n - i]      = bi[2*n - i +1] - (ncut + 1);
33891d098868SHong Zhang     bjtmp            = bj + bdiag[i];
33901d098868SHong Zhang     batmp            = ba + bdiag[i];
33911d098868SHong Zhang     *bjtmp           = i;
33928aee2decSHong Zhang     *batmp           = diag_tmp; /* rtmp[i]; */
3393c9c72f8fSHong Zhang     if (*batmp == 0.0) {
3394c9c72f8fSHong Zhang       *batmp = dt+shift;
3395bd1bc851SBarry Smith       /* printf(" row %d add shift %g\n",i,shift); */
3396c9c72f8fSHong Zhang     }
3397b78a477dSHong Zhang     *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */
33986da515a1SHong Zhang     /* printf(" (%d,%g),",*bjtmp,*batmp); */
33991d098868SHong Zhang 
34001d098868SHong Zhang     bjtmp = bj + bdiag[i+1]+1;
34011d098868SHong Zhang     batmp = ba + bdiag[i+1]+1;
34028369b28dSHong Zhang     for (k=0; k<ncut; k++) {
34031d098868SHong Zhang       bjtmp[k] = jtmp[nzi_bl+1+k];
34041d098868SHong Zhang       batmp[k] = vtmp[nzi_bl+1+k];
34056da515a1SHong Zhang       /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */
3406599ef60dSHong Zhang     }
34076da515a1SHong Zhang     /* printf("\n"); */
3408599ef60dSHong Zhang 
3409599ef60dSHong Zhang     im[i] = nzi;   /* used by PetscLLAddSortedLU() */
3410599ef60dSHong Zhang     /*
34111d098868SHong Zhang     printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]);
34128369b28dSHong Zhang     printf(" ----------------------------\n");
3413599ef60dSHong Zhang     */
34148369b28dSHong Zhang   } /* for (i=0; i<n; i++) */
34151d098868SHong Zhang     /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */
3416e32f2f54SBarry 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]);
3417599ef60dSHong Zhang 
3418599ef60dSHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3419599ef60dSHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3420599ef60dSHong Zhang 
3421599ef60dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
34220e83c824SBarry Smith   ierr = PetscFree2(im,jtmp);CHKERRQ(ierr);
34230e83c824SBarry Smith   ierr = PetscFree2(rtmp,vtmp);CHKERRQ(ierr);
34241bfa06eaSJed Brown   ierr = PetscFree(bdiag_rev);CHKERRQ(ierr);
3425599ef60dSHong Zhang 
3426599ef60dSHong Zhang   ierr     = PetscLogFlops(B->cmap->n);CHKERRQ(ierr);
34271d098868SHong Zhang   b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n];
3428599ef60dSHong Zhang 
3429599ef60dSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
3430599ef60dSHong Zhang   ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr);
3431599ef60dSHong Zhang   if (row_identity && icol_identity) {
343235233d99SShri Abhyankar     B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
3433599ef60dSHong Zhang   } else {
343435233d99SShri Abhyankar     B->ops->solve = MatSolve_SeqAIJ;
3435599ef60dSHong Zhang   }
3436599ef60dSHong Zhang 
34371d098868SHong Zhang   B->ops->solveadd          = 0;
34381d098868SHong Zhang   B->ops->solvetranspose    = 0;
34391d098868SHong Zhang   B->ops->solvetransposeadd = 0;
34401d098868SHong Zhang   B->ops->matsolve          = 0;
3441599ef60dSHong Zhang   B->assembled              = PETSC_TRUE;
3442599ef60dSHong Zhang   B->preallocated           = PETSC_TRUE;
3443599ef60dSHong Zhang   PetscFunctionReturn(0);
3444599ef60dSHong Zhang }
3445599ef60dSHong Zhang 
34463c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */
3447fe97e370SBarry Smith /*
3448bebe2cf6SSatish 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
3449fe97e370SBarry Smith */
3450fe97e370SBarry Smith 
34517087cfbeSBarry Smith PetscErrorCode  MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info)
34523c2a7987SHong Zhang {
34533c2a7987SHong Zhang   PetscErrorCode ierr;
34543c2a7987SHong Zhang 
34553c2a7987SHong Zhang   PetscFunctionBegin;
34563c2a7987SHong Zhang   ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr);
34573c2a7987SHong Zhang   PetscFunctionReturn(0);
34583c2a7987SHong Zhang }
34593c2a7987SHong Zhang 
34603c2a7987SHong Zhang /*
34613c2a7987SHong Zhang    same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors
34623c2a7987SHong Zhang    - intend to replace existing MatLUFactorNumeric_SeqAIJ()
34633c2a7987SHong Zhang */
3464fe97e370SBarry Smith /*
3465bebe2cf6SSatish 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
3466fe97e370SBarry Smith */
3467fe97e370SBarry Smith 
34687087cfbeSBarry Smith PetscErrorCode  MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info)
34693c2a7987SHong Zhang {
34703c2a7987SHong Zhang   Mat            C     =fact;
34713c2a7987SHong Zhang   Mat_SeqAIJ     *a    =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data;
34723c2a7987SHong Zhang   IS             isrow = b->row,isicol = b->icol;
34733c2a7987SHong Zhang   PetscErrorCode ierr;
34743c2a7987SHong Zhang   const PetscInt *r,*ic,*ics;
3475b78a477dSHong Zhang   PetscInt       i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
3476b78a477dSHong Zhang   PetscInt       *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj;
34773c2a7987SHong Zhang   MatScalar      *rtmp,*pc,multiplier,*v,*pv,*aa=a->a;
3478182b8fbaSHong Zhang   PetscReal      dt=info->dt,shift=info->shiftamount;
3479ace3abfcSBarry Smith   PetscBool      row_identity, col_identity;
34803c2a7987SHong Zhang 
34813c2a7987SHong Zhang   PetscFunctionBegin;
34823c2a7987SHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
34833c2a7987SHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
3484854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&rtmp);CHKERRQ(ierr);
34853c2a7987SHong Zhang   ics  = ic;
34863c2a7987SHong Zhang 
34873c2a7987SHong Zhang   for (i=0; i<n; i++) {
3488b78a477dSHong Zhang     /* initialize rtmp array */
3489b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i];       /* num of nozeros in L(i,:) */
34903c2a7987SHong Zhang     bjtmp = bj + bi[i];
3491b78a477dSHong Zhang     for  (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0;
3492b78a477dSHong Zhang     rtmp[i] = 0.0;
3493b78a477dSHong Zhang     nzu     = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */
3494b78a477dSHong Zhang     bjtmp   = bj + bdiag[i+1] + 1;
3495b78a477dSHong Zhang     for  (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0;
34963c2a7987SHong Zhang 
3497b78a477dSHong Zhang     /* load in initial unfactored row of A */
34986da515a1SHong Zhang     /* printf("row %d\n",i); */
34993c2a7987SHong Zhang     nz    = ai[r[i]+1] - ai[r[i]];
35003c2a7987SHong Zhang     ajtmp = aj + ai[r[i]];
35013c2a7987SHong Zhang     v     = aa + ai[r[i]];
35023c2a7987SHong Zhang     for (j=0; j<nz; j++) {
3503b78a477dSHong Zhang       rtmp[ics[*ajtmp++]] = v[j];
35046da515a1SHong Zhang       /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */
35053c2a7987SHong Zhang     }
35066da515a1SHong Zhang     /* printf("\n"); */
35073c2a7987SHong Zhang 
3508b78a477dSHong Zhang     /* numerical factorization */
3509b78a477dSHong Zhang     bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */
3510b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i]; /* num of entries in L(i,:) */
3511b78a477dSHong Zhang     k     = 0;
3512b78a477dSHong Zhang     while (k < nzl) {
35133c2a7987SHong Zhang       row = *bjtmp++;
35146da515a1SHong Zhang       /* printf("  prow %d\n",row); */
35153c2a7987SHong Zhang       pc         = rtmp + row;
3516b78a477dSHong Zhang       pv         = b->a + bdiag[row]; /* 1./(diag of the pivot row) */
3517b78a477dSHong Zhang       multiplier = (*pc) * (*pv);
35183c2a7987SHong Zhang       *pc        = multiplier;
3519b78a477dSHong Zhang       if (PetscAbsScalar(multiplier) > dt) {
3520b78a477dSHong Zhang         pj = bj + bdiag[row+1] + 1;         /* point to 1st entry of U(row,:) */
3521b78a477dSHong Zhang         pv = b->a + bdiag[row+1] + 1;
3522b78a477dSHong Zhang         nz = bdiag[row] - bdiag[row+1] - 1;         /* num of entries in U(row,:), excluding diagonal */
3523b78a477dSHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
352428b1a77fSLisandro Dalcin         ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
35253c2a7987SHong Zhang       }
3526b78a477dSHong Zhang       k++;
35273c2a7987SHong Zhang     }
35283c2a7987SHong Zhang 
3529b78a477dSHong Zhang     /* finished row so stick it into b->a */
3530b78a477dSHong Zhang     /* L-part */
3531b78a477dSHong Zhang     pv  = b->a + bi[i];
3532b78a477dSHong Zhang     pj  = bj + bi[i];
3533b78a477dSHong Zhang     nzl = bi[i+1] - bi[i];
3534b78a477dSHong Zhang     for (j=0; j<nzl; j++) {
3535b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
35366da515a1SHong Zhang       /* printf(" (%d,%g),",pj[j],pv[j]); */
35373c2a7987SHong Zhang     }
3538b78a477dSHong Zhang 
3539b78a477dSHong Zhang     /* diagonal: invert diagonal entries for simplier triangular solves */
3540b78a477dSHong Zhang     if (rtmp[i] == 0.0) rtmp[i] = dt+shift;
3541b78a477dSHong Zhang     b->a[bdiag[i]] = 1.0/rtmp[i];
35426da515a1SHong Zhang     /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */
3543b78a477dSHong Zhang 
3544b78a477dSHong Zhang     /* U-part */
3545b78a477dSHong Zhang     pv  = b->a + bdiag[i+1] + 1;
3546b78a477dSHong Zhang     pj  = bj + bdiag[i+1] + 1;
3547b78a477dSHong Zhang     nzu = bdiag[i] - bdiag[i+1] - 1;
3548b78a477dSHong Zhang     for (j=0; j<nzu; j++) {
3549b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
35506da515a1SHong Zhang       /* printf(" (%d,%g),",pj[j],pv[j]); */
3551b78a477dSHong Zhang     }
35526da515a1SHong Zhang     /* printf("\n"); */
3553b78a477dSHong Zhang   }
3554b78a477dSHong Zhang 
35553c2a7987SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
35563c2a7987SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
35573c2a7987SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3558b78a477dSHong Zhang 
35593c2a7987SHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
35603c2a7987SHong Zhang   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
35613c2a7987SHong Zhang   if (row_identity && col_identity) {
356235233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
35633c2a7987SHong Zhang   } else {
356435233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ;
35653c2a7987SHong Zhang   }
3566b78a477dSHong Zhang   C->ops->solveadd          = 0;
3567b78a477dSHong Zhang   C->ops->solvetranspose    = 0;
3568b78a477dSHong Zhang   C->ops->solvetransposeadd = 0;
3569b78a477dSHong Zhang   C->ops->matsolve          = 0;
35703c2a7987SHong Zhang   C->assembled              = PETSC_TRUE;
35713c2a7987SHong Zhang   C->preallocated           = PETSC_TRUE;
35722205254eSKarl Rupp 
35733c2a7987SHong Zhang   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
35743c2a7987SHong Zhang   PetscFunctionReturn(0);
35753c2a7987SHong Zhang }
3576