xref: /petsc/src/mat/impls/aij/seq/aijfact.c (revision 28b400f66ebc7ae0049166a2294dfcd3df27e64b)
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;
15889a8dedSBarry Smith   PetscInt          i,j,jj,k, kk,n = mat->rmap->n, current = 0, newcurrent = 0,*order;
168fa24674SBarry Smith   const PetscInt    *ai = a->i, *aj = a->j;
178fa24674SBarry Smith   const PetscScalar *aa = a->a;
18ace3abfcSBarry Smith   PetscBool         *done;
19889a8dedSBarry Smith   PetscReal         best,past = 0,future;
203a40ed3dSBarry Smith 
218fa24674SBarry Smith   PetscFunctionBegin;
228fa24674SBarry Smith   /* pick initial row */
238fa24674SBarry Smith   best = -1;
248fa24674SBarry Smith   for (i=0; i<n; i++) {
2575567043SBarry Smith     future = 0.0;
268fa24674SBarry Smith     for (j=ai[i]; j<ai[i+1]; j++) {
27db4deed7SKarl Rupp       if (aj[j] != i) future += PetscAbsScalar(aa[j]);
28db4deed7SKarl Rupp       else              past  = PetscAbsScalar(aa[j]);
298fa24674SBarry Smith     }
308fa24674SBarry Smith     if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */
318fa24674SBarry Smith     if (past/future > best) {
328fa24674SBarry Smith       best    = past/future;
338fa24674SBarry Smith       current = i;
348fa24674SBarry Smith     }
358fa24674SBarry Smith   }
368fa24674SBarry Smith 
375f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(n,&done));
385f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscArrayzero(done,n));
395f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(n,&order));
408fa24674SBarry Smith   order[0] = current;
418fa24674SBarry Smith   for (i=0; i<n-1; i++) {
428fa24674SBarry Smith     done[current] = PETSC_TRUE;
438fa24674SBarry Smith     best          = -1;
448fa24674SBarry Smith     /* loop over all neighbors of current pivot */
458fa24674SBarry Smith     for (j=ai[current]; j<ai[current+1]; j++) {
468fa24674SBarry Smith       jj = aj[j];
478fa24674SBarry Smith       if (done[jj]) continue;
488fa24674SBarry Smith       /* loop over columns of potential next row computing weights for below and above diagonal */
498fa24674SBarry Smith       past = future = 0.0;
508fa24674SBarry Smith       for (k=ai[jj]; k<ai[jj+1]; k++) {
518fa24674SBarry Smith         kk = aj[k];
528fa24674SBarry Smith         if (done[kk]) past += PetscAbsScalar(aa[k]);
538fa24674SBarry Smith         else if (kk != jj) future += PetscAbsScalar(aa[k]);
548fa24674SBarry Smith       }
558fa24674SBarry Smith       if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */
568fa24674SBarry Smith       if (past/future > best) {
578fa24674SBarry Smith         best       = past/future;
588fa24674SBarry Smith         newcurrent = jj;
598fa24674SBarry Smith       }
608fa24674SBarry Smith     }
618fa24674SBarry Smith     if (best == -1) { /* no neighbors to select from so select best of all that remain */
628fa24674SBarry Smith       best = -1;
638fa24674SBarry Smith       for (k=0; k<n; k++) {
648fa24674SBarry Smith         if (done[k]) continue;
6575567043SBarry Smith         future = 0.0;
6675567043SBarry Smith         past   = 0.0;
678fa24674SBarry Smith         for (j=ai[k]; j<ai[k+1]; j++) {
688fa24674SBarry Smith           kk = aj[j];
698fa24674SBarry Smith           if (done[kk])       past += PetscAbsScalar(aa[j]);
708fa24674SBarry Smith           else if (kk != k) future += PetscAbsScalar(aa[j]);
718fa24674SBarry Smith         }
728fa24674SBarry Smith         if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */
738fa24674SBarry Smith         if (past/future > best) {
748fa24674SBarry Smith           best       = past/future;
758fa24674SBarry Smith           newcurrent = k;
768fa24674SBarry Smith         }
778fa24674SBarry Smith       }
788fa24674SBarry Smith     }
792c71b3e2SJacob Faibussowitsch     PetscCheckFalse(current == newcurrent,PETSC_COMM_SELF,PETSC_ERR_PLIB,"newcurrent cannot be current");
808fa24674SBarry Smith     current    = newcurrent;
818fa24674SBarry Smith     order[i+1] = current;
828fa24674SBarry Smith   }
835f80ce2aSJacob Faibussowitsch   CHKERRQ(ISCreateGeneral(PETSC_COMM_SELF,n,order,PETSC_COPY_VALUES,irow));
848fa24674SBarry Smith   *icol = *irow;
855f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectReference((PetscObject)*irow));
865f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree(done));
875f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree(order));
88d64ed03dSBarry Smith   PetscFunctionReturn(0);
893b2fbd54SBarry Smith }
903b2fbd54SBarry Smith 
914ac6704cSBarry Smith static PetscErrorCode MatFactorGetSolverType_petsc(Mat A,MatSolverType *type)
924ac6704cSBarry Smith {
934ac6704cSBarry Smith   PetscFunctionBegin;
944ac6704cSBarry Smith   *type = MATSOLVERPETSC;
954ac6704cSBarry Smith   PetscFunctionReturn(0);
964ac6704cSBarry Smith }
974ac6704cSBarry Smith 
98cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatGetFactor_seqaij_petsc(Mat A,MatFactorType ftype,Mat *B)
99b24902e0SBarry Smith {
100d0f46423SBarry Smith   PetscInt       n = A->rmap->n;
101b24902e0SBarry Smith 
102b24902e0SBarry Smith   PetscFunctionBegin;
103891bceaeSHong Zhang #if defined(PETSC_USE_COMPLEX)
1042c71b3e2SJacob Faibussowitsch   PetscCheckFalse(A->hermitian && !A->symmetric && (ftype == MAT_FACTOR_CHOLESKY||ftype == MAT_FACTOR_ICC),PETSC_COMM_SELF,PETSC_ERR_SUP,"Hermitian CHOLESKY or ICC Factor is not supported");
105891bceaeSHong Zhang #endif
1065f80ce2aSJacob Faibussowitsch   CHKERRQ(MatCreate(PetscObjectComm((PetscObject)A),B));
1075f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSetSizes(*B,n,n,n,n));
108599ef60dSHong Zhang   if (ftype == MAT_FACTOR_LU || ftype == MAT_FACTOR_ILU || ftype == MAT_FACTOR_ILUDT) {
1095f80ce2aSJacob Faibussowitsch     CHKERRQ(MatSetType(*B,MATSEQAIJ));
1102205254eSKarl Rupp 
11135233d99SShri Abhyankar     (*B)->ops->ilufactorsymbolic = MatILUFactorSymbolic_SeqAIJ;
11235233d99SShri Abhyankar     (*B)->ops->lufactorsymbolic  = MatLUFactorSymbolic_SeqAIJ;
1132205254eSKarl Rupp 
1145f80ce2aSJacob Faibussowitsch     CHKERRQ(MatSetBlockSizesFromMats(*B,A,A));
1155f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscStrallocpy(MATORDERINGND,(char**)&(*B)->preferredordering[MAT_FACTOR_LU]));
1165f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscStrallocpy(MATORDERINGNATURAL,(char**)&(*B)->preferredordering[MAT_FACTOR_ILU]));
1175f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscStrallocpy(MATORDERINGNATURAL,(char**)&(*B)->preferredordering[MAT_FACTOR_ILUDT]));
118b24902e0SBarry Smith   } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
1195f80ce2aSJacob Faibussowitsch     CHKERRQ(MatSetType(*B,MATSEQSBAIJ));
1205f80ce2aSJacob Faibussowitsch     CHKERRQ(MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,NULL));
1212205254eSKarl Rupp 
12235233d99SShri Abhyankar     (*B)->ops->iccfactorsymbolic      = MatICCFactorSymbolic_SeqAIJ;
12335233d99SShri Abhyankar     (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ;
1245f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscStrallocpy(MATORDERINGND,(char**)&(*B)->preferredordering[MAT_FACTOR_CHOLESKY]));
1255f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscStrallocpy(MATORDERINGNATURAL,(char**)&(*B)->preferredordering[MAT_FACTOR_ICC]));
126e32f2f54SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported");
127d5f3da31SBarry Smith   (*B)->factortype = ftype;
12800c67f3bSHong Zhang 
1295f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree((*B)->solvertype));
1305f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscStrallocpy(MATSOLVERPETSC,&(*B)->solvertype));
131f73b0415SBarry Smith   (*B)->canuseordering = PETSC_TRUE;
1325f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectComposeFunction((PetscObject)*B,"MatFactorGetSolverType_C",MatFactorGetSolverType_petsc));
133b24902e0SBarry Smith   PetscFunctionReturn(0);
134b24902e0SBarry Smith }
135b24902e0SBarry Smith 
136ad04f41aSHong Zhang PetscErrorCode MatLUFactorSymbolic_SeqAIJ_inplace(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
137289bc588SBarry Smith {
138416022c9SBarry Smith   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
139289bc588SBarry Smith   IS                 isicol;
1405d0c19d7SBarry Smith   const PetscInt     *r,*ic;
1415d0c19d7SBarry Smith   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j;
1421393bc97SHong Zhang   PetscInt           *bi,*bj,*ajtmp;
1431393bc97SHong Zhang   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
1449e046878SBarry Smith   PetscReal          f;
1454875ba38SHong Zhang   PetscInt           nlnk,*lnk,k,**bi_ptr;
1460298fd71SBarry Smith   PetscFreeSpaceList free_space=NULL,current_space=NULL;
1471393bc97SHong Zhang   PetscBT            lnkbt;
148ece542f9SHong Zhang   PetscBool          missing;
149289bc588SBarry Smith 
1503a40ed3dSBarry Smith   PetscFunctionBegin;
1512c71b3e2SJacob Faibussowitsch   PetscCheckFalse(A->rmap->N != A->cmap->N,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square");
1525f80ce2aSJacob Faibussowitsch   CHKERRQ(MatMissingDiagonal(A,&missing,&i));
153*28b400f6SJacob Faibussowitsch   PetscCheck(!missing,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %" PetscInt_FMT,i);
154ece542f9SHong Zhang 
1555f80ce2aSJacob Faibussowitsch   CHKERRQ(ISInvertPermutation(iscol,PETSC_DECIDE,&isicol));
1565f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isrow,&r));
1575f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isicol,&ic));
158289bc588SBarry Smith 
159289bc588SBarry Smith   /* get new row pointers */
1605f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(n+1,&bi));
1611393bc97SHong Zhang   bi[0] = 0;
1621393bc97SHong Zhang 
1631393bc97SHong Zhang   /* bdiag is location of diagonal in factor */
1645f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(n+1,&bdiag));
1651393bc97SHong Zhang   bdiag[0] = 0;
1661393bc97SHong Zhang 
1671393bc97SHong Zhang   /* linked list for storing column indices of the active row */
1681393bc97SHong Zhang   nlnk = n + 1;
1695f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLLCreate(n,n,nlnk,lnk,lnkbt));
1701393bc97SHong Zhang 
1715f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc2(n+1,&bi_ptr,n+1,&im));
1721393bc97SHong Zhang 
1731393bc97SHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
174d3d32019SBarry Smith   f             = info->fill;
175aa91b3e7SRichard Tran Mills   if (n==1)   f = 1; /* prevent failure in corner case of 1x1 matrix with fill < 0.5 */
1765f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space));
1771393bc97SHong Zhang   current_space = free_space;
178289bc588SBarry Smith 
179289bc588SBarry Smith   for (i=0; i<n; i++) {
1801393bc97SHong Zhang     /* copy previous fill into linked list */
181289bc588SBarry Smith     nzi = 0;
1821393bc97SHong Zhang     nnz = ai[r[i]+1] - ai[r[i]];
1831393bc97SHong Zhang     ajtmp = aj + ai[r[i]];
18494bad497SJacob Faibussowitsch     CHKERRQ(PetscLLAddPerm(nnz,ajtmp,ic,n,&nlnk,lnk,lnkbt));
1851393bc97SHong Zhang     nzi  += nlnk;
1861393bc97SHong Zhang 
1871393bc97SHong Zhang     /* add pivot rows into linked list */
1881393bc97SHong Zhang     row = lnk[n];
1891393bc97SHong Zhang     while (row < i) {
190d1170560SHong Zhang       nzbd  = bdiag[row] - bi[row] + 1;   /* num of entries in the row with column index <= row */
191d1170560SHong Zhang       ajtmp = bi_ptr[row] + nzbd;   /* points to the entry next to the diagonal */
19294bad497SJacob Faibussowitsch       CHKERRQ(PetscLLAddSortedLU(ajtmp,row,&nlnk,lnk,lnkbt,i,nzbd,im));
1931393bc97SHong Zhang       nzi  += nlnk;
1941393bc97SHong Zhang       row   = lnk[row];
195289bc588SBarry Smith     }
1961393bc97SHong Zhang     bi[i+1] = bi[i] + nzi;
1971393bc97SHong Zhang     im[i]   = nzi;
1981393bc97SHong Zhang 
1991393bc97SHong Zhang     /* mark bdiag */
2001393bc97SHong Zhang     nzbd = 0;
2011393bc97SHong Zhang     nnz  = nzi;
2021393bc97SHong Zhang     k    = lnk[n];
2031393bc97SHong Zhang     while (nnz-- && k < i) {
2041393bc97SHong Zhang       nzbd++;
2051393bc97SHong Zhang       k = lnk[k];
2061393bc97SHong Zhang     }
2071393bc97SHong Zhang     bdiag[i] = bi[i] + nzbd;
2081393bc97SHong Zhang 
2091393bc97SHong Zhang     /* if free space is not available, make more free space */
2101393bc97SHong Zhang     if (current_space->local_remaining<nzi) {
211f91af8c7SBarry Smith       nnz  = PetscIntMultTruncate(n - i,nzi); /* estimated and max additional space needed */
2125f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscFreeSpaceGet(nnz,&current_space));
2131393bc97SHong Zhang       reallocs++;
2141393bc97SHong Zhang     }
2151393bc97SHong Zhang 
2161393bc97SHong Zhang     /* copy data into free space, then initialize lnk */
2175f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt));
2182205254eSKarl Rupp 
2191393bc97SHong Zhang     bi_ptr[i]                       = current_space->array;
2201393bc97SHong Zhang     current_space->array           += nzi;
2211393bc97SHong Zhang     current_space->local_used      += nzi;
2221393bc97SHong Zhang     current_space->local_remaining -= nzi;
223289bc588SBarry Smith   }
2246cf91177SBarry Smith #if defined(PETSC_USE_INFO)
225464e8d44SSatish Balay   if (ai[n] != 0) {
2261393bc97SHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
2275f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"Reallocs %" PetscInt_FMT " Fill ratio:given %g needed %g\n",reallocs,(double)f,(double)af));
2285f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"Run with -pc_factor_fill %g or use \n",(double)af));
2295f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"PCFactorSetFill(pc,%g);\n",(double)af));
2305f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"for best performance.\n"));
23105bf559cSSatish Balay   } else {
2325f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"Empty matrix\n"));
23305bf559cSSatish Balay   }
23463ba0a88SBarry Smith #endif
2352fb3ed76SBarry Smith 
2365f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isrow,&r));
2375f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isicol,&ic));
2381987afe7SBarry Smith 
2391393bc97SHong Zhang   /* destroy list of free space and other temporary array(s) */
2405f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(bi[n]+1,&bj));
2415f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFreeSpaceContiguous(&free_space,bj));
2425f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLLDestroy(lnk,lnkbt));
2435f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree2(bi_ptr,im));
244289bc588SBarry Smith 
245289bc588SBarry Smith   /* put together the new matrix */
2465f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,NULL));
2475f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogObjectParent((PetscObject)B,(PetscObject)isicol));
248719d5645SBarry Smith   b    = (Mat_SeqAIJ*)(B)->data;
2492205254eSKarl Rupp 
250e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
251e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
2527c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
2532205254eSKarl Rupp 
2545f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(bi[n]+1,&b->a));
2551393bc97SHong Zhang   b->j    = bj;
2561393bc97SHong Zhang   b->i    = bi;
2571393bc97SHong Zhang   b->diag = bdiag;
258f4259b30SLisandro Dalcin   b->ilen = NULL;
259f4259b30SLisandro Dalcin   b->imax = NULL;
260416022c9SBarry Smith   b->row  = isrow;
261416022c9SBarry Smith   b->col  = iscol;
2625f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectReference((PetscObject)isrow));
2635f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectReference((PetscObject)iscol));
26482bf6240SBarry Smith   b->icol = isicol;
2655f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(n+1,&b->solve_work));
2661393bc97SHong Zhang 
2671393bc97SHong Zhang   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
2685f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogObjectMemory((PetscObject)B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar))));
2691393bc97SHong Zhang   b->maxnz = b->nz = bi[n];
2707fd98bd6SLois Curfman McInnes 
271d5f3da31SBarry Smith   (B)->factortype            = MAT_FACTOR_LU;
272719d5645SBarry Smith   (B)->info.factor_mallocs   = reallocs;
273719d5645SBarry Smith   (B)->info.fill_ratio_given = f;
274703deb49SSatish Balay 
2759f7d0b68SHong Zhang   if (ai[n]) {
276719d5645SBarry Smith     (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
277eea2913aSSatish Balay   } else {
278719d5645SBarry Smith     (B)->info.fill_ratio_needed = 0.0;
279eea2913aSSatish Balay   }
280ad04f41aSHong Zhang   (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace;
281f2344125SBarry Smith   if (a->inode.size) {
282f2344125SBarry Smith     (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
283f2344125SBarry Smith   }
2843a40ed3dSBarry Smith   PetscFunctionReturn(0);
285289bc588SBarry Smith }
2861393bc97SHong Zhang 
28735233d99SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
288ce3d78c0SShri Abhyankar {
289ce3d78c0SShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
290ce3d78c0SShri Abhyankar   IS                 isicol;
2918758e1faSBarry Smith   const PetscInt     *r,*ic,*ai=a->i,*aj=a->j,*ajtmp;
2928758e1faSBarry Smith   PetscInt           i,n=A->rmap->n;
2938758e1faSBarry Smith   PetscInt           *bi,*bj;
294ce3d78c0SShri Abhyankar   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
295ce3d78c0SShri Abhyankar   PetscReal          f;
296ce3d78c0SShri Abhyankar   PetscInt           nlnk,*lnk,k,**bi_ptr;
2970298fd71SBarry Smith   PetscFreeSpaceList free_space=NULL,current_space=NULL;
298ce3d78c0SShri Abhyankar   PetscBT            lnkbt;
299ece542f9SHong Zhang   PetscBool          missing;
300ce3d78c0SShri Abhyankar 
301ce3d78c0SShri Abhyankar   PetscFunctionBegin;
3022c71b3e2SJacob Faibussowitsch   PetscCheckFalse(A->rmap->N != A->cmap->N,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square");
3035f80ce2aSJacob Faibussowitsch   CHKERRQ(MatMissingDiagonal(A,&missing,&i));
304*28b400f6SJacob Faibussowitsch   PetscCheck(!missing,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %" PetscInt_FMT,i);
305ece542f9SHong Zhang 
3065f80ce2aSJacob Faibussowitsch   CHKERRQ(ISInvertPermutation(iscol,PETSC_DECIDE,&isicol));
3075f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isrow,&r));
3085f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isicol,&ic));
309ce3d78c0SShri Abhyankar 
3101bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
3115f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(n+1,&bi));
3125f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(n+1,&bdiag));
313a6df321fSShri Abhyankar   bi[0] = bdiag[0] = 0;
3149b48462bSShri Abhyankar 
3159b48462bSShri Abhyankar   /* linked list for storing column indices of the active row */
3169b48462bSShri Abhyankar   nlnk = n + 1;
3175f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLLCreate(n,n,nlnk,lnk,lnkbt));
3189b48462bSShri Abhyankar 
3195f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc2(n+1,&bi_ptr,n+1,&im));
3209b48462bSShri Abhyankar 
3219b48462bSShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
3229b48462bSShri Abhyankar   f             = info->fill;
323aa91b3e7SRichard Tran Mills   if (n==1)   f = 1; /* prevent failure in corner case of 1x1 matrix with fill < 0.5 */
3245f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space));
3259b48462bSShri Abhyankar   current_space = free_space;
3269b48462bSShri Abhyankar 
3279b48462bSShri Abhyankar   for (i=0; i<n; i++) {
3289b48462bSShri Abhyankar     /* copy previous fill into linked list */
3299b48462bSShri Abhyankar     nzi = 0;
3309b48462bSShri Abhyankar     nnz = ai[r[i]+1] - ai[r[i]];
3319b48462bSShri Abhyankar     ajtmp = aj + ai[r[i]];
33294bad497SJacob Faibussowitsch     CHKERRQ(PetscLLAddPerm(nnz,ajtmp,ic,n,&nlnk,lnk,lnkbt));
3339b48462bSShri Abhyankar     nzi  += nlnk;
3349b48462bSShri Abhyankar 
3359b48462bSShri Abhyankar     /* add pivot rows into linked list */
3369b48462bSShri Abhyankar     row = lnk[n];
3379b48462bSShri Abhyankar     while (row < i) {
3389b48462bSShri Abhyankar       nzbd  = bdiag[row] + 1; /* num of entries in the row with column index <= row */
3399b48462bSShri Abhyankar       ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
34094bad497SJacob Faibussowitsch       CHKERRQ(PetscLLAddSortedLU(ajtmp,row,&nlnk,lnk,lnkbt,i,nzbd,im));
3419b48462bSShri Abhyankar       nzi  += nlnk;
3429b48462bSShri Abhyankar       row   = lnk[row];
3439b48462bSShri Abhyankar     }
3449b48462bSShri Abhyankar     bi[i+1] = bi[i] + nzi;
3459b48462bSShri Abhyankar     im[i]   = nzi;
3469b48462bSShri Abhyankar 
3479b48462bSShri Abhyankar     /* mark bdiag */
3489b48462bSShri Abhyankar     nzbd = 0;
3499b48462bSShri Abhyankar     nnz  = nzi;
3509b48462bSShri Abhyankar     k    = lnk[n];
3519b48462bSShri Abhyankar     while (nnz-- && k < i) {
3529b48462bSShri Abhyankar       nzbd++;
3539b48462bSShri Abhyankar       k = lnk[k];
3549b48462bSShri Abhyankar     }
3559b48462bSShri Abhyankar     bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */
3569b48462bSShri Abhyankar 
3579b48462bSShri Abhyankar     /* if free space is not available, make more free space */
3589b48462bSShri Abhyankar     if (current_space->local_remaining<nzi) {
3592f18eb33SBarry Smith       /* estimated additional space needed */
3601df4278eSBarry Smith       nnz  = PetscIntMultTruncate(2,PetscIntMultTruncate(n-1,nzi));
3615f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscFreeSpaceGet(nnz,&current_space));
3629b48462bSShri Abhyankar       reallocs++;
3639b48462bSShri Abhyankar     }
3649b48462bSShri Abhyankar 
3659b48462bSShri Abhyankar     /* copy data into free space, then initialize lnk */
3665f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt));
3672205254eSKarl Rupp 
3689b48462bSShri Abhyankar     bi_ptr[i]                       = current_space->array;
3699b48462bSShri Abhyankar     current_space->array           += nzi;
3709b48462bSShri Abhyankar     current_space->local_used      += nzi;
3719b48462bSShri Abhyankar     current_space->local_remaining -= nzi;
3729b48462bSShri Abhyankar   }
3739b48462bSShri Abhyankar 
3745f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isrow,&r));
3755f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isicol,&ic));
3769b48462bSShri Abhyankar 
3779263d837SHong Zhang   /*   copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */
3785f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(bi[n]+1,&bj));
3795f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag));
3805f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLLDestroy(lnk,lnkbt));
3815f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree2(bi_ptr,im));
3829b48462bSShri Abhyankar 
3839b48462bSShri Abhyankar   /* put together the new matrix */
3845f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,NULL));
3855f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogObjectParent((PetscObject)B,(PetscObject)isicol));
3869b48462bSShri Abhyankar   b    = (Mat_SeqAIJ*)(B)->data;
3872205254eSKarl Rupp 
3889b48462bSShri Abhyankar   b->free_a       = PETSC_TRUE;
3899b48462bSShri Abhyankar   b->free_ij      = PETSC_TRUE;
3909b48462bSShri Abhyankar   b->singlemalloc = PETSC_FALSE;
3912205254eSKarl Rupp 
3925f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(bdiag[0]+1,&b->a));
3932205254eSKarl Rupp 
3949b48462bSShri Abhyankar   b->j    = bj;
3959b48462bSShri Abhyankar   b->i    = bi;
3969b48462bSShri Abhyankar   b->diag = bdiag;
397f4259b30SLisandro Dalcin   b->ilen = NULL;
398f4259b30SLisandro Dalcin   b->imax = NULL;
3999b48462bSShri Abhyankar   b->row  = isrow;
4009b48462bSShri Abhyankar   b->col  = iscol;
4015f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectReference((PetscObject)isrow));
4025f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectReference((PetscObject)iscol));
4039b48462bSShri Abhyankar   b->icol = isicol;
4045f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(n+1,&b->solve_work));
4059b48462bSShri Abhyankar 
4069b48462bSShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
4075f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogObjectMemory((PetscObject)B,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar))));
4089b48462bSShri Abhyankar   b->maxnz = b->nz = bdiag[0]+1;
4092205254eSKarl Rupp 
410d5f3da31SBarry Smith   B->factortype            = MAT_FACTOR_LU;
411f284f12aSHong Zhang   B->info.factor_mallocs   = reallocs;
412f284f12aSHong Zhang   B->info.fill_ratio_given = f;
4139b48462bSShri Abhyankar 
4149b48462bSShri Abhyankar   if (ai[n]) {
415f284f12aSHong Zhang     B->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
4169b48462bSShri Abhyankar   } else {
417f284f12aSHong Zhang     B->info.fill_ratio_needed = 0.0;
4189b48462bSShri Abhyankar   }
4199263d837SHong Zhang #if defined(PETSC_USE_INFO)
4209263d837SHong Zhang   if (ai[n] != 0) {
4219263d837SHong Zhang     PetscReal af = B->info.fill_ratio_needed;
4225f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"Reallocs %" PetscInt_FMT " Fill ratio:given %g needed %g\n",reallocs,(double)f,(double)af));
4235f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"Run with -pc_factor_fill %g or use \n",(double)af));
4245f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"PCFactorSetFill(pc,%g);\n",(double)af));
4255f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"for best performance.\n"));
4269263d837SHong Zhang   } else {
4275f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"Empty matrix\n"));
4289263d837SHong Zhang   }
4299263d837SHong Zhang #endif
43035233d99SShri Abhyankar   B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ;
431f2344125SBarry Smith   if (a->inode.size) {
432d3ac4fa3SBarry Smith     B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode;
433d3ac4fa3SBarry Smith   }
4345f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSeqAIJCheckInode_FactorLU(B));
4359b48462bSShri Abhyankar   PetscFunctionReturn(0);
4369b48462bSShri Abhyankar }
4379b48462bSShri Abhyankar 
4385b5bc046SBarry Smith /*
4395b5bc046SBarry Smith     Trouble in factorization, should we dump the original matrix?
4405b5bc046SBarry Smith */
4415b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A)
4425b5bc046SBarry Smith {
443ace3abfcSBarry Smith   PetscBool      flg = PETSC_FALSE;
4445b5bc046SBarry Smith 
4455b5bc046SBarry Smith   PetscFunctionBegin;
4465f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsGetBool(((PetscObject)A)->options,NULL,"-mat_factor_dump_on_error",&flg,NULL));
4475b5bc046SBarry Smith   if (flg) {
4485b5bc046SBarry Smith     PetscViewer viewer;
4495b5bc046SBarry Smith     char        filename[PETSC_MAX_PATH_LEN];
4505b5bc046SBarry Smith 
4515f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank));
4525f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscViewerBinaryOpen(PetscObjectComm((PetscObject)A),filename,FILE_MODE_WRITE,&viewer));
4535f80ce2aSJacob Faibussowitsch     CHKERRQ(MatView(A,viewer));
4545f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscViewerDestroy(&viewer));
4555b5bc046SBarry Smith   }
4565b5bc046SBarry Smith   PetscFunctionReturn(0);
4575b5bc046SBarry Smith }
4585b5bc046SBarry Smith 
45935233d99SShri Abhyankar PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
460c9c72f8fSHong Zhang {
461c9c72f8fSHong Zhang   Mat             C     =B;
462c9c72f8fSHong Zhang   Mat_SeqAIJ      *a    =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data;
463c9c72f8fSHong Zhang   IS              isrow = b->row,isicol = b->icol;
464c9c72f8fSHong Zhang   const PetscInt  *r,*ic,*ics;
465bbd65245SShri Abhyankar   const PetscInt  n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bdiag=b->diag;
466bbd65245SShri Abhyankar   PetscInt        i,j,k,nz,nzL,row,*pj;
467bbd65245SShri Abhyankar   const PetscInt  *ajtmp,*bjtmp;
468bbd65245SShri Abhyankar   MatScalar       *rtmp,*pc,multiplier,*pv;
469bbd65245SShri Abhyankar   const MatScalar *aa=a->a,*v;
470ace3abfcSBarry Smith   PetscBool       row_identity,col_identity;
471d90ac83dSHong Zhang   FactorShiftCtx  sctx;
4724f81c4b7SBarry Smith   const PetscInt  *ddiag;
473d4ad06f7SHong Zhang   PetscReal       rs;
474d4ad06f7SHong Zhang   MatScalar       d;
475d4ad06f7SHong Zhang 
476c9c72f8fSHong Zhang   PetscFunctionBegin;
477d6e5152cSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
4785f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMemzero(&sctx,sizeof(FactorShiftCtx)));
479d4ad06f7SHong Zhang 
480f4db908eSBarry Smith   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
481d4ad06f7SHong Zhang     ddiag          = a->diag;
482d4ad06f7SHong Zhang     sctx.shift_top = info->zeropivot;
483d4ad06f7SHong Zhang     for (i=0; i<n; i++) {
484d4ad06f7SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
485d4ad06f7SHong Zhang       d  = (aa)[ddiag[i]];
486d4ad06f7SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
487d4ad06f7SHong Zhang       v  = aa+ai[i];
488d4ad06f7SHong Zhang       nz = ai[i+1] - ai[i];
4892205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
490d4ad06f7SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
491d4ad06f7SHong Zhang     }
492d4ad06f7SHong Zhang     sctx.shift_top *= 1.1;
493d4ad06f7SHong Zhang     sctx.nshift_max = 5;
494d4ad06f7SHong Zhang     sctx.shift_lo   = 0.;
495d4ad06f7SHong Zhang     sctx.shift_hi   = 1.;
496d4ad06f7SHong Zhang   }
497d4ad06f7SHong Zhang 
4985f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isrow,&r));
4995f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isicol,&ic));
5005f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(n+1,&rtmp));
501c9c72f8fSHong Zhang   ics  = ic;
502c9c72f8fSHong Zhang 
503d4ad06f7SHong Zhang   do {
50407b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
505c9c72f8fSHong Zhang     for (i=0; i<n; i++) {
506c9c72f8fSHong Zhang       /* zero rtmp */
507c9c72f8fSHong Zhang       /* L part */
508c9c72f8fSHong Zhang       nz    = bi[i+1] - bi[i];
509c9c72f8fSHong Zhang       bjtmp = bj + bi[i];
510c9c72f8fSHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
511c9c72f8fSHong Zhang 
512c9c72f8fSHong Zhang       /* U part */
51362fbd6afSShri Abhyankar       nz    = bdiag[i]-bdiag[i+1];
51462fbd6afSShri Abhyankar       bjtmp = bj + bdiag[i+1]+1;
515813a1f2bSShri Abhyankar       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
516813a1f2bSShri Abhyankar 
517813a1f2bSShri Abhyankar       /* load in initial (unfactored row) */
518813a1f2bSShri Abhyankar       nz    = ai[r[i]+1] - ai[r[i]];
519813a1f2bSShri Abhyankar       ajtmp = aj + ai[r[i]];
520813a1f2bSShri Abhyankar       v     = aa + ai[r[i]];
521813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
522813a1f2bSShri Abhyankar         rtmp[ics[ajtmp[j]]] = v[j];
523813a1f2bSShri Abhyankar       }
524813a1f2bSShri Abhyankar       /* ZeropivotApply() */
52598a93161SHong Zhang       rtmp[i] += sctx.shift_amount;  /* shift the diagonal of the matrix */
526813a1f2bSShri Abhyankar 
527813a1f2bSShri Abhyankar       /* elimination */
528813a1f2bSShri Abhyankar       bjtmp = bj + bi[i];
529813a1f2bSShri Abhyankar       row   = *bjtmp++;
530813a1f2bSShri Abhyankar       nzL   = bi[i+1] - bi[i];
531f268cba8SShri Abhyankar       for (k=0; k < nzL; k++) {
532813a1f2bSShri Abhyankar         pc = rtmp + row;
533813a1f2bSShri Abhyankar         if (*pc != 0.0) {
534813a1f2bSShri Abhyankar           pv         = b->a + bdiag[row];
535813a1f2bSShri Abhyankar           multiplier = *pc * (*pv);
536813a1f2bSShri Abhyankar           *pc        = multiplier;
5372205254eSKarl Rupp 
53862fbd6afSShri Abhyankar           pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */
53962fbd6afSShri Abhyankar           pv = b->a + bdiag[row+1]+1;
54062fbd6afSShri Abhyankar           nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
5412205254eSKarl Rupp 
542813a1f2bSShri Abhyankar           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
5435f80ce2aSJacob Faibussowitsch           CHKERRQ(PetscLogFlops(1+2.0*nz));
544813a1f2bSShri Abhyankar         }
545f268cba8SShri Abhyankar         row = *bjtmp++;
546813a1f2bSShri Abhyankar       }
547813a1f2bSShri Abhyankar 
548813a1f2bSShri Abhyankar       /* finished row so stick it into b->a */
549813a1f2bSShri Abhyankar       rs = 0.0;
550813a1f2bSShri Abhyankar       /* L part */
551813a1f2bSShri Abhyankar       pv = b->a + bi[i];
552813a1f2bSShri Abhyankar       pj = b->j + bi[i];
553813a1f2bSShri Abhyankar       nz = bi[i+1] - bi[i];
554813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
555813a1f2bSShri Abhyankar         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
556813a1f2bSShri Abhyankar       }
557813a1f2bSShri Abhyankar 
558813a1f2bSShri Abhyankar       /* U part */
55962fbd6afSShri Abhyankar       pv = b->a + bdiag[i+1]+1;
56062fbd6afSShri Abhyankar       pj = b->j + bdiag[i+1]+1;
56162fbd6afSShri Abhyankar       nz = bdiag[i] - bdiag[i+1]-1;
562813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
563813a1f2bSShri Abhyankar         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
564813a1f2bSShri Abhyankar       }
565813a1f2bSShri Abhyankar 
566813a1f2bSShri Abhyankar       sctx.rs = rs;
567813a1f2bSShri Abhyankar       sctx.pv = rtmp[i];
5685f80ce2aSJacob Faibussowitsch       CHKERRQ(MatPivotCheck(B,A,info,&sctx,i));
56907b50cabSHong Zhang       if (sctx.newshift) break; /* break for-loop */
57007b50cabSHong Zhang       rtmp[i] = sctx.pv; /* sctx.pv might be updated in the case of MAT_SHIFT_INBLOCKS */
571813a1f2bSShri Abhyankar 
572a5b23f4aSJose E. Roman       /* Mark diagonal and invert diagonal for simpler triangular solves */
573813a1f2bSShri Abhyankar       pv  = b->a + bdiag[i];
574813a1f2bSShri Abhyankar       *pv = 1.0/rtmp[i];
575813a1f2bSShri Abhyankar 
576813a1f2bSShri Abhyankar     } /* endof for (i=0; i<n; i++) { */
577813a1f2bSShri Abhyankar 
5788ff23777SHong Zhang     /* MatPivotRefine() */
57907b50cabSHong Zhang     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
580813a1f2bSShri Abhyankar       /*
581813a1f2bSShri Abhyankar        * if no shift in this attempt & shifting & started shifting & can refine,
582813a1f2bSShri Abhyankar        * then try lower shift
583813a1f2bSShri Abhyankar        */
584813a1f2bSShri Abhyankar       sctx.shift_hi       = sctx.shift_fraction;
585813a1f2bSShri Abhyankar       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
586813a1f2bSShri Abhyankar       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
58707b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
588813a1f2bSShri Abhyankar       sctx.nshift++;
589813a1f2bSShri Abhyankar     }
59007b50cabSHong Zhang   } while (sctx.newshift);
591813a1f2bSShri Abhyankar 
5925f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree(rtmp));
5935f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isicol,&ic));
5945f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isrow,&r));
595d3ac4fa3SBarry Smith 
5965f80ce2aSJacob Faibussowitsch   CHKERRQ(ISIdentity(isrow,&row_identity));
5975f80ce2aSJacob Faibussowitsch   CHKERRQ(ISIdentity(isicol,&col_identity));
598abb87a52SBarry Smith   if (b->inode.size) {
599abb87a52SBarry Smith     C->ops->solve = MatSolve_SeqAIJ_Inode;
600abb87a52SBarry Smith   } else if (row_identity && col_identity) {
60135233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
602813a1f2bSShri Abhyankar   } else {
60335233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ;
604813a1f2bSShri Abhyankar   }
60535233d99SShri Abhyankar   C->ops->solveadd          = MatSolveAdd_SeqAIJ;
60635233d99SShri Abhyankar   C->ops->solvetranspose    = MatSolveTranspose_SeqAIJ;
60735233d99SShri Abhyankar   C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ;
60835233d99SShri Abhyankar   C->ops->matsolve          = MatMatSolve_SeqAIJ;
609813a1f2bSShri Abhyankar   C->assembled              = PETSC_TRUE;
610813a1f2bSShri Abhyankar   C->preallocated           = PETSC_TRUE;
6112205254eSKarl Rupp 
6125f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(C->cmap->n));
613813a1f2bSShri Abhyankar 
61414d2772eSHong Zhang   /* MatShiftView(A,info,&sctx) */
615813a1f2bSShri Abhyankar   if (sctx.nshift) {
616f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
6175f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscInfo(A,"number of shift_pd tries %" PetscInt_FMT ", 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));
618f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
6195f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscInfo(A,"number of shift_nz tries %" PetscInt_FMT ", shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount));
620f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS) {
6215f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscInfo(A,"number of shift_inblocks applied %" PetscInt_FMT ", each shift_amount %g\n",sctx.nshift,(double)info->shiftamount));
622813a1f2bSShri Abhyankar     }
623813a1f2bSShri Abhyankar   }
624813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
625813a1f2bSShri Abhyankar }
626813a1f2bSShri Abhyankar 
627ad04f41aSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info)
628289bc588SBarry Smith {
629719d5645SBarry Smith   Mat             C     =B;
630416022c9SBarry Smith   Mat_SeqAIJ      *a    =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data;
63182bf6240SBarry Smith   IS              isrow = b->row,isicol = b->icol;
6325d0c19d7SBarry Smith   const PetscInt  *r,*ic,*ics;
633d278edc7SBarry Smith   PetscInt        nz,row,i,j,n=A->rmap->n,diag;
634d278edc7SBarry Smith   const PetscInt  *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
635d278edc7SBarry Smith   const PetscInt  *ajtmp,*bjtmp,*diag_offset = b->diag,*pj;
636d278edc7SBarry Smith   MatScalar       *pv,*rtmp,*pc,multiplier,d;
637d278edc7SBarry Smith   const MatScalar *v,*aa=a->a;
63833f9893dSHong Zhang   PetscReal       rs=0.0;
639d90ac83dSHong Zhang   FactorShiftCtx  sctx;
6404f81c4b7SBarry Smith   const PetscInt  *ddiag;
641ace3abfcSBarry Smith   PetscBool       row_identity, col_identity;
642289bc588SBarry Smith 
6433a40ed3dSBarry Smith   PetscFunctionBegin;
644504a3fadSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
6455f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMemzero(&sctx,sizeof(FactorShiftCtx)));
646ada7143aSSatish Balay 
647f4db908eSBarry Smith   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
64842898933SHong Zhang     ddiag          = a->diag;
6499f7d0b68SHong Zhang     sctx.shift_top = info->zeropivot;
6506cc28720Svictorle     for (i=0; i<n; i++) {
6519f95998fSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
6529f7d0b68SHong Zhang       d  = (aa)[ddiag[i]];
6531a890ab1SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
6549f7d0b68SHong Zhang       v  = aa+ai[i];
6559f7d0b68SHong Zhang       nz = ai[i+1] - ai[i];
6562205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
6571a890ab1SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
6586cc28720Svictorle     }
659118f5789SBarry Smith     sctx.shift_top *= 1.1;
660d2276718SHong Zhang     sctx.nshift_max = 5;
661d2276718SHong Zhang     sctx.shift_lo   = 0.;
662d2276718SHong Zhang     sctx.shift_hi   = 1.;
663a0a356efSHong Zhang   }
664a0a356efSHong Zhang 
6655f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isrow,&r));
6665f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isicol,&ic));
6675f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(n+1,&rtmp));
668504a3fadSHong Zhang   ics  = ic;
669504a3fadSHong Zhang 
670085256b3SBarry Smith   do {
67107b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
672289bc588SBarry Smith     for (i=0; i<n; i++) {
673b3bf805bSHong Zhang       nz    = bi[i+1] - bi[i];
674b3bf805bSHong Zhang       bjtmp = bj + bi[i];
6759f7d0b68SHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
676289bc588SBarry Smith 
677289bc588SBarry Smith       /* load in initial (unfactored row) */
6789f7d0b68SHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
6799f7d0b68SHong Zhang       ajtmp = aj + ai[r[i]];
6809f7d0b68SHong Zhang       v     = aa + ai[r[i]];
681085256b3SBarry Smith       for (j=0; j<nz; j++) {
682b3bf805bSHong Zhang         rtmp[ics[ajtmp[j]]] = v[j];
683085256b3SBarry Smith       }
684d2276718SHong Zhang       rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */
685289bc588SBarry Smith 
686b3bf805bSHong Zhang       row = *bjtmp++;
687f2582111SSatish Balay       while  (row < i) {
6888c37ef55SBarry Smith         pc = rtmp + row;
6898c37ef55SBarry Smith         if (*pc != 0.0) {
690010a20caSHong Zhang           pv         = b->a + diag_offset[row];
691010a20caSHong Zhang           pj         = b->j + diag_offset[row] + 1;
69235aab85fSBarry Smith           multiplier = *pc / *pv++;
6938c37ef55SBarry Smith           *pc        = multiplier;
694b3bf805bSHong Zhang           nz         = bi[row+1] - diag_offset[row] - 1;
6959f7d0b68SHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
6965f80ce2aSJacob Faibussowitsch           CHKERRQ(PetscLogFlops(1+2.0*nz));
697289bc588SBarry Smith         }
698b3bf805bSHong Zhang         row = *bjtmp++;
699289bc588SBarry Smith       }
700416022c9SBarry Smith       /* finished row so stick it into b->a */
701b3bf805bSHong Zhang       pv   = b->a + bi[i];
702b3bf805bSHong Zhang       pj   = b->j + bi[i];
703b3bf805bSHong Zhang       nz   = bi[i+1] - bi[i];
704b3bf805bSHong Zhang       diag = diag_offset[i] - bi[i];
705e57ff13aSBarry Smith       rs   = 0.0;
706d3d32019SBarry Smith       for (j=0; j<nz; j++) {
7079f7d0b68SHong Zhang         pv[j] = rtmp[pj[j]];
7089f7d0b68SHong Zhang         rs   += PetscAbsScalar(pv[j]);
709d3d32019SBarry Smith       }
710e57ff13aSBarry Smith       rs -= PetscAbsScalar(pv[diag]);
711d2276718SHong Zhang 
712d2276718SHong Zhang       sctx.rs = rs;
713d2276718SHong Zhang       sctx.pv = pv[diag];
7145f80ce2aSJacob Faibussowitsch       CHKERRQ(MatPivotCheck(B,A,info,&sctx,i));
71507b50cabSHong Zhang       if (sctx.newshift) break;
716504a3fadSHong Zhang       pv[diag] = sctx.pv;
71735aab85fSBarry Smith     }
718d2276718SHong Zhang 
71907b50cabSHong Zhang     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
7206cc28720Svictorle       /*
7216c037d1bSvictorle        * if no shift in this attempt & shifting & started shifting & can refine,
7226cc28720Svictorle        * then try lower shift
7236cc28720Svictorle        */
7240481f469SBarry Smith       sctx.shift_hi       = sctx.shift_fraction;
7250481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
7260481f469SBarry Smith       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
72707b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
728d2276718SHong Zhang       sctx.nshift++;
7296cc28720Svictorle     }
73007b50cabSHong Zhang   } while (sctx.newshift);
7310f11f9aeSSatish Balay 
732a5b23f4aSJose E. Roman   /* invert diagonal entries for simpler triangular solves */
73335aab85fSBarry Smith   for (i=0; i<n; i++) {
734010a20caSHong Zhang     b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]];
73535aab85fSBarry Smith   }
7365f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree(rtmp));
7375f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isicol,&ic));
7385f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isrow,&r));
739d3ac4fa3SBarry Smith 
7405f80ce2aSJacob Faibussowitsch   CHKERRQ(ISIdentity(isrow,&row_identity));
7415f80ce2aSJacob Faibussowitsch   CHKERRQ(ISIdentity(isicol,&col_identity));
7428d8c7c43SBarry Smith   if (row_identity && col_identity) {
743ad04f41aSHong Zhang     C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_inplace;
7448d8c7c43SBarry Smith   } else {
745ad04f41aSHong Zhang     C->ops->solve = MatSolve_SeqAIJ_inplace;
746db4efbfdSBarry Smith   }
747ad04f41aSHong Zhang   C->ops->solveadd          = MatSolveAdd_SeqAIJ_inplace;
748ad04f41aSHong Zhang   C->ops->solvetranspose    = MatSolveTranspose_SeqAIJ_inplace;
749ad04f41aSHong Zhang   C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace;
750ad04f41aSHong Zhang   C->ops->matsolve          = MatMatSolve_SeqAIJ_inplace;
7512205254eSKarl Rupp 
752c456f294SBarry Smith   C->assembled    = PETSC_TRUE;
7535c9eb25fSBarry Smith   C->preallocated = PETSC_TRUE;
7542205254eSKarl Rupp 
7555f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(C->cmap->n));
756d2276718SHong Zhang   if (sctx.nshift) {
757f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
7585f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscInfo(A,"number of shift_pd tries %" PetscInt_FMT ", 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));
759f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
7605f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscInfo(A,"number of shift_nz tries %" PetscInt_FMT ", shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount));
7616cc28720Svictorle     }
7620697b6caSHong Zhang   }
763d3ac4fa3SBarry Smith   (C)->ops->solve          = MatSolve_SeqAIJ_inplace;
764d3ac4fa3SBarry Smith   (C)->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace;
7652205254eSKarl Rupp 
7665f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSeqAIJCheckInode(C));
7673a40ed3dSBarry Smith   PetscFunctionReturn(0);
768289bc588SBarry Smith }
7690889b5dcSvictorle 
770137fb511SHong Zhang /*
771137fb511SHong Zhang    This routine implements inplace ILU(0) with row or/and column permutations.
772137fb511SHong Zhang    Input:
773137fb511SHong Zhang      A - original matrix
774137fb511SHong Zhang    Output;
775137fb511SHong Zhang      A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i]
776137fb511SHong Zhang          a->j (col index) is permuted by the inverse of colperm, then sorted
777137fb511SHong Zhang          a->a reordered accordingly with a->j
778137fb511SHong Zhang          a->diag (ptr to diagonal elements) is updated.
779137fb511SHong Zhang */
7800481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info)
781137fb511SHong Zhang {
782b051339dSHong Zhang   Mat_SeqAIJ      *a    =(Mat_SeqAIJ*)A->data;
783b051339dSHong Zhang   IS              isrow = a->row,isicol = a->icol;
7845d0c19d7SBarry Smith   const PetscInt  *r,*ic,*ics;
7855d0c19d7SBarry Smith   PetscInt        i,j,n=A->rmap->n,*ai=a->i,*aj=a->j;
7865d0c19d7SBarry Smith   PetscInt        *ajtmp,nz,row;
787b051339dSHong Zhang   PetscInt        *diag = a->diag,nbdiag,*pj;
788a77337e4SBarry Smith   PetscScalar     *rtmp,*pc,multiplier,d;
789504a3fadSHong Zhang   MatScalar       *pv,*v;
790137fb511SHong Zhang   PetscReal       rs;
791d90ac83dSHong Zhang   FactorShiftCtx  sctx;
792504a3fadSHong Zhang   const MatScalar *aa=a->a,*vtmp;
793137fb511SHong Zhang 
794137fb511SHong Zhang   PetscFunctionBegin;
7952c71b3e2SJacob Faibussowitsch   PetscCheckFalse(A != B,PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address");
796504a3fadSHong Zhang 
797504a3fadSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
7985f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMemzero(&sctx,sizeof(FactorShiftCtx)));
799504a3fadSHong Zhang 
800504a3fadSHong Zhang   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
801504a3fadSHong Zhang     const PetscInt *ddiag = a->diag;
802504a3fadSHong Zhang     sctx.shift_top = info->zeropivot;
803504a3fadSHong Zhang     for (i=0; i<n; i++) {
804504a3fadSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
805504a3fadSHong Zhang       d    = (aa)[ddiag[i]];
806504a3fadSHong Zhang       rs   = -PetscAbsScalar(d) - PetscRealPart(d);
807504a3fadSHong Zhang       vtmp = aa+ai[i];
808504a3fadSHong Zhang       nz   = ai[i+1] - ai[i];
8092205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(vtmp[j]);
810504a3fadSHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
811504a3fadSHong Zhang     }
812504a3fadSHong Zhang     sctx.shift_top *= 1.1;
813504a3fadSHong Zhang     sctx.nshift_max = 5;
814504a3fadSHong Zhang     sctx.shift_lo   = 0.;
815504a3fadSHong Zhang     sctx.shift_hi   = 1.;
816504a3fadSHong Zhang   }
817504a3fadSHong Zhang 
8185f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isrow,&r));
8195f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isicol,&ic));
8205f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(n+1,&rtmp));
8215f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscArrayzero(rtmp,n+1));
822b051339dSHong Zhang   ics  = ic;
823137fb511SHong Zhang 
824504a3fadSHong Zhang #if defined(MV)
82575567043SBarry Smith   sctx.shift_top      = 0.;
826e60cf9a0SBarry Smith   sctx.nshift_max     = 0;
82775567043SBarry Smith   sctx.shift_lo       = 0.;
82875567043SBarry Smith   sctx.shift_hi       = 0.;
82975567043SBarry Smith   sctx.shift_fraction = 0.;
830137fb511SHong Zhang 
831f4db908eSBarry Smith   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
83275567043SBarry Smith     sctx.shift_top = 0.;
833137fb511SHong Zhang     for (i=0; i<n; i++) {
834137fb511SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
835b051339dSHong Zhang       d  = (a->a)[diag[i]];
836137fb511SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
837b051339dSHong Zhang       v  = a->a+ai[i];
838b051339dSHong Zhang       nz = ai[i+1] - ai[i];
8392205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
840137fb511SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
841137fb511SHong Zhang     }
842137fb511SHong Zhang     if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot;
843137fb511SHong Zhang     sctx.shift_top *= 1.1;
844137fb511SHong Zhang     sctx.nshift_max = 5;
845137fb511SHong Zhang     sctx.shift_lo   = 0.;
846137fb511SHong Zhang     sctx.shift_hi   = 1.;
847137fb511SHong Zhang   }
848137fb511SHong Zhang 
84975567043SBarry Smith   sctx.shift_amount = 0.;
850137fb511SHong Zhang   sctx.nshift       = 0;
851504a3fadSHong Zhang #endif
852504a3fadSHong Zhang 
853137fb511SHong Zhang   do {
85407b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
855137fb511SHong Zhang     for (i=0; i<n; i++) {
856b051339dSHong Zhang       /* load in initial unfactored row */
857b051339dSHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
858b051339dSHong Zhang       ajtmp = aj + ai[r[i]];
859b051339dSHong Zhang       v     = a->a + ai[r[i]];
860137fb511SHong Zhang       /* sort permuted ajtmp and values v accordingly */
861b051339dSHong Zhang       for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]];
8625f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscSortIntWithScalarArray(nz,ajtmp,v));
863137fb511SHong Zhang 
864b051339dSHong Zhang       diag[r[i]] = ai[r[i]];
865137fb511SHong Zhang       for (j=0; j<nz; j++) {
866137fb511SHong Zhang         rtmp[ajtmp[j]] = v[j];
867b051339dSHong Zhang         if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */
868137fb511SHong Zhang       }
869137fb511SHong Zhang       rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */
870137fb511SHong Zhang 
871b051339dSHong Zhang       row = *ajtmp++;
872137fb511SHong Zhang       while  (row < i) {
873137fb511SHong Zhang         pc = rtmp + row;
874137fb511SHong Zhang         if (*pc != 0.0) {
875b051339dSHong Zhang           pv = a->a + diag[r[row]];
876b051339dSHong Zhang           pj = aj + diag[r[row]] + 1;
877137fb511SHong Zhang 
878137fb511SHong Zhang           multiplier = *pc / *pv++;
879137fb511SHong Zhang           *pc        = multiplier;
880b051339dSHong Zhang           nz         = ai[r[row]+1] - diag[r[row]] - 1;
881b051339dSHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
8825f80ce2aSJacob Faibussowitsch           CHKERRQ(PetscLogFlops(1+2.0*nz));
883137fb511SHong Zhang         }
884b051339dSHong Zhang         row = *ajtmp++;
885137fb511SHong Zhang       }
886b051339dSHong Zhang       /* finished row so overwrite it onto a->a */
887b051339dSHong Zhang       pv     = a->a + ai[r[i]];
888b051339dSHong Zhang       pj     = aj + ai[r[i]];
889b051339dSHong Zhang       nz     = ai[r[i]+1] - ai[r[i]];
890b051339dSHong Zhang       nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */
891137fb511SHong Zhang 
892137fb511SHong Zhang       rs = 0.0;
893137fb511SHong Zhang       for (j=0; j<nz; j++) {
894b051339dSHong Zhang         pv[j] = rtmp[pj[j]];
895b051339dSHong Zhang         if (j != nbdiag) rs += PetscAbsScalar(pv[j]);
896137fb511SHong Zhang       }
897137fb511SHong Zhang 
898137fb511SHong Zhang       sctx.rs = rs;
899b051339dSHong Zhang       sctx.pv = pv[nbdiag];
9005f80ce2aSJacob Faibussowitsch       CHKERRQ(MatPivotCheck(B,A,info,&sctx,i));
90107b50cabSHong Zhang       if (sctx.newshift) break;
902504a3fadSHong Zhang       pv[nbdiag] = sctx.pv;
903137fb511SHong Zhang     }
904137fb511SHong Zhang 
90507b50cabSHong Zhang     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
906137fb511SHong Zhang       /*
907137fb511SHong Zhang        * if no shift in this attempt & shifting & started shifting & can refine,
908137fb511SHong Zhang        * then try lower shift
909137fb511SHong Zhang        */
9100481f469SBarry Smith       sctx.shift_hi       = sctx.shift_fraction;
9110481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
9120481f469SBarry Smith       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
91307b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
914137fb511SHong Zhang       sctx.nshift++;
915137fb511SHong Zhang     }
91607b50cabSHong Zhang   } while (sctx.newshift);
917137fb511SHong Zhang 
918a5b23f4aSJose E. Roman   /* invert diagonal entries for simpler triangular solves */
919137fb511SHong Zhang   for (i=0; i<n; i++) {
920b051339dSHong Zhang     a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]];
921137fb511SHong Zhang   }
922137fb511SHong Zhang 
9235f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree(rtmp));
9245f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isicol,&ic));
9255f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isrow,&r));
9262205254eSKarl Rupp 
927b051339dSHong Zhang   A->ops->solve             = MatSolve_SeqAIJ_InplaceWithPerm;
928ad04f41aSHong Zhang   A->ops->solveadd          = MatSolveAdd_SeqAIJ_inplace;
929ad04f41aSHong Zhang   A->ops->solvetranspose    = MatSolveTranspose_SeqAIJ_inplace;
930ad04f41aSHong Zhang   A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace;
9312205254eSKarl Rupp 
932b051339dSHong Zhang   A->assembled    = PETSC_TRUE;
9335c9eb25fSBarry Smith   A->preallocated = PETSC_TRUE;
9342205254eSKarl Rupp 
9355f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(A->cmap->n));
936137fb511SHong Zhang   if (sctx.nshift) {
937f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
9385f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscInfo(A,"number of shift_pd tries %" PetscInt_FMT ", 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));
939f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
9405f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscInfo(A,"number of shift_nz tries %" PetscInt_FMT ", shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount));
941137fb511SHong Zhang     }
942137fb511SHong Zhang   }
943137fb511SHong Zhang   PetscFunctionReturn(0);
944137fb511SHong Zhang }
945137fb511SHong Zhang 
9460a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
9470481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info)
948da3a660dSBarry Smith {
949416022c9SBarry Smith   Mat            C;
950416022c9SBarry Smith 
9513a40ed3dSBarry Smith   PetscFunctionBegin;
9525f80ce2aSJacob Faibussowitsch   CHKERRQ(MatGetFactor(A,MATSOLVERPETSC,MAT_FACTOR_LU,&C));
9535f80ce2aSJacob Faibussowitsch   CHKERRQ(MatLUFactorSymbolic(C,A,row,col,info));
9545f80ce2aSJacob Faibussowitsch   CHKERRQ(MatLUFactorNumeric(C,A,info));
9552205254eSKarl Rupp 
956db4efbfdSBarry Smith   A->ops->solve          = C->ops->solve;
957db4efbfdSBarry Smith   A->ops->solvetranspose = C->ops->solvetranspose;
9582205254eSKarl Rupp 
9595f80ce2aSJacob Faibussowitsch   CHKERRQ(MatHeaderMerge(A,&C));
9605f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogObjectParent((PetscObject)A,(PetscObject)((Mat_SeqAIJ*)(A->data))->icol));
9613a40ed3dSBarry Smith   PetscFunctionReturn(0);
962da3a660dSBarry Smith }
9630a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
9641d098868SHong Zhang 
965ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_inplace(Mat A,Vec bb,Vec xx)
9668c37ef55SBarry Smith {
967416022c9SBarry Smith   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
968416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
9695d0c19d7SBarry Smith   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
9705d0c19d7SBarry Smith   PetscInt          nz;
9715d0c19d7SBarry Smith   const PetscInt    *rout,*cout,*r,*c;
972dd6ea824SBarry Smith   PetscScalar       *x,*tmp,*tmps,sum;
973d9fead3dSBarry Smith   const PetscScalar *b;
974dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
9758c37ef55SBarry Smith 
9763a40ed3dSBarry Smith   PetscFunctionBegin;
9773a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
97891bf9adeSBarry Smith 
9795f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayRead(bb,&b));
9805f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayWrite(xx,&x));
981416022c9SBarry Smith   tmp  = a->solve_work;
9828c37ef55SBarry Smith 
9835f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isrow,&rout)); r = rout;
9845f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(iscol,&cout)); c = cout + (n-1);
9858c37ef55SBarry Smith 
9868c37ef55SBarry Smith   /* forward solve the lower triangular */
9878c37ef55SBarry Smith   tmp[0] = b[*r++];
988010a20caSHong Zhang   tmps   = tmp;
9898c37ef55SBarry Smith   for (i=1; i<n; i++) {
990010a20caSHong Zhang     v   = aa + ai[i];
991010a20caSHong Zhang     vi  = aj + ai[i];
99253e7425aSBarry Smith     nz  = a->diag[i] - ai[i];
99353e7425aSBarry Smith     sum = b[*r++];
994003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
9958c37ef55SBarry Smith     tmp[i] = sum;
9968c37ef55SBarry Smith   }
9978c37ef55SBarry Smith 
9988c37ef55SBarry Smith   /* backward solve the upper triangular */
9998c37ef55SBarry Smith   for (i=n-1; i>=0; i--) {
1000010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
1001010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
1002416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
10038c37ef55SBarry Smith     sum = tmp[i];
1004003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1005010a20caSHong Zhang     x[*c--] = tmp[i] = sum*aa[a->diag[i]];
10068c37ef55SBarry Smith   }
10078c37ef55SBarry Smith 
10085f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isrow,&rout));
10095f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(iscol,&cout));
10105f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayRead(bb,&b));
10115f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayWrite(xx,&x));
10125f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(2.0*a->nz - A->cmap->n));
10133a40ed3dSBarry Smith   PetscFunctionReturn(0);
10148c37ef55SBarry Smith }
1015026e39d0SSatish Balay 
1016ad04f41aSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ_inplace(Mat A,Mat B,Mat X)
10172bebee5dSHong Zhang {
10182bebee5dSHong Zhang   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
10192bebee5dSHong Zhang   IS                iscol = a->col,isrow = a->row;
10205d0c19d7SBarry Smith   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
1021910cf402Sprj-   PetscInt          nz,neq,ldb,ldx;
10225d0c19d7SBarry Smith   const PetscInt    *rout,*cout,*r,*c;
1023910cf402Sprj-   PetscScalar       *x,*tmp = a->solve_work,*tmps,sum;
1024910cf402Sprj-   const PetscScalar *b,*aa = a->a,*v;
1025910cf402Sprj-   PetscBool         isdense;
10262bebee5dSHong Zhang 
10272bebee5dSHong Zhang   PetscFunctionBegin;
10282bebee5dSHong Zhang   if (!n) PetscFunctionReturn(0);
10295f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectTypeCompare((PetscObject)B,MATSEQDENSE,&isdense));
1030*28b400f6SJacob Faibussowitsch   PetscCheck(isdense,PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix");
1031f9fb9879SHong Zhang   if (X != B) {
10325f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscObjectTypeCompare((PetscObject)X,MATSEQDENSE,&isdense));
1033*28b400f6SJacob Faibussowitsch     PetscCheck(isdense,PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix");
1034f9fb9879SHong Zhang   }
10355f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDenseGetArrayRead(B,&b));
10365f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDenseGetLDA(B,&ldb));
10375f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDenseGetArray(X,&x));
10385f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDenseGetLDA(X,&ldx));
10395f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isrow,&rout)); r = rout;
10405f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(iscol,&cout)); c = cout;
1041a36b22ccSSatish Balay   for (neq=0; neq<B->cmap->n; neq++) {
10422bebee5dSHong Zhang     /* forward solve the lower triangular */
10432bebee5dSHong Zhang     tmp[0] = b[r[0]];
10442bebee5dSHong Zhang     tmps   = tmp;
10452bebee5dSHong Zhang     for (i=1; i<n; i++) {
10462bebee5dSHong Zhang       v   = aa + ai[i];
10472bebee5dSHong Zhang       vi  = aj + ai[i];
10482bebee5dSHong Zhang       nz  = a->diag[i] - ai[i];
10492bebee5dSHong Zhang       sum = b[r[i]];
1050003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10512bebee5dSHong Zhang       tmp[i] = sum;
10522bebee5dSHong Zhang     }
10532bebee5dSHong Zhang     /* backward solve the upper triangular */
10542bebee5dSHong Zhang     for (i=n-1; i>=0; i--) {
10552bebee5dSHong Zhang       v   = aa + a->diag[i] + 1;
10562bebee5dSHong Zhang       vi  = aj + a->diag[i] + 1;
10572bebee5dSHong Zhang       nz  = ai[i+1] - a->diag[i] - 1;
10582bebee5dSHong Zhang       sum = tmp[i];
1059003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10602bebee5dSHong Zhang       x[c[i]] = tmp[i] = sum*aa[a->diag[i]];
10612bebee5dSHong Zhang     }
1062910cf402Sprj-     b += ldb;
1063910cf402Sprj-     x += ldx;
10642bebee5dSHong Zhang   }
10655f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isrow,&rout));
10665f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(iscol,&cout));
10675f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDenseRestoreArrayRead(B,&b));
10685f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDenseRestoreArray(X,&x));
10695f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(B->cmap->n*(2.0*a->nz - n)));
10702bebee5dSHong Zhang   PetscFunctionReturn(0);
10712bebee5dSHong Zhang }
10722bebee5dSHong Zhang 
107335233d99SShri Abhyankar PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X)
10749bd0847aSShri Abhyankar {
10759bd0847aSShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
10769bd0847aSShri Abhyankar   IS                iscol = a->col,isrow = a->row;
10779bd0847aSShri Abhyankar   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag;
1078910cf402Sprj-   PetscInt          nz,neq,ldb,ldx;
10799bd0847aSShri Abhyankar   const PetscInt    *rout,*cout,*r,*c;
1080910cf402Sprj-   PetscScalar       *x,*tmp = a->solve_work,sum;
1081910cf402Sprj-   const PetscScalar *b,*aa = a->a,*v;
1082910cf402Sprj-   PetscBool         isdense;
10839bd0847aSShri Abhyankar 
10849bd0847aSShri Abhyankar   PetscFunctionBegin;
10859bd0847aSShri Abhyankar   if (!n) PetscFunctionReturn(0);
10865f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectTypeCompare((PetscObject)B,MATSEQDENSE,&isdense));
1087*28b400f6SJacob Faibussowitsch   PetscCheck(isdense,PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix");
1088f9fb9879SHong Zhang   if (X != B) {
10895f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscObjectTypeCompare((PetscObject)X,MATSEQDENSE,&isdense));
1090*28b400f6SJacob Faibussowitsch     PetscCheck(isdense,PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix");
1091f9fb9879SHong Zhang   }
10925f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDenseGetArrayRead(B,&b));
10935f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDenseGetLDA(B,&ldb));
10945f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDenseGetArray(X,&x));
10955f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDenseGetLDA(X,&ldx));
10965f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isrow,&rout)); r = rout;
10975f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(iscol,&cout)); c = cout;
10989bd0847aSShri Abhyankar   for (neq=0; neq<B->cmap->n; neq++) {
10999bd0847aSShri Abhyankar     /* forward solve the lower triangular */
11009bd0847aSShri Abhyankar     tmp[0] = b[r[0]];
11019bd0847aSShri Abhyankar     v      = aa;
11029bd0847aSShri Abhyankar     vi     = aj;
11039bd0847aSShri Abhyankar     for (i=1; i<n; i++) {
11049bd0847aSShri Abhyankar       nz  = ai[i+1] - ai[i];
11059bd0847aSShri Abhyankar       sum = b[r[i]];
11069bd0847aSShri Abhyankar       PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
11079bd0847aSShri Abhyankar       tmp[i] = sum;
11089bd0847aSShri Abhyankar       v     += nz; vi += nz;
11099bd0847aSShri Abhyankar     }
11109bd0847aSShri Abhyankar     /* backward solve the upper triangular */
11119bd0847aSShri Abhyankar     for (i=n-1; i>=0; i--) {
11129bd0847aSShri Abhyankar       v   = aa + adiag[i+1]+1;
11139bd0847aSShri Abhyankar       vi  = aj + adiag[i+1]+1;
11149bd0847aSShri Abhyankar       nz  = adiag[i]-adiag[i+1]-1;
11159bd0847aSShri Abhyankar       sum = tmp[i];
11169bd0847aSShri Abhyankar       PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
11179bd0847aSShri Abhyankar       x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */
11189bd0847aSShri Abhyankar     }
1119910cf402Sprj-     b += ldb;
1120910cf402Sprj-     x += ldx;
11219bd0847aSShri Abhyankar   }
11225f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isrow,&rout));
11235f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(iscol,&cout));
11245f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDenseRestoreArrayRead(B,&b));
11255f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDenseRestoreArray(X,&x));
11265f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(B->cmap->n*(2.0*a->nz - n)));
11279bd0847aSShri Abhyankar   PetscFunctionReturn(0);
11289bd0847aSShri Abhyankar }
11299bd0847aSShri Abhyankar 
1130137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx)
1131137fb511SHong Zhang {
1132137fb511SHong Zhang   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
1133137fb511SHong Zhang   IS                iscol = a->col,isrow = a->row;
11345d0c19d7SBarry Smith   const PetscInt    *r,*c,*rout,*cout;
11355d0c19d7SBarry Smith   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
11365d0c19d7SBarry Smith   PetscInt          nz,row;
1137fdc842d1SBarry Smith   PetscScalar       *x,*tmp,*tmps,sum;
1138fdc842d1SBarry Smith   const PetscScalar *b;
1139dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
1140137fb511SHong Zhang 
1141137fb511SHong Zhang   PetscFunctionBegin;
1142137fb511SHong Zhang   if (!n) PetscFunctionReturn(0);
1143137fb511SHong Zhang 
11445f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayRead(bb,&b));
11455f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayWrite(xx,&x));
1146137fb511SHong Zhang   tmp  = a->solve_work;
1147137fb511SHong Zhang 
11485f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isrow,&rout)); r = rout;
11495f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(iscol,&cout)); c = cout + (n-1);
1150137fb511SHong Zhang 
1151137fb511SHong Zhang   /* forward solve the lower triangular */
1152137fb511SHong Zhang   tmp[0] = b[*r++];
1153137fb511SHong Zhang   tmps   = tmp;
1154137fb511SHong Zhang   for (row=1; row<n; row++) {
1155137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1156137fb511SHong Zhang     v   = aa + ai[i];
1157137fb511SHong Zhang     vi  = aj + ai[i];
1158137fb511SHong Zhang     nz  = a->diag[i] - ai[i];
1159137fb511SHong Zhang     sum = b[*r++];
1160003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1161137fb511SHong Zhang     tmp[row] = sum;
1162137fb511SHong Zhang   }
1163137fb511SHong Zhang 
1164137fb511SHong Zhang   /* backward solve the upper triangular */
1165137fb511SHong Zhang   for (row=n-1; row>=0; row--) {
1166137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1167137fb511SHong Zhang     v   = aa + a->diag[i] + 1;
1168137fb511SHong Zhang     vi  = aj + a->diag[i] + 1;
1169137fb511SHong Zhang     nz  = ai[i+1] - a->diag[i] - 1;
1170137fb511SHong Zhang     sum = tmp[row];
1171003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1172137fb511SHong Zhang     x[*c--] = tmp[row] = sum*aa[a->diag[i]];
1173137fb511SHong Zhang   }
1174137fb511SHong Zhang 
11755f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isrow,&rout));
11765f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(iscol,&cout));
11775f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayRead(bb,&b));
11785f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayWrite(xx,&x));
11795f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(2.0*a->nz - A->cmap->n));
1180137fb511SHong Zhang   PetscFunctionReturn(0);
1181137fb511SHong Zhang }
1182137fb511SHong Zhang 
1183930ae53cSBarry Smith /* ----------------------------------------------------------- */
1184c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/ftn-kernels/fsolve.h>
1185ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_inplace(Mat A,Vec bb,Vec xx)
1186930ae53cSBarry Smith {
1187930ae53cSBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
1188d0f46423SBarry Smith   PetscInt          n   = A->rmap->n;
1189b377110cSBarry Smith   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag;
1190356650c2SBarry Smith   PetscScalar       *x;
1191356650c2SBarry Smith   const PetscScalar *b;
1192dd6ea824SBarry Smith   const MatScalar   *aa = a->a;
1193aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
1194356650c2SBarry Smith   PetscInt        adiag_i,i,nz,ai_i;
1195b377110cSBarry Smith   const PetscInt  *vi;
1196dd6ea824SBarry Smith   const MatScalar *v;
1197dd6ea824SBarry Smith   PetscScalar     sum;
1198d85afc42SSatish Balay #endif
1199930ae53cSBarry Smith 
12003a40ed3dSBarry Smith   PetscFunctionBegin;
12013a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
1202930ae53cSBarry Smith 
12035f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayRead(bb,&b));
12045f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayWrite(xx,&x));
1205930ae53cSBarry Smith 
1206aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
12071c4feecaSSatish Balay   fortransolveaij_(&n,x,ai,aj,adiag,aa,b);
12081c4feecaSSatish Balay #else
1209930ae53cSBarry Smith   /* forward solve the lower triangular */
1210930ae53cSBarry Smith   x[0] = b[0];
1211930ae53cSBarry Smith   for (i=1; i<n; i++) {
1212930ae53cSBarry Smith     ai_i = ai[i];
1213930ae53cSBarry Smith     v    = aa + ai_i;
1214930ae53cSBarry Smith     vi   = aj + ai_i;
1215930ae53cSBarry Smith     nz   = adiag[i] - ai_i;
1216930ae53cSBarry Smith     sum  = b[i];
1217003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1218930ae53cSBarry Smith     x[i] = sum;
1219930ae53cSBarry Smith   }
1220930ae53cSBarry Smith 
1221930ae53cSBarry Smith   /* backward solve the upper triangular */
1222930ae53cSBarry Smith   for (i=n-1; i>=0; i--) {
1223930ae53cSBarry Smith     adiag_i = adiag[i];
1224d708749eSBarry Smith     v       = aa + adiag_i + 1;
1225d708749eSBarry Smith     vi      = aj + adiag_i + 1;
1226930ae53cSBarry Smith     nz      = ai[i+1] - adiag_i - 1;
1227930ae53cSBarry Smith     sum     = x[i];
1228003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1229930ae53cSBarry Smith     x[i] = sum*aa[adiag_i];
1230930ae53cSBarry Smith   }
12311c4feecaSSatish Balay #endif
12325f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(2.0*a->nz - A->cmap->n));
12335f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayRead(bb,&b));
12345f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayWrite(xx,&x));
12353a40ed3dSBarry Smith   PetscFunctionReturn(0);
1236930ae53cSBarry Smith }
1237930ae53cSBarry Smith 
1238ad04f41aSHong Zhang PetscErrorCode MatSolveAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec yy,Vec xx)
1239da3a660dSBarry Smith {
1240416022c9SBarry Smith   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
1241416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
124204fbf559SBarry Smith   PetscInt          i, n = A->rmap->n,j;
12435d0c19d7SBarry Smith   PetscInt          nz;
124404fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j;
124504fbf559SBarry Smith   PetscScalar       *x,*tmp,sum;
124604fbf559SBarry Smith   const PetscScalar *b;
1247dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
1248da3a660dSBarry Smith 
12493a40ed3dSBarry Smith   PetscFunctionBegin;
12505f80ce2aSJacob Faibussowitsch   if (yy != xx) CHKERRQ(VecCopy(yy,xx));
1251da3a660dSBarry Smith 
12525f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayRead(bb,&b));
12535f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArray(xx,&x));
1254416022c9SBarry Smith   tmp  = a->solve_work;
1255da3a660dSBarry Smith 
12565f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isrow,&rout)); r = rout;
12575f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(iscol,&cout)); c = cout + (n-1);
1258da3a660dSBarry Smith 
1259da3a660dSBarry Smith   /* forward solve the lower triangular */
1260da3a660dSBarry Smith   tmp[0] = b[*r++];
1261da3a660dSBarry Smith   for (i=1; i<n; i++) {
1262010a20caSHong Zhang     v   = aa + ai[i];
1263010a20caSHong Zhang     vi  = aj + ai[i];
1264416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
1265da3a660dSBarry Smith     sum = b[*r++];
126604fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1267da3a660dSBarry Smith     tmp[i] = sum;
1268da3a660dSBarry Smith   }
1269da3a660dSBarry Smith 
1270da3a660dSBarry Smith   /* backward solve the upper triangular */
1271da3a660dSBarry Smith   for (i=n-1; i>=0; i--) {
1272010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
1273010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
1274416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
1275da3a660dSBarry Smith     sum = tmp[i];
127604fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1277010a20caSHong Zhang     tmp[i]   = sum*aa[a->diag[i]];
1278da3a660dSBarry Smith     x[*c--] += tmp[i];
1279da3a660dSBarry Smith   }
1280da3a660dSBarry Smith 
12815f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isrow,&rout));
12825f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(iscol,&cout));
12835f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayRead(bb,&b));
12845f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArray(xx,&x));
12855f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(2.0*a->nz));
12863a40ed3dSBarry Smith   PetscFunctionReturn(0);
1287da3a660dSBarry Smith }
128804fbf559SBarry Smith 
128935233d99SShri Abhyankar PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx)
12903c0119dfSShri Abhyankar {
12913c0119dfSShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
12923c0119dfSShri Abhyankar   IS                iscol = a->col,isrow = a->row;
12933c0119dfSShri Abhyankar   PetscInt          i, n = A->rmap->n,j;
12943c0119dfSShri Abhyankar   PetscInt          nz;
12953c0119dfSShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag;
12963c0119dfSShri Abhyankar   PetscScalar       *x,*tmp,sum;
12973c0119dfSShri Abhyankar   const PetscScalar *b;
12983c0119dfSShri Abhyankar   const MatScalar   *aa = a->a,*v;
12993c0119dfSShri Abhyankar 
13003c0119dfSShri Abhyankar   PetscFunctionBegin;
13015f80ce2aSJacob Faibussowitsch   if (yy != xx) CHKERRQ(VecCopy(yy,xx));
13023c0119dfSShri Abhyankar 
13035f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayRead(bb,&b));
13045f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArray(xx,&x));
13053c0119dfSShri Abhyankar   tmp  = a->solve_work;
13063c0119dfSShri Abhyankar 
13075f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isrow,&rout)); r = rout;
13085f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(iscol,&cout)); c = cout;
13093c0119dfSShri Abhyankar 
13103c0119dfSShri Abhyankar   /* forward solve the lower triangular */
13113c0119dfSShri Abhyankar   tmp[0] = b[r[0]];
13123c0119dfSShri Abhyankar   v      = aa;
13133c0119dfSShri Abhyankar   vi     = aj;
13143c0119dfSShri Abhyankar   for (i=1; i<n; i++) {
13153c0119dfSShri Abhyankar     nz  = ai[i+1] - ai[i];
13163c0119dfSShri Abhyankar     sum = b[r[i]];
13173c0119dfSShri Abhyankar     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
13183c0119dfSShri Abhyankar     tmp[i] = sum;
13192205254eSKarl Rupp     v     += nz;
13202205254eSKarl Rupp     vi    += nz;
13213c0119dfSShri Abhyankar   }
13223c0119dfSShri Abhyankar 
13233c0119dfSShri Abhyankar   /* backward solve the upper triangular */
13243c0119dfSShri Abhyankar   v  = aa + adiag[n-1];
13253c0119dfSShri Abhyankar   vi = aj + adiag[n-1];
13263c0119dfSShri Abhyankar   for (i=n-1; i>=0; i--) {
13273c0119dfSShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
13283c0119dfSShri Abhyankar     sum = tmp[i];
13293c0119dfSShri Abhyankar     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
13303c0119dfSShri Abhyankar     tmp[i]   = sum*v[nz];
13313c0119dfSShri Abhyankar     x[c[i]] += tmp[i];
13323c0119dfSShri Abhyankar     v       += nz+1; vi += nz+1;
13333c0119dfSShri Abhyankar   }
13343c0119dfSShri Abhyankar 
13355f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isrow,&rout));
13365f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(iscol,&cout));
13375f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayRead(bb,&b));
13385f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArray(xx,&x));
13395f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(2.0*a->nz));
13403c0119dfSShri Abhyankar   PetscFunctionReturn(0);
13413c0119dfSShri Abhyankar }
13423c0119dfSShri Abhyankar 
1343ad04f41aSHong Zhang PetscErrorCode MatSolveTranspose_SeqAIJ_inplace(Mat A,Vec bb,Vec xx)
1344da3a660dSBarry Smith {
1345416022c9SBarry Smith   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
13462235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
134704fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
134804fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
134904fbf559SBarry Smith   PetscInt          nz;
135004fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1351dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
135204fbf559SBarry Smith   const PetscScalar *b;
1353da3a660dSBarry Smith 
13543a40ed3dSBarry Smith   PetscFunctionBegin;
13555f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayRead(bb,&b));
13565f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayWrite(xx,&x));
1357416022c9SBarry Smith   tmp  = a->solve_work;
1358da3a660dSBarry Smith 
13595f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isrow,&rout)); r = rout;
13605f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(iscol,&cout)); c = cout;
1361da3a660dSBarry Smith 
1362da3a660dSBarry Smith   /* copy the b into temp work space according to permutation */
13632235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1364da3a660dSBarry Smith 
1365da3a660dSBarry Smith   /* forward solve the U^T */
1366da3a660dSBarry Smith   for (i=0; i<n; i++) {
1367010a20caSHong Zhang     v   = aa + diag[i];
1368010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1369f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
1370c38d4ed2SBarry Smith     s1  = tmp[i];
1371ef66eb69SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
137204fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1373c38d4ed2SBarry Smith     tmp[i] = s1;
1374da3a660dSBarry Smith   }
1375da3a660dSBarry Smith 
1376da3a660dSBarry Smith   /* backward solve the L^T */
1377da3a660dSBarry Smith   for (i=n-1; i>=0; i--) {
1378010a20caSHong Zhang     v  = aa + diag[i] - 1;
1379010a20caSHong Zhang     vi = aj + diag[i] - 1;
1380f1af5d2fSBarry Smith     nz = diag[i] - ai[i];
1381f1af5d2fSBarry Smith     s1 = tmp[i];
138204fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
1383da3a660dSBarry Smith   }
1384da3a660dSBarry Smith 
1385da3a660dSBarry Smith   /* copy tmp into x according to permutation */
1386591294e4SBarry Smith   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1387da3a660dSBarry Smith 
13885f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isrow,&rout));
13895f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(iscol,&cout));
13905f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayRead(bb,&b));
13915f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayWrite(xx,&x));
1392da3a660dSBarry Smith 
13935f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(2.0*a->nz-A->cmap->n));
13943a40ed3dSBarry Smith   PetscFunctionReturn(0);
1395da3a660dSBarry Smith }
1396da3a660dSBarry Smith 
139735233d99SShri Abhyankar PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx)
1398d1fa9404SShri Abhyankar {
1399d1fa9404SShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
1400d1fa9404SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
1401d1fa9404SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi;
1402d1fa9404SShri Abhyankar   PetscInt          i,n = A->rmap->n,j;
1403d1fa9404SShri Abhyankar   PetscInt          nz;
1404d1fa9404SShri Abhyankar   PetscScalar       *x,*tmp,s1;
1405d1fa9404SShri Abhyankar   const MatScalar   *aa = a->a,*v;
1406d1fa9404SShri Abhyankar   const PetscScalar *b;
1407d1fa9404SShri Abhyankar 
1408d1fa9404SShri Abhyankar   PetscFunctionBegin;
14095f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayRead(bb,&b));
14105f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayWrite(xx,&x));
1411d1fa9404SShri Abhyankar   tmp  = a->solve_work;
1412d1fa9404SShri Abhyankar 
14135f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isrow,&rout)); r = rout;
14145f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(iscol,&cout)); c = cout;
1415d1fa9404SShri Abhyankar 
1416d1fa9404SShri Abhyankar   /* copy the b into temp work space according to permutation */
1417d1fa9404SShri Abhyankar   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1418d1fa9404SShri Abhyankar 
1419d1fa9404SShri Abhyankar   /* forward solve the U^T */
1420d1fa9404SShri Abhyankar   for (i=0; i<n; i++) {
1421d1fa9404SShri Abhyankar     v   = aa + adiag[i+1] + 1;
1422d1fa9404SShri Abhyankar     vi  = aj + adiag[i+1] + 1;
1423d1fa9404SShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
1424d1fa9404SShri Abhyankar     s1  = tmp[i];
1425d1fa9404SShri Abhyankar     s1 *= v[nz];  /* multiply by inverse of diagonal entry */
1426d1fa9404SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1427d1fa9404SShri Abhyankar     tmp[i] = s1;
1428d1fa9404SShri Abhyankar   }
1429d1fa9404SShri Abhyankar 
1430d1fa9404SShri Abhyankar   /* backward solve the L^T */
1431d1fa9404SShri Abhyankar   for (i=n-1; i>=0; i--) {
1432d1fa9404SShri Abhyankar     v  = aa + ai[i];
1433d1fa9404SShri Abhyankar     vi = aj + ai[i];
1434d1fa9404SShri Abhyankar     nz = ai[i+1] - ai[i];
1435d1fa9404SShri Abhyankar     s1 = tmp[i];
1436d1fa9404SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1437d1fa9404SShri Abhyankar   }
1438d1fa9404SShri Abhyankar 
1439d1fa9404SShri Abhyankar   /* copy tmp into x according to permutation */
1440d1fa9404SShri Abhyankar   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1441d1fa9404SShri Abhyankar 
14425f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isrow,&rout));
14435f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(iscol,&cout));
14445f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayRead(bb,&b));
14455f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayWrite(xx,&x));
1446d1fa9404SShri Abhyankar 
14475f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(2.0*a->nz-A->cmap->n));
1448d1fa9404SShri Abhyankar   PetscFunctionReturn(0);
1449d1fa9404SShri Abhyankar }
1450d1fa9404SShri Abhyankar 
1451ad04f41aSHong Zhang PetscErrorCode MatSolveTransposeAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec zz,Vec xx)
1452da3a660dSBarry Smith {
1453416022c9SBarry Smith   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
14542235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
145504fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
145604fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
145704fbf559SBarry Smith   PetscInt          nz;
145804fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1459dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
146004fbf559SBarry Smith   const PetscScalar *b;
14616abc6512SBarry Smith 
14623a40ed3dSBarry Smith   PetscFunctionBegin;
14635f80ce2aSJacob Faibussowitsch   if (zz != xx) CHKERRQ(VecCopy(zz,xx));
14645f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayRead(bb,&b));
14655f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArray(xx,&x));
1466416022c9SBarry Smith   tmp  = a->solve_work;
14676abc6512SBarry Smith 
14685f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isrow,&rout)); r = rout;
14695f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(iscol,&cout)); c = cout;
14706abc6512SBarry Smith 
14716abc6512SBarry Smith   /* copy the b into temp work space according to permutation */
14722235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
14736abc6512SBarry Smith 
14746abc6512SBarry Smith   /* forward solve the U^T */
14756abc6512SBarry Smith   for (i=0; i<n; i++) {
1476010a20caSHong Zhang     v   = aa + diag[i];
1477010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1478f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
147904fbf559SBarry Smith     s1  = tmp[i];
148004fbf559SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
148104fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
148204fbf559SBarry Smith     tmp[i] = s1;
14836abc6512SBarry Smith   }
14846abc6512SBarry Smith 
14856abc6512SBarry Smith   /* backward solve the L^T */
14866abc6512SBarry Smith   for (i=n-1; i>=0; i--) {
1487010a20caSHong Zhang     v  = aa + diag[i] - 1;
1488010a20caSHong Zhang     vi = aj + diag[i] - 1;
1489f1af5d2fSBarry Smith     nz = diag[i] - ai[i];
149004fbf559SBarry Smith     s1 = tmp[i];
149104fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
14926abc6512SBarry Smith   }
14936abc6512SBarry Smith 
14946abc6512SBarry Smith   /* copy tmp into x according to permutation */
14956abc6512SBarry Smith   for (i=0; i<n; i++) x[r[i]] += tmp[i];
14966abc6512SBarry Smith 
14975f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isrow,&rout));
14985f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(iscol,&cout));
14995f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayRead(bb,&b));
15005f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArray(xx,&x));
15016abc6512SBarry Smith 
15025f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(2.0*a->nz-A->cmap->n));
15033a40ed3dSBarry Smith   PetscFunctionReturn(0);
1504da3a660dSBarry Smith }
150504fbf559SBarry Smith 
150635233d99SShri Abhyankar PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx)
1507533e6011SShri Abhyankar {
1508533e6011SShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
1509533e6011SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
1510533e6011SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi;
1511533e6011SShri Abhyankar   PetscInt          i,n = A->rmap->n,j;
1512533e6011SShri Abhyankar   PetscInt          nz;
1513533e6011SShri Abhyankar   PetscScalar       *x,*tmp,s1;
1514533e6011SShri Abhyankar   const MatScalar   *aa = a->a,*v;
1515533e6011SShri Abhyankar   const PetscScalar *b;
1516533e6011SShri Abhyankar 
1517533e6011SShri Abhyankar   PetscFunctionBegin;
15185f80ce2aSJacob Faibussowitsch   if (zz != xx) CHKERRQ(VecCopy(zz,xx));
15195f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayRead(bb,&b));
15205f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArray(xx,&x));
1521533e6011SShri Abhyankar   tmp  = a->solve_work;
1522533e6011SShri Abhyankar 
15235f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isrow,&rout)); r = rout;
15245f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(iscol,&cout)); c = cout;
1525533e6011SShri Abhyankar 
1526533e6011SShri Abhyankar   /* copy the b into temp work space according to permutation */
1527533e6011SShri Abhyankar   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1528533e6011SShri Abhyankar 
1529533e6011SShri Abhyankar   /* forward solve the U^T */
1530533e6011SShri Abhyankar   for (i=0; i<n; i++) {
1531533e6011SShri Abhyankar     v   = aa + adiag[i+1] + 1;
1532533e6011SShri Abhyankar     vi  = aj + adiag[i+1] + 1;
1533533e6011SShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
1534533e6011SShri Abhyankar     s1  = tmp[i];
1535533e6011SShri Abhyankar     s1 *= v[nz];  /* multiply by inverse of diagonal entry */
1536533e6011SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1537533e6011SShri Abhyankar     tmp[i] = s1;
1538533e6011SShri Abhyankar   }
1539533e6011SShri Abhyankar 
1540533e6011SShri Abhyankar   /* backward solve the L^T */
1541533e6011SShri Abhyankar   for (i=n-1; i>=0; i--) {
1542533e6011SShri Abhyankar     v  = aa + ai[i];
1543533e6011SShri Abhyankar     vi = aj + ai[i];
1544533e6011SShri Abhyankar     nz = ai[i+1] - ai[i];
1545533e6011SShri Abhyankar     s1 = tmp[i];
1546c5e3b2a3SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1547533e6011SShri Abhyankar   }
1548533e6011SShri Abhyankar 
1549533e6011SShri Abhyankar   /* copy tmp into x according to permutation */
1550533e6011SShri Abhyankar   for (i=0; i<n; i++) x[r[i]] += tmp[i];
1551533e6011SShri Abhyankar 
15525f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isrow,&rout));
15535f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(iscol,&cout));
15545f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayRead(bb,&b));
15555f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArray(xx,&x));
1556533e6011SShri Abhyankar 
15575f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(2.0*a->nz-A->cmap->n));
1558533e6011SShri Abhyankar   PetscFunctionReturn(0);
1559533e6011SShri Abhyankar }
1560533e6011SShri Abhyankar 
15619e25ed09SBarry Smith /* ----------------------------------------------------------------*/
1562b47f5a65SHong Zhang 
15638fc3a347SHong Zhang /*
15648a76ed4fSHong Zhang    ilu() under revised new data structure.
1565813a1f2bSShri Abhyankar    Factored arrays bj and ba are stored as
1566813a1f2bSShri Abhyankar      L(0,:), L(1,:), ...,L(n-1,:),  U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:)
1567813a1f2bSShri Abhyankar 
1568813a1f2bSShri Abhyankar    bi=fact->i is an array of size n+1, in which
1569813a1f2bSShri Abhyankar    bi+
1570baabb860SHong Zhang      bi[i]:  points to 1st entry of L(i,:),i=0,...,n-1
1571baabb860SHong Zhang      bi[n]:  points to L(n-1,n-1)+1
1572baabb860SHong Zhang 
1573813a1f2bSShri Abhyankar   bdiag=fact->diag is an array of size n+1,in which
1574baabb860SHong Zhang      bdiag[i]: points to diagonal of U(i,:), i=0,...,n-1
1575a24f213cSHong Zhang      bdiag[n]: points to entry of U(n-1,0)-1
1576813a1f2bSShri Abhyankar 
1577813a1f2bSShri Abhyankar    U(i,:) contains bdiag[i] as its last entry, i.e.,
1578813a1f2bSShri Abhyankar     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
1579813a1f2bSShri Abhyankar */
158035233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
1581813a1f2bSShri Abhyankar {
1582813a1f2bSShri Abhyankar   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data,*b;
1583bbd65245SShri Abhyankar   const PetscInt n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag;
1584bbd65245SShri Abhyankar   PetscInt       i,j,k=0,nz,*bi,*bj,*bdiag;
15851df811f5SHong Zhang   IS             isicol;
1586813a1f2bSShri Abhyankar 
1587813a1f2bSShri Abhyankar   PetscFunctionBegin;
15885f80ce2aSJacob Faibussowitsch   CHKERRQ(ISInvertPermutation(iscol,PETSC_DECIDE,&isicol));
15895f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE));
1590813a1f2bSShri Abhyankar   b    = (Mat_SeqAIJ*)(fact)->data;
1591813a1f2bSShri Abhyankar 
1592813a1f2bSShri Abhyankar   /* allocate matrix arrays for new data structure */
15935f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc3(ai[n]+1,&b->a,ai[n]+1,&b->j,n+1,&b->i));
15945f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogObjectMemory((PetscObject)fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(n+1)*sizeof(PetscInt)));
15952205254eSKarl Rupp 
1596813a1f2bSShri Abhyankar   b->singlemalloc = PETSC_TRUE;
1597813a1f2bSShri Abhyankar   if (!b->diag) {
15985f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscMalloc1(n+1,&b->diag));
15995f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscLogObjectMemory((PetscObject)fact,(n+1)*sizeof(PetscInt)));
1600813a1f2bSShri Abhyankar   }
1601813a1f2bSShri Abhyankar   bdiag = b->diag;
1602813a1f2bSShri Abhyankar 
1603813a1f2bSShri Abhyankar   if (n > 0) {
16045f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscArrayzero(b->a,ai[n]));
1605813a1f2bSShri Abhyankar   }
1606813a1f2bSShri Abhyankar 
1607813a1f2bSShri Abhyankar   /* set bi and bj with new data structure */
1608813a1f2bSShri Abhyankar   bi = b->i;
1609813a1f2bSShri Abhyankar   bj = b->j;
1610813a1f2bSShri Abhyankar 
1611813a1f2bSShri Abhyankar   /* L part */
1612e60cf9a0SBarry Smith   bi[0] = 0;
1613813a1f2bSShri Abhyankar   for (i=0; i<n; i++) {
1614813a1f2bSShri Abhyankar     nz      = adiag[i] - ai[i];
1615813a1f2bSShri Abhyankar     bi[i+1] = bi[i] + nz;
1616813a1f2bSShri Abhyankar     aj      = a->j + ai[i];
1617813a1f2bSShri Abhyankar     for (j=0; j<nz; j++) {
1618bbd65245SShri Abhyankar       /*   *bj = aj[j]; bj++; */
1619bbd65245SShri Abhyankar       bj[k++] = aj[j];
1620813a1f2bSShri Abhyankar     }
1621813a1f2bSShri Abhyankar   }
1622813a1f2bSShri Abhyankar 
1623813a1f2bSShri Abhyankar   /* U part */
162462fbd6afSShri Abhyankar   bdiag[n] = bi[n]-1;
1625813a1f2bSShri Abhyankar   for (i=n-1; i>=0; i--) {
1626813a1f2bSShri Abhyankar     nz = ai[i+1] - adiag[i] - 1;
1627813a1f2bSShri Abhyankar     aj = a->j + adiag[i] + 1;
1628813a1f2bSShri Abhyankar     for (j=0; j<nz; j++) {
1629bbd65245SShri Abhyankar       /*      *bj = aj[j]; bj++; */
1630bbd65245SShri Abhyankar       bj[k++] = aj[j];
1631813a1f2bSShri Abhyankar     }
1632813a1f2bSShri Abhyankar     /* diag[i] */
1633bbd65245SShri Abhyankar     /*    *bj = i; bj++; */
1634bbd65245SShri Abhyankar     bj[k++]  = i;
1635a24f213cSHong Zhang     bdiag[i] = bdiag[i+1] + nz + 1;
1636813a1f2bSShri Abhyankar   }
16371df811f5SHong Zhang 
1638d5f3da31SBarry Smith   fact->factortype             = MAT_FACTOR_ILU;
16391df811f5SHong Zhang   fact->info.factor_mallocs    = 0;
16401df811f5SHong Zhang   fact->info.fill_ratio_given  = info->fill;
16411df811f5SHong Zhang   fact->info.fill_ratio_needed = 1.0;
164235233d99SShri Abhyankar   fact->ops->lufactornumeric   = MatLUFactorNumeric_SeqAIJ;
16435f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSeqAIJCheckInode_FactorLU(fact));
16441df811f5SHong Zhang 
16451df811f5SHong Zhang   b       = (Mat_SeqAIJ*)(fact)->data;
16461df811f5SHong Zhang   b->row  = isrow;
16471df811f5SHong Zhang   b->col  = iscol;
16481df811f5SHong Zhang   b->icol = isicol;
16495f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(fact->rmap->n+1,&b->solve_work));
16505f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectReference((PetscObject)isrow));
16515f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectReference((PetscObject)iscol));
1652813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
1653813a1f2bSShri Abhyankar }
1654813a1f2bSShri Abhyankar 
165535233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
1656813a1f2bSShri Abhyankar {
1657813a1f2bSShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
1658813a1f2bSShri Abhyankar   IS                 isicol;
1659813a1f2bSShri Abhyankar   const PetscInt     *r,*ic;
16601df811f5SHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j;
1661813a1f2bSShri Abhyankar   PetscInt           *bi,*cols,nnz,*cols_lvl;
1662813a1f2bSShri Abhyankar   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
1663813a1f2bSShri Abhyankar   PetscInt           i,levels,diagonal_fill;
1664035f4963SHong Zhang   PetscBool          col_identity,row_identity,missing;
1665813a1f2bSShri Abhyankar   PetscReal          f;
16660298fd71SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=NULL;
1667813a1f2bSShri Abhyankar   PetscBT            lnkbt;
1668813a1f2bSShri Abhyankar   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
16690298fd71SBarry Smith   PetscFreeSpaceList free_space    =NULL,current_space=NULL;
16700298fd71SBarry Smith   PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL;
1671813a1f2bSShri Abhyankar 
1672813a1f2bSShri Abhyankar   PetscFunctionBegin;
16732c71b3e2SJacob Faibussowitsch   PetscCheckFalse(A->rmap->n != A->cmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %" PetscInt_FMT " columns %" PetscInt_FMT,A->rmap->n,A->cmap->n);
16745f80ce2aSJacob Faibussowitsch   CHKERRQ(MatMissingDiagonal(A,&missing,&i));
1675*28b400f6SJacob Faibussowitsch   PetscCheck(!missing,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %" PetscInt_FMT,i);
1676ad04f41aSHong Zhang 
1677813a1f2bSShri Abhyankar   levels = (PetscInt)info->levels;
16785f80ce2aSJacob Faibussowitsch   CHKERRQ(ISIdentity(isrow,&row_identity));
16795f80ce2aSJacob Faibussowitsch   CHKERRQ(ISIdentity(iscol,&col_identity));
1680813a1f2bSShri Abhyankar   if (!levels && row_identity && col_identity) {
1681813a1f2bSShri Abhyankar     /* special case: ilu(0) with natural ordering */
16825f80ce2aSJacob Faibussowitsch     CHKERRQ(MatILUFactorSymbolic_SeqAIJ_ilu0(fact,A,isrow,iscol,info));
1683124a3351SHong Zhang     if (a->inode.size) {
1684124a3351SHong Zhang       fact->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode;
1685124a3351SHong Zhang     }
1686813a1f2bSShri Abhyankar     PetscFunctionReturn(0);
1687813a1f2bSShri Abhyankar   }
1688813a1f2bSShri Abhyankar 
16895f80ce2aSJacob Faibussowitsch   CHKERRQ(ISInvertPermutation(iscol,PETSC_DECIDE,&isicol));
16905f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isrow,&r));
16915f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isicol,&ic));
1692813a1f2bSShri Abhyankar 
16931bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
16945f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(n+1,&bi));
16955f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(n+1,&bdiag));
1696e60cf9a0SBarry Smith   bi[0] = bdiag[0] = 0;
16975f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc2(n,&bj_ptr,n,&bjlvl_ptr));
1698813a1f2bSShri Abhyankar 
1699813a1f2bSShri Abhyankar   /* create a linked list for storing column indices of the active row */
1700813a1f2bSShri Abhyankar   nlnk = n + 1;
17015f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt));
1702813a1f2bSShri Abhyankar 
1703813a1f2bSShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
17041df811f5SHong Zhang   f                 = info->fill;
17051df811f5SHong Zhang   diagonal_fill     = (PetscInt)info->diagonal_fill;
17065f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space));
1707813a1f2bSShri Abhyankar   current_space     = free_space;
17085f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space_lvl));
1709813a1f2bSShri Abhyankar   current_space_lvl = free_space_lvl;
1710813a1f2bSShri Abhyankar   for (i=0; i<n; i++) {
1711813a1f2bSShri Abhyankar     nzi = 0;
1712813a1f2bSShri Abhyankar     /* copy current row into linked list */
1713813a1f2bSShri Abhyankar     nnz = ai[r[i]+1] - ai[r[i]];
1714*28b400f6SJacob Faibussowitsch     PetscCheck(nnz,PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %" PetscInt_FMT " in permuted ordering %" PetscInt_FMT,r[i],i);
1715813a1f2bSShri Abhyankar     cols   = aj + ai[r[i]];
1716813a1f2bSShri Abhyankar     lnk[i] = -1; /* marker to indicate if diagonal exists */
171794bad497SJacob Faibussowitsch     CHKERRQ(PetscIncompleteLLInit(nnz,cols,n,ic,&nlnk,lnk,lnk_lvl,lnkbt));
1718813a1f2bSShri Abhyankar     nzi   += nlnk;
1719813a1f2bSShri Abhyankar 
1720813a1f2bSShri Abhyankar     /* make sure diagonal entry is included */
1721813a1f2bSShri Abhyankar     if (diagonal_fill && lnk[i] == -1) {
1722813a1f2bSShri Abhyankar       fm = n;
1723813a1f2bSShri Abhyankar       while (lnk[fm] < i) fm = lnk[fm];
1724813a1f2bSShri Abhyankar       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
1725813a1f2bSShri Abhyankar       lnk[fm]    = i;
1726813a1f2bSShri Abhyankar       lnk_lvl[i] = 0;
1727813a1f2bSShri Abhyankar       nzi++; dcount++;
1728813a1f2bSShri Abhyankar     }
1729813a1f2bSShri Abhyankar 
1730813a1f2bSShri Abhyankar     /* add pivot rows into the active row */
1731813a1f2bSShri Abhyankar     nzbd = 0;
1732813a1f2bSShri Abhyankar     prow = lnk[n];
1733813a1f2bSShri Abhyankar     while (prow < i) {
1734813a1f2bSShri Abhyankar       nnz      = bdiag[prow];
1735813a1f2bSShri Abhyankar       cols     = bj_ptr[prow] + nnz + 1;
1736813a1f2bSShri Abhyankar       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
1737813a1f2bSShri Abhyankar       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
173894bad497SJacob Faibussowitsch       CHKERRQ(PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,&nlnk,lnk,lnk_lvl,lnkbt,prow));
1739813a1f2bSShri Abhyankar       nzi     += nlnk;
1740813a1f2bSShri Abhyankar       prow     = lnk[prow];
1741813a1f2bSShri Abhyankar       nzbd++;
1742813a1f2bSShri Abhyankar     }
1743813a1f2bSShri Abhyankar     bdiag[i] = nzbd;
1744813a1f2bSShri Abhyankar     bi[i+1]  = bi[i] + nzi;
1745813a1f2bSShri Abhyankar     /* if free space is not available, make more free space */
1746813a1f2bSShri Abhyankar     if (current_space->local_remaining<nzi) {
1747f91af8c7SBarry Smith       nnz  = PetscIntMultTruncate(2,PetscIntMultTruncate(nzi,n - i)); /* estimated and max additional space needed */
17485f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscFreeSpaceGet(nnz,&current_space));
17495f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscFreeSpaceGet(nnz,&current_space_lvl));
1750813a1f2bSShri Abhyankar       reallocs++;
1751813a1f2bSShri Abhyankar     }
1752813a1f2bSShri Abhyankar 
1753813a1f2bSShri Abhyankar     /* copy data into free_space and free_space_lvl, then initialize lnk */
17545f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt));
1755813a1f2bSShri Abhyankar     bj_ptr[i]    = current_space->array;
1756813a1f2bSShri Abhyankar     bjlvl_ptr[i] = current_space_lvl->array;
1757813a1f2bSShri Abhyankar 
1758813a1f2bSShri Abhyankar     /* make sure the active row i has diagonal entry */
17592c71b3e2SJacob Faibussowitsch     PetscCheckFalse(*(bj_ptr[i]+bdiag[i]) != i,PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Row %" PetscInt_FMT " has missing diagonal in factored matrix\ntry running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i);
1760813a1f2bSShri Abhyankar 
1761813a1f2bSShri Abhyankar     current_space->array               += nzi;
1762813a1f2bSShri Abhyankar     current_space->local_used          += nzi;
1763813a1f2bSShri Abhyankar     current_space->local_remaining     -= nzi;
1764813a1f2bSShri Abhyankar     current_space_lvl->array           += nzi;
1765813a1f2bSShri Abhyankar     current_space_lvl->local_used      += nzi;
1766813a1f2bSShri Abhyankar     current_space_lvl->local_remaining -= nzi;
1767813a1f2bSShri Abhyankar   }
1768813a1f2bSShri Abhyankar 
17695f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isrow,&r));
17705f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isicol,&ic));
1771813a1f2bSShri Abhyankar   /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */
17725f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(bi[n]+1,&bj));
17735f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag));
1774813a1f2bSShri Abhyankar 
17755f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscIncompleteLLDestroy(lnk,lnkbt));
17765f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFreeSpaceDestroy(free_space_lvl));
17775f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree2(bj_ptr,bjlvl_ptr));
1778813a1f2bSShri Abhyankar 
1779813a1f2bSShri Abhyankar #if defined(PETSC_USE_INFO)
1780813a1f2bSShri Abhyankar   {
1781aef85c9fSShri Abhyankar     PetscReal af = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
17825f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"Reallocs %" PetscInt_FMT " Fill ratio:given %g needed %g\n",reallocs,(double)f,(double)af));
17835f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"Run with -[sub_]pc_factor_fill %g or use \n",(double)af));
17845f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"PCFactorSetFill([sub]pc,%g);\n",(double)af));
17855f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"for best performance.\n"));
1786813a1f2bSShri Abhyankar     if (diagonal_fill) {
17875f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscInfo(A,"Detected and replaced %" PetscInt_FMT " missing diagonals\n",dcount));
1788813a1f2bSShri Abhyankar     }
1789813a1f2bSShri Abhyankar   }
1790813a1f2bSShri Abhyankar #endif
1791813a1f2bSShri Abhyankar   /* put together the new matrix */
17925f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,NULL));
17935f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogObjectParent((PetscObject)fact,(PetscObject)isicol));
1794813a1f2bSShri Abhyankar   b    = (Mat_SeqAIJ*)(fact)->data;
17952205254eSKarl Rupp 
1796813a1f2bSShri Abhyankar   b->free_a       = PETSC_TRUE;
1797813a1f2bSShri Abhyankar   b->free_ij      = PETSC_TRUE;
1798813a1f2bSShri Abhyankar   b->singlemalloc = PETSC_FALSE;
17992205254eSKarl Rupp 
18005f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(bdiag[0]+1,&b->a));
18012205254eSKarl Rupp 
1802813a1f2bSShri Abhyankar   b->j    = bj;
1803813a1f2bSShri Abhyankar   b->i    = bi;
1804813a1f2bSShri Abhyankar   b->diag = bdiag;
1805f4259b30SLisandro Dalcin   b->ilen = NULL;
1806f4259b30SLisandro Dalcin   b->imax = NULL;
1807813a1f2bSShri Abhyankar   b->row  = isrow;
1808813a1f2bSShri Abhyankar   b->col  = iscol;
18095f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectReference((PetscObject)isrow));
18105f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectReference((PetscObject)iscol));
1811813a1f2bSShri Abhyankar   b->icol = isicol;
18122205254eSKarl Rupp 
18135f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(n+1,&b->solve_work));
1814813a1f2bSShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.
1815813a1f2bSShri Abhyankar      Allocate bdiag, solve_work, new a, new j */
18165f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogObjectMemory((PetscObject)fact,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar))));
1817f268cba8SShri Abhyankar   b->maxnz = b->nz = bdiag[0]+1;
18182205254eSKarl Rupp 
1819813a1f2bSShri Abhyankar   (fact)->info.factor_mallocs    = reallocs;
1820813a1f2bSShri Abhyankar   (fact)->info.fill_ratio_given  = f;
1821f268cba8SShri Abhyankar   (fact)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
182235233d99SShri Abhyankar   (fact)->ops->lufactornumeric   = MatLUFactorNumeric_SeqAIJ;
1823f2344125SBarry Smith   if (a->inode.size) {
1824f2344125SBarry Smith     (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode;
1825f2344125SBarry Smith   }
18265f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSeqAIJCheckInode_FactorLU(fact));
1827813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
1828813a1f2bSShri Abhyankar }
1829813a1f2bSShri Abhyankar 
1830ad04f41aSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
18316516239fSHong Zhang {
18326516239fSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
18336516239fSHong Zhang   IS                 isicol;
18346516239fSHong Zhang   const PetscInt     *r,*ic;
1835ece542f9SHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j;
18366516239fSHong Zhang   PetscInt           *bi,*cols,nnz,*cols_lvl;
18376516239fSHong Zhang   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
18386516239fSHong Zhang   PetscInt           i,levels,diagonal_fill;
1839ace3abfcSBarry Smith   PetscBool          col_identity,row_identity;
18406516239fSHong Zhang   PetscReal          f;
18410298fd71SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=NULL;
18426516239fSHong Zhang   PetscBT            lnkbt;
18436516239fSHong Zhang   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
18440298fd71SBarry Smith   PetscFreeSpaceList free_space    =NULL,current_space=NULL;
18450298fd71SBarry Smith   PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL;
1846ace3abfcSBarry Smith   PetscBool          missing;
18476516239fSHong Zhang 
18486516239fSHong Zhang   PetscFunctionBegin;
18492c71b3e2SJacob Faibussowitsch   PetscCheckFalse(A->rmap->n != A->cmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %" PetscInt_FMT " columns %" PetscInt_FMT,A->rmap->n,A->cmap->n);
18505f80ce2aSJacob Faibussowitsch   CHKERRQ(MatMissingDiagonal(A,&missing,&i));
1851*28b400f6SJacob Faibussowitsch   PetscCheck(!missing,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %" PetscInt_FMT,i);
1852ece542f9SHong Zhang 
18536516239fSHong Zhang   f             = info->fill;
18546516239fSHong Zhang   levels        = (PetscInt)info->levels;
18556516239fSHong Zhang   diagonal_fill = (PetscInt)info->diagonal_fill;
18562205254eSKarl Rupp 
18575f80ce2aSJacob Faibussowitsch   CHKERRQ(ISInvertPermutation(iscol,PETSC_DECIDE,&isicol));
18586516239fSHong Zhang 
18595f80ce2aSJacob Faibussowitsch   CHKERRQ(ISIdentity(isrow,&row_identity));
18605f80ce2aSJacob Faibussowitsch   CHKERRQ(ISIdentity(iscol,&col_identity));
18616516239fSHong Zhang   if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */
18625f80ce2aSJacob Faibussowitsch     CHKERRQ(MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE));
18632205254eSKarl Rupp 
1864ad04f41aSHong Zhang     (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ_inplace;
1865f2344125SBarry Smith     if (a->inode.size) {
1866f2344125SBarry Smith       (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
1867f2344125SBarry Smith     }
1868d5f3da31SBarry Smith     fact->factortype               = MAT_FACTOR_ILU;
1869719d5645SBarry Smith     (fact)->info.factor_mallocs    = 0;
1870719d5645SBarry Smith     (fact)->info.fill_ratio_given  = info->fill;
1871719d5645SBarry Smith     (fact)->info.fill_ratio_needed = 1.0;
18722205254eSKarl Rupp 
1873719d5645SBarry Smith     b    = (Mat_SeqAIJ*)(fact)->data;
187408480c60SBarry Smith     b->row  = isrow;
187508480c60SBarry Smith     b->col  = iscol;
187682bf6240SBarry Smith     b->icol = isicol;
18775f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscMalloc1((fact)->rmap->n+1,&b->solve_work));
18785f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscObjectReference((PetscObject)isrow));
18795f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscObjectReference((PetscObject)iscol));
18803a40ed3dSBarry Smith     PetscFunctionReturn(0);
188108480c60SBarry Smith   }
188208480c60SBarry Smith 
18835f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isrow,&r));
18845f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isicol,&ic));
18859e25ed09SBarry Smith 
18861bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
18875f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(n+1,&bi));
18885f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(n+1,&bdiag));
1889a6df321fSShri Abhyankar   bi[0] = bdiag[0] = 0;
18906cf997b0SBarry Smith 
18915f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc2(n,&bj_ptr,n,&bjlvl_ptr));
18925a8e39fbSHong Zhang 
18935a8e39fbSHong Zhang   /* create a linked list for storing column indices of the active row */
18945a8e39fbSHong Zhang   nlnk = n + 1;
18955f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt));
18965a8e39fbSHong Zhang 
18975a8e39fbSHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
18985f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space));
18995a8e39fbSHong Zhang   current_space     = free_space;
19005f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space_lvl));
19015a8e39fbSHong Zhang   current_space_lvl = free_space_lvl;
19025a8e39fbSHong Zhang 
19035a8e39fbSHong Zhang   for (i=0; i<n; i++) {
19045a8e39fbSHong Zhang     nzi = 0;
19056cf997b0SBarry Smith     /* copy current row into linked list */
19065a8e39fbSHong Zhang     nnz = ai[r[i]+1] - ai[r[i]];
1907*28b400f6SJacob Faibussowitsch     PetscCheck(nnz,PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %" PetscInt_FMT " in permuted ordering %" PetscInt_FMT,r[i],i);
19085a8e39fbSHong Zhang     cols   = aj + ai[r[i]];
19095a8e39fbSHong Zhang     lnk[i] = -1; /* marker to indicate if diagonal exists */
191094bad497SJacob Faibussowitsch     CHKERRQ(PetscIncompleteLLInit(nnz,cols,n,ic,&nlnk,lnk,lnk_lvl,lnkbt));
19115a8e39fbSHong Zhang     nzi   += nlnk;
19126cf997b0SBarry Smith 
19136cf997b0SBarry Smith     /* make sure diagonal entry is included */
19145a8e39fbSHong Zhang     if (diagonal_fill && lnk[i] == -1) {
19156cf997b0SBarry Smith       fm = n;
19165a8e39fbSHong Zhang       while (lnk[fm] < i) fm = lnk[fm];
19175a8e39fbSHong Zhang       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
19185a8e39fbSHong Zhang       lnk[fm]    = i;
19195a8e39fbSHong Zhang       lnk_lvl[i] = 0;
19205a8e39fbSHong Zhang       nzi++; dcount++;
19216cf997b0SBarry Smith     }
19226cf997b0SBarry Smith 
19235a8e39fbSHong Zhang     /* add pivot rows into the active row */
19245a8e39fbSHong Zhang     nzbd = 0;
19255a8e39fbSHong Zhang     prow = lnk[n];
19265a8e39fbSHong Zhang     while (prow < i) {
19275a8e39fbSHong Zhang       nnz      = bdiag[prow];
19285a8e39fbSHong Zhang       cols     = bj_ptr[prow] + nnz + 1;
19295a8e39fbSHong Zhang       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
19305a8e39fbSHong Zhang       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
193194bad497SJacob Faibussowitsch       CHKERRQ(PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,&nlnk,lnk,lnk_lvl,lnkbt,prow));
19325a8e39fbSHong Zhang       nzi     += nlnk;
19335a8e39fbSHong Zhang       prow     = lnk[prow];
19345a8e39fbSHong Zhang       nzbd++;
193556beaf04SBarry Smith     }
19365a8e39fbSHong Zhang     bdiag[i] = nzbd;
19375a8e39fbSHong Zhang     bi[i+1]  = bi[i] + nzi;
1938ecf371e4SBarry Smith 
19395a8e39fbSHong Zhang     /* if free space is not available, make more free space */
19405a8e39fbSHong Zhang     if (current_space->local_remaining<nzi) {
1941f91af8c7SBarry Smith       nnz  = PetscIntMultTruncate(nzi,n - i); /* estimated and max additional space needed */
19425f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscFreeSpaceGet(nnz,&current_space));
19435f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscFreeSpaceGet(nnz,&current_space_lvl));
19445a8e39fbSHong Zhang       reallocs++;
19455a8e39fbSHong Zhang     }
1946ecf371e4SBarry Smith 
19475a8e39fbSHong Zhang     /* copy data into free_space and free_space_lvl, then initialize lnk */
19485f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt));
19495a8e39fbSHong Zhang     bj_ptr[i]    = current_space->array;
19505a8e39fbSHong Zhang     bjlvl_ptr[i] = current_space_lvl->array;
19515a8e39fbSHong Zhang 
19525a8e39fbSHong Zhang     /* make sure the active row i has diagonal entry */
19532c71b3e2SJacob Faibussowitsch     PetscCheckFalse(*(bj_ptr[i]+bdiag[i]) != i,PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Row %" PetscInt_FMT " has missing diagonal in factored matrix\ntry running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i);
19545a8e39fbSHong Zhang 
19555a8e39fbSHong Zhang     current_space->array               += nzi;
19565a8e39fbSHong Zhang     current_space->local_used          += nzi;
19575a8e39fbSHong Zhang     current_space->local_remaining     -= nzi;
19585a8e39fbSHong Zhang     current_space_lvl->array           += nzi;
19595a8e39fbSHong Zhang     current_space_lvl->local_used      += nzi;
19605a8e39fbSHong Zhang     current_space_lvl->local_remaining -= nzi;
19619e25ed09SBarry Smith   }
19625a8e39fbSHong Zhang 
19635f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isrow,&r));
19645f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isicol,&ic));
19655a8e39fbSHong Zhang 
19665a8e39fbSHong Zhang   /* destroy list of free space and other temporary arrays */
19675f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(bi[n]+1,&bj));
19685f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFreeSpaceContiguous(&free_space,bj)); /* copy free_space -> bj */
19695f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscIncompleteLLDestroy(lnk,lnkbt));
19705f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFreeSpaceDestroy(free_space_lvl));
19715f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree2(bj_ptr,bjlvl_ptr));
19729e25ed09SBarry Smith 
19736cf91177SBarry Smith #if defined(PETSC_USE_INFO)
1974f64065bbSBarry Smith   {
19755a8e39fbSHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
19765f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"Reallocs %" PetscInt_FMT " Fill ratio:given %g needed %g\n",reallocs,(double)f,(double)af));
19775f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"Run with -[sub_]pc_factor_fill %g or use \n",(double)af));
19785f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"PCFactorSetFill([sub]pc,%g);\n",(double)af));
19795f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"for best performance.\n"));
1980335d9088SBarry Smith     if (diagonal_fill) {
19815f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscInfo(A,"Detected and replaced %" PetscInt_FMT " missing diagonals\n",dcount));
1982335d9088SBarry Smith     }
1983f64065bbSBarry Smith   }
198463ba0a88SBarry Smith #endif
1985bbb6d6a8SBarry Smith 
19869e25ed09SBarry Smith   /* put together the new matrix */
19875f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,NULL));
19885f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogObjectParent((PetscObject)fact,(PetscObject)isicol));
1989719d5645SBarry Smith   b    = (Mat_SeqAIJ*)(fact)->data;
19902205254eSKarl Rupp 
1991e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
1992e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
19937c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
19942205254eSKarl Rupp 
19955f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(bi[n],&b->a));
19965a8e39fbSHong Zhang   b->j = bj;
19975a8e39fbSHong Zhang   b->i = bi;
19985a8e39fbSHong Zhang   for (i=0; i<n; i++) bdiag[i] += bi[i];
19995a8e39fbSHong Zhang   b->diag = bdiag;
2000f4259b30SLisandro Dalcin   b->ilen = NULL;
2001f4259b30SLisandro Dalcin   b->imax = NULL;
2002416022c9SBarry Smith   b->row  = isrow;
2003416022c9SBarry Smith   b->col  = iscol;
20045f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectReference((PetscObject)isrow));
20055f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectReference((PetscObject)iscol));
200682bf6240SBarry Smith   b->icol = isicol;
20075f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(n+1,&b->solve_work));
2008416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
20095a8e39fbSHong Zhang      Allocate bdiag, solve_work, new a, new j */
20105f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogObjectMemory((PetscObject)fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar))));
20115a8e39fbSHong Zhang   b->maxnz = b->nz = bi[n];
20122205254eSKarl Rupp 
2013719d5645SBarry Smith   (fact)->info.factor_mallocs    = reallocs;
2014719d5645SBarry Smith   (fact)->info.fill_ratio_given  = f;
2015719d5645SBarry Smith   (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
2016ad04f41aSHong Zhang   (fact)->ops->lufactornumeric   =  MatLUFactorNumeric_SeqAIJ_inplace;
2017f2344125SBarry Smith   if (a->inode.size) {
2018f2344125SBarry Smith     (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
2019d3ac4fa3SBarry Smith   }
20203a40ed3dSBarry Smith   PetscFunctionReturn(0);
20219e25ed09SBarry Smith }
2022d5d45c9bSBarry Smith 
202335233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
20245f44c854SHong Zhang {
20255f44c854SHong Zhang   Mat            C = B;
20265f44c854SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
20275f44c854SHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
20285f44c854SHong Zhang   IS             ip=b->row,iip = b->icol;
20295f44c854SHong Zhang   const PetscInt *rip,*riip;
20305f44c854SHong Zhang   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp;
20315f44c854SHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
20325f44c854SHong Zhang   PetscInt       k,jmin,jmax,*c2r,*il,col,nexti,ili,nz;
20335f44c854SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
2034ace3abfcSBarry Smith   PetscBool      perm_identity;
2035d90ac83dSHong Zhang   FactorShiftCtx sctx;
203698a93161SHong Zhang   PetscReal      rs;
203798a93161SHong Zhang   MatScalar      d,*v;
203898a93161SHong Zhang 
20395f44c854SHong Zhang   PetscFunctionBegin;
204098a93161SHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
20415f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMemzero(&sctx,sizeof(FactorShiftCtx)));
204298a93161SHong Zhang 
2043f4db908eSBarry Smith   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
204498a93161SHong Zhang     sctx.shift_top = info->zeropivot;
204598a93161SHong Zhang     for (i=0; i<mbs; i++) {
204698a93161SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
204798a93161SHong Zhang       d  = (aa)[a->diag[i]];
204898a93161SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
204998a93161SHong Zhang       v  = aa+ai[i];
205098a93161SHong Zhang       nz = ai[i+1] - ai[i];
20512205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
205298a93161SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
205398a93161SHong Zhang     }
205498a93161SHong Zhang     sctx.shift_top *= 1.1;
205598a93161SHong Zhang     sctx.nshift_max = 5;
205698a93161SHong Zhang     sctx.shift_lo   = 0.;
205798a93161SHong Zhang     sctx.shift_hi   = 1.;
205898a93161SHong Zhang   }
20595f44c854SHong Zhang 
20605f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(ip,&rip));
20615f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(iip,&riip));
20625f44c854SHong Zhang 
20635f44c854SHong Zhang   /* allocate working arrays
20645f44c854SHong Zhang      c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col
20655f44c854SHong 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
20665f44c854SHong Zhang   */
20675f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc3(mbs,&rtmp,mbs,&il,mbs,&c2r));
20685f44c854SHong Zhang 
206998a93161SHong Zhang   do {
207007b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
207198a93161SHong Zhang 
2072c5546cabSHong Zhang     for (i=0; i<mbs; i++) c2r[i] = mbs;
20732e987568SSatish Balay     if (mbs) il[0] = 0;
20745f44c854SHong Zhang 
20755f44c854SHong Zhang     for (k = 0; k<mbs; k++) {
20765f44c854SHong Zhang       /* zero rtmp */
20775f44c854SHong Zhang       nz    = bi[k+1] - bi[k];
20785f44c854SHong Zhang       bjtmp = bj + bi[k];
20795f44c854SHong Zhang       for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
20805f44c854SHong Zhang 
20815f44c854SHong Zhang       /* load in initial unfactored row */
20825f44c854SHong Zhang       bval = ba + bi[k];
20835f44c854SHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
20845f44c854SHong Zhang       for (j = jmin; j < jmax; j++) {
20855f44c854SHong Zhang         col = riip[aj[j]];
20865f44c854SHong Zhang         if (col >= k) { /* only take upper triangular entry */
20875f44c854SHong Zhang           rtmp[col] = aa[j];
20885f44c854SHong Zhang           *bval++   = 0.0; /* for in-place factorization */
20895f44c854SHong Zhang         }
20905f44c854SHong Zhang       }
209198a93161SHong Zhang       /* shift the diagonal of the matrix: ZeropivotApply() */
209298a93161SHong Zhang       rtmp[k] += sctx.shift_amount;  /* shift the diagonal of the matrix */
20935f44c854SHong Zhang 
20945f44c854SHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
20955f44c854SHong Zhang       dk = rtmp[k];
20965f44c854SHong Zhang       i  = c2r[k]; /* first row to be added to k_th row  */
20975f44c854SHong Zhang 
20985f44c854SHong Zhang       while (i < k) {
20995f44c854SHong Zhang         nexti = c2r[i]; /* next row to be added to k_th row */
21005f44c854SHong Zhang 
21015f44c854SHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
21025f44c854SHong Zhang         ili     = il[i]; /* index of first nonzero element in U(i,k:bms-1) */
21035f44c854SHong Zhang         uikdi   = -ba[ili]*ba[bdiag[i]]; /* diagonal(k) */
21045f44c854SHong Zhang         dk     += uikdi*ba[ili]; /* update diag[k] */
21055f44c854SHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
21065f44c854SHong Zhang 
21075f44c854SHong Zhang         /* add multiple of row i to k-th row */
21085f44c854SHong Zhang         jmin = ili + 1; jmax = bi[i+1];
21095f44c854SHong Zhang         if (jmin < jmax) {
21105f44c854SHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
21115f44c854SHong Zhang           /* update il and c2r for row i */
21125f44c854SHong Zhang           il[i] = jmin;
21135f44c854SHong Zhang           j     = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i;
21145f44c854SHong Zhang         }
21155f44c854SHong Zhang         i = nexti;
21165f44c854SHong Zhang       }
21175f44c854SHong Zhang 
21185f44c854SHong Zhang       /* copy data into U(k,:) */
211998a93161SHong Zhang       rs   = 0.0;
21205f44c854SHong Zhang       jmin = bi[k]; jmax = bi[k+1]-1;
21215f44c854SHong Zhang       if (jmin < jmax) {
21225f44c854SHong Zhang         for (j=jmin; j<jmax; j++) {
212398a93161SHong Zhang           col = bj[j]; ba[j] = rtmp[col]; rs += PetscAbsScalar(ba[j]);
21245f44c854SHong Zhang         }
21255f44c854SHong Zhang         /* add the k-th row into il and c2r */
21265f44c854SHong Zhang         il[k] = jmin;
21275f44c854SHong Zhang         i     = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k;
21285f44c854SHong Zhang       }
212998a93161SHong Zhang 
213098a93161SHong Zhang       /* MatPivotCheck() */
213198a93161SHong Zhang       sctx.rs = rs;
213298a93161SHong Zhang       sctx.pv = dk;
21335f80ce2aSJacob Faibussowitsch       CHKERRQ(MatPivotCheck(B,A,info,&sctx,i));
213407b50cabSHong Zhang       if (sctx.newshift) break;
213598a93161SHong Zhang       dk = sctx.pv;
213698a93161SHong Zhang 
213798a93161SHong Zhang       ba[bdiag[k]] = 1.0/dk; /* U(k,k) */
213898a93161SHong Zhang     }
213907b50cabSHong Zhang   } while (sctx.newshift);
21405f44c854SHong Zhang 
21415f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree3(rtmp,il,c2r));
21425f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(ip,&rip));
21435f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(iip,&riip));
21445f44c854SHong Zhang 
21455f80ce2aSJacob Faibussowitsch   CHKERRQ(ISIdentity(ip,&perm_identity));
21465f44c854SHong Zhang   if (perm_identity) {
214735233d99SShri Abhyankar     B->ops->solve          = MatSolve_SeqSBAIJ_1_NaturalOrdering;
214835233d99SShri Abhyankar     B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering;
21492169352eSHong Zhang     B->ops->forwardsolve   = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering;
21502169352eSHong Zhang     B->ops->backwardsolve  = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering;
21515f44c854SHong Zhang   } else {
215235233d99SShri Abhyankar     B->ops->solve          = MatSolve_SeqSBAIJ_1;
215335233d99SShri Abhyankar     B->ops->solvetranspose = MatSolve_SeqSBAIJ_1;
215435233d99SShri Abhyankar     B->ops->forwardsolve   = MatForwardSolve_SeqSBAIJ_1;
215535233d99SShri Abhyankar     B->ops->backwardsolve  = MatBackwardSolve_SeqSBAIJ_1;
21565f44c854SHong Zhang   }
21575f44c854SHong Zhang 
21585f44c854SHong Zhang   C->assembled    = PETSC_TRUE;
21595f44c854SHong Zhang   C->preallocated = PETSC_TRUE;
21602205254eSKarl Rupp 
21615f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(C->rmap->n));
216298a93161SHong Zhang 
216398a93161SHong Zhang   /* MatPivotView() */
216498a93161SHong Zhang   if (sctx.nshift) {
2165f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
21665f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscInfo(A,"number of shift_pd tries %" PetscInt_FMT ", 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));
2167f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
21685f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscInfo(A,"number of shift_nz tries %" PetscInt_FMT ", shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount));
2169f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS) {
21705f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscInfo(A,"number of shift_inblocks applied %" PetscInt_FMT ", each shift_amount %g\n",sctx.nshift,(double)info->shiftamount));
217198a93161SHong Zhang     }
217298a93161SHong Zhang   }
21735f44c854SHong Zhang   PetscFunctionReturn(0);
21745f44c854SHong Zhang }
21755f44c854SHong Zhang 
21760a3c351aSHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info)
2177a6175056SHong Zhang {
2178719d5645SBarry Smith   Mat            C = B;
21790968510aSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
21802ed133dbSHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
21819bfd6278SHong Zhang   IS             ip=b->row,iip = b->icol;
21825d0c19d7SBarry Smith   const PetscInt *rip,*riip;
2183c5546cabSHong Zhang   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol,*bjtmp;
21842ed133dbSHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
21852ed133dbSHong Zhang   PetscInt       k,jmin,jmax,*jl,*il,col,nexti,ili,nz;
2186622e2028SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
2187ace3abfcSBarry Smith   PetscBool      perm_identity;
21880e95ead3SHong Zhang   FactorShiftCtx sctx;
21890e95ead3SHong Zhang   PetscReal      rs;
21900e95ead3SHong Zhang   MatScalar      d,*v;
2191d5d45c9bSBarry Smith 
2192a6175056SHong Zhang   PetscFunctionBegin;
21930e95ead3SHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
21945f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMemzero(&sctx,sizeof(FactorShiftCtx)));
21950e95ead3SHong Zhang 
21960e95ead3SHong Zhang   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
21970e95ead3SHong Zhang     sctx.shift_top = info->zeropivot;
21980e95ead3SHong Zhang     for (i=0; i<mbs; i++) {
21990e95ead3SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
22000e95ead3SHong Zhang       d  = (aa)[a->diag[i]];
22010e95ead3SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
22020e95ead3SHong Zhang       v  = aa+ai[i];
22030e95ead3SHong Zhang       nz = ai[i+1] - ai[i];
22042205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
22050e95ead3SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
22060e95ead3SHong Zhang     }
22070e95ead3SHong Zhang     sctx.shift_top *= 1.1;
22080e95ead3SHong Zhang     sctx.nshift_max = 5;
22090e95ead3SHong Zhang     sctx.shift_lo   = 0.;
22100e95ead3SHong Zhang     sctx.shift_hi   = 1.;
22110e95ead3SHong Zhang   }
2212ee45ca4aSHong Zhang 
22135f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(ip,&rip));
22145f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(iip,&riip));
22152ed133dbSHong Zhang 
22162ed133dbSHong Zhang   /* initialization */
22175f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc3(mbs,&rtmp,mbs,&il,mbs,&jl));
22180e95ead3SHong Zhang 
22192ed133dbSHong Zhang   do {
222007b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
22210e95ead3SHong Zhang 
2222c5546cabSHong Zhang     for (i=0; i<mbs; i++) jl[i] = mbs;
2223c5546cabSHong Zhang     il[0] = 0;
22242ed133dbSHong Zhang 
22252ed133dbSHong Zhang     for (k = 0; k<mbs; k++) {
2226c5546cabSHong Zhang       /* zero rtmp */
2227c5546cabSHong Zhang       nz    = bi[k+1] - bi[k];
2228c5546cabSHong Zhang       bjtmp = bj + bi[k];
2229c5546cabSHong Zhang       for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
2230c5546cabSHong Zhang 
2231c05c3958SHong Zhang       bval = ba + bi[k];
22322ed133dbSHong Zhang       /* initialize k-th row by the perm[k]-th row of A */
22332ed133dbSHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
22342ed133dbSHong Zhang       for (j = jmin; j < jmax; j++) {
22359bfd6278SHong Zhang         col = riip[aj[j]];
22362ed133dbSHong Zhang         if (col >= k) { /* only take upper triangular entry */
22372ed133dbSHong Zhang           rtmp[col] = aa[j];
2238c05c3958SHong Zhang           *bval++   = 0.0; /* for in-place factorization */
22392ed133dbSHong Zhang         }
22402ed133dbSHong Zhang       }
224139e3d630SHong Zhang       /* shift the diagonal of the matrix */
2242540703acSHong Zhang       if (sctx.nshift) rtmp[k] += sctx.shift_amount;
22432ed133dbSHong Zhang 
22442ed133dbSHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
22452ed133dbSHong Zhang       dk = rtmp[k];
22462ed133dbSHong Zhang       i  = jl[k]; /* first row to be added to k_th row  */
22472ed133dbSHong Zhang 
22482ed133dbSHong Zhang       while (i < k) {
22492ed133dbSHong Zhang         nexti = jl[i]; /* next row to be added to k_th row */
22502ed133dbSHong Zhang 
22512ed133dbSHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
22522ed133dbSHong Zhang         ili     = il[i]; /* index of first nonzero element in U(i,k:bms-1) */
22532ed133dbSHong Zhang         uikdi   = -ba[ili]*ba[bi[i]]; /* diagonal(k) */
22542ed133dbSHong Zhang         dk     += uikdi*ba[ili];
22552ed133dbSHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
22562ed133dbSHong Zhang 
22572ed133dbSHong Zhang         /* add multiple of row i to k-th row */
22582ed133dbSHong Zhang         jmin = ili + 1; jmax = bi[i+1];
22592ed133dbSHong Zhang         if (jmin < jmax) {
22602ed133dbSHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
22612ed133dbSHong Zhang           /* update il and jl for row i */
22622ed133dbSHong Zhang           il[i] = jmin;
22632ed133dbSHong Zhang           j     = bj[jmin]; jl[i] = jl[j]; jl[j] = i;
22642ed133dbSHong Zhang         }
22652ed133dbSHong Zhang         i = nexti;
22662ed133dbSHong Zhang       }
22672ed133dbSHong Zhang 
22680697b6caSHong Zhang       /* shift the diagonals when zero pivot is detected */
22690697b6caSHong Zhang       /* compute rs=sum of abs(off-diagonal) */
22700697b6caSHong Zhang       rs   = 0.0;
22713c9e8598SHong Zhang       jmin = bi[k]+1;
22723c9e8598SHong Zhang       nz   = bi[k+1] - jmin;
22733c9e8598SHong Zhang       bcol = bj + jmin;
227404fbf559SBarry Smith       for (j=0; j<nz; j++) {
227504fbf559SBarry Smith         rs += PetscAbsScalar(rtmp[bcol[j]]);
22763c9e8598SHong Zhang       }
2277540703acSHong Zhang 
2278540703acSHong Zhang       sctx.rs = rs;
2279540703acSHong Zhang       sctx.pv = dk;
22805f80ce2aSJacob Faibussowitsch       CHKERRQ(MatPivotCheck(B,A,info,&sctx,k));
228107b50cabSHong Zhang       if (sctx.newshift) break;
22820e95ead3SHong Zhang       dk = sctx.pv;
22833c9e8598SHong Zhang 
22843c9e8598SHong Zhang       /* copy data into U(k,:) */
228539e3d630SHong Zhang       ba[bi[k]] = 1.0/dk; /* U(k,k) */
228639e3d630SHong Zhang       jmin      = bi[k]+1; jmax = bi[k+1];
228739e3d630SHong Zhang       if (jmin < jmax) {
228839e3d630SHong Zhang         for (j=jmin; j<jmax; j++) {
2289c6ae9bfcSHong Zhang           col = bj[j]; ba[j] = rtmp[col];
22903c9e8598SHong Zhang         }
229139e3d630SHong Zhang         /* add the k-th row into il and jl */
22923c9e8598SHong Zhang         il[k] = jmin;
22933c9e8598SHong Zhang         i     = bj[jmin]; jl[k] = jl[i]; jl[i] = k;
22943c9e8598SHong Zhang       }
22953c9e8598SHong Zhang     }
229607b50cabSHong Zhang   } while (sctx.newshift);
22970e95ead3SHong Zhang 
22985f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree3(rtmp,il,jl));
22995f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(ip,&rip));
23005f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(iip,&riip));
2301db4efbfdSBarry Smith 
23025f80ce2aSJacob Faibussowitsch   CHKERRQ(ISIdentity(ip,&perm_identity));
2303db4efbfdSBarry Smith   if (perm_identity) {
23040a3c351aSHong Zhang     B->ops->solve          = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23050a3c351aSHong Zhang     B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23060a3c351aSHong Zhang     B->ops->forwardsolve   = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23070a3c351aSHong Zhang     B->ops->backwardsolve  = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
2308db4efbfdSBarry Smith   } else {
23090a3c351aSHong Zhang     B->ops->solve          = MatSolve_SeqSBAIJ_1_inplace;
23100a3c351aSHong Zhang     B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_inplace;
23110a3c351aSHong Zhang     B->ops->forwardsolve   = MatForwardSolve_SeqSBAIJ_1_inplace;
23120a3c351aSHong Zhang     B->ops->backwardsolve  = MatBackwardSolve_SeqSBAIJ_1_inplace;
2313db4efbfdSBarry Smith   }
2314db4efbfdSBarry Smith 
23153c9e8598SHong Zhang   C->assembled    = PETSC_TRUE;
23163c9e8598SHong Zhang   C->preallocated = PETSC_TRUE;
23172205254eSKarl Rupp 
23185f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(C->rmap->n));
2319540703acSHong Zhang   if (sctx.nshift) {
2320f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
23215f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscInfo(A,"number of shiftnz tries %" PetscInt_FMT ", shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount));
2322f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
23235f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscInfo(A,"number of shiftpd tries %" PetscInt_FMT ", shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount));
23240697b6caSHong Zhang     }
23253c9e8598SHong Zhang   }
23263c9e8598SHong Zhang   PetscFunctionReturn(0);
23273c9e8598SHong Zhang }
2328a6175056SHong Zhang 
23298a76ed4fSHong Zhang /*
23308a76ed4fSHong Zhang    icc() under revised new data structure.
23318a76ed4fSHong Zhang    Factored arrays bj and ba are stored as
23328a76ed4fSHong Zhang      U(0,:),...,U(i,:),U(n-1,:)
23338a76ed4fSHong Zhang 
23348a76ed4fSHong Zhang    ui=fact->i is an array of size n+1, in which
23358a76ed4fSHong Zhang    ui+
23368a76ed4fSHong Zhang      ui[i]:  points to 1st entry of U(i,:),i=0,...,n-1
23378a76ed4fSHong Zhang      ui[n]:  points to U(n-1,n-1)+1
23388a76ed4fSHong Zhang 
23398a76ed4fSHong Zhang   udiag=fact->diag is an array of size n,in which
23408a76ed4fSHong Zhang      udiag[i]: points to diagonal of U(i,:), i=0,...,n-1
23418a76ed4fSHong Zhang 
23428a76ed4fSHong Zhang    U(i,:) contains udiag[i] as its last entry, i.e.,
23438a76ed4fSHong Zhang     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
23448a76ed4fSHong Zhang */
23458a76ed4fSHong Zhang 
234635233d99SShri Abhyankar PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2347a6175056SHong Zhang {
23480968510aSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
2349ed59401aSHong Zhang   Mat_SeqSBAIJ       *b;
2350ace3abfcSBarry Smith   PetscBool          perm_identity,missing;
23515f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
23525d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
2353ed59401aSHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
23540298fd71SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=NULL,d;
23555a8e39fbSHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
2356ed59401aSHong Zhang   PetscReal          fill          =info->fill,levels=info->levels;
23570298fd71SBarry Smith   PetscFreeSpaceList free_space    =NULL,current_space=NULL;
23580298fd71SBarry Smith   PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL;
23597a48dd6fSHong Zhang   PetscBT            lnkbt;
2360b635d36bSHong Zhang   IS                 iperm;
2361a6175056SHong Zhang 
2362a6175056SHong Zhang   PetscFunctionBegin;
23632c71b3e2SJacob Faibussowitsch   PetscCheckFalse(A->rmap->n != A->cmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %" PetscInt_FMT " columns %" PetscInt_FMT,A->rmap->n,A->cmap->n);
23645f80ce2aSJacob Faibussowitsch   CHKERRQ(MatMissingDiagonal(A,&missing,&d));
2365*28b400f6SJacob Faibussowitsch   PetscCheck(!missing,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %" PetscInt_FMT,d);
23665f80ce2aSJacob Faibussowitsch   CHKERRQ(ISIdentity(perm,&perm_identity));
23675f80ce2aSJacob Faibussowitsch   CHKERRQ(ISInvertPermutation(perm,PETSC_DECIDE,&iperm));
23687a48dd6fSHong Zhang 
23695f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(am+1,&ui));
23705f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(am+1,&udiag));
237139e3d630SHong Zhang   ui[0] = 0;
237239e3d630SHong Zhang 
23738a76ed4fSHong Zhang   /* ICC(0) without matrix ordering: simply rearrange column indices */
2374622e2028SHong Zhang   if (!levels && perm_identity) {
23755f44c854SHong Zhang     for (i=0; i<am; i++) {
2376c5546cabSHong Zhang       ncols    = ai[i+1] - a->diag[i];
2377c5546cabSHong Zhang       ui[i+1]  = ui[i] + ncols;
2378c5546cabSHong Zhang       udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */
2379dc1e2a3fSHong Zhang     }
23805f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscMalloc1(ui[am]+1,&uj));
2381dc1e2a3fSHong Zhang     cols = uj;
2382dc1e2a3fSHong Zhang     for (i=0; i<am; i++) {
23835f44c854SHong Zhang       aj    = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */
2384dc1e2a3fSHong Zhang       ncols = ai[i+1] - a->diag[i] -1;
23855f44c854SHong Zhang       for (j=0; j<ncols; j++) *cols++ = aj[j];
2386910cf402Sprj-       *cols++ = i; /* diagonal is located as the last entry of U(i,:) */
23875f44c854SHong Zhang     }
23885f44c854SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
23895f80ce2aSJacob Faibussowitsch     CHKERRQ(ISGetIndices(iperm,&riip));
23905f80ce2aSJacob Faibussowitsch     CHKERRQ(ISGetIndices(perm,&rip));
23915f44c854SHong Zhang 
23925f44c854SHong Zhang     /* initialization */
23935f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscMalloc1(am+1,&ajtmp));
23945f44c854SHong Zhang 
23955f44c854SHong Zhang     /* jl: linked list for storing indices of the pivot rows
23965f44c854SHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
23975f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscMalloc4(am,&uj_ptr,am,&uj_lvl_ptr,am,&jl,am,&il));
23985f44c854SHong Zhang     for (i=0; i<am; i++) {
23995f44c854SHong Zhang       jl[i] = am; il[i] = 0;
24005f44c854SHong Zhang     }
24015f44c854SHong Zhang 
24025f44c854SHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
24035f44c854SHong Zhang     nlnk = am + 1;
24045f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt));
24055f44c854SHong Zhang 
240695b5ac22SHong Zhang     /* initial FreeSpace size is fill*(ai[am]+am)/2 */
24075f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,(ai[am]+am)/2),&free_space));
24085f44c854SHong Zhang     current_space     = free_space;
24095f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,(ai[am]+am)/2),&free_space_lvl));
24105f44c854SHong Zhang     current_space_lvl = free_space_lvl;
24115f44c854SHong Zhang 
24125f44c854SHong Zhang     for (k=0; k<am; k++) {  /* for each active row k */
24135f44c854SHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
24145f44c854SHong Zhang       nzk   = 0;
24155f44c854SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
2416*28b400f6SJacob Faibussowitsch       PetscCheck(ncols,PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %" PetscInt_FMT " in permuted ordering %" PetscInt_FMT,rip[k],k);
24175f44c854SHong Zhang       ncols_upper = 0;
24185f44c854SHong Zhang       for (j=0; j<ncols; j++) {
24195f44c854SHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
24205f44c854SHong Zhang         if (riip[i] >= k) { /* only take upper triangular entry */
24215f44c854SHong Zhang           ajtmp[ncols_upper] = i;
24225f44c854SHong Zhang           ncols_upper++;
24235f44c854SHong Zhang         }
24245f44c854SHong Zhang       }
242594bad497SJacob Faibussowitsch       CHKERRQ(PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,&nlnk,lnk,lnk_lvl,lnkbt));
24265f44c854SHong Zhang       nzk += nlnk;
24275f44c854SHong Zhang 
24285f44c854SHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
24295f44c854SHong Zhang       prow = jl[k]; /* 1st pivot row */
24305f44c854SHong Zhang 
24315f44c854SHong Zhang       while (prow < k) {
24325f44c854SHong Zhang         nextprow = jl[prow];
24335f44c854SHong Zhang 
24345f44c854SHong Zhang         /* merge prow into k-th row */
24355f44c854SHong Zhang         jmin  = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */
24365f44c854SHong Zhang         jmax  = ui[prow+1];
24375f44c854SHong Zhang         ncols = jmax-jmin;
24385f44c854SHong Zhang         i     = jmin - ui[prow];
24395f44c854SHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
24405f44c854SHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
24415f44c854SHong Zhang         j     = *(uj - 1);
244294bad497SJacob Faibussowitsch         CHKERRQ(PetscICCLLAddSorted(ncols,cols,levels,uj,am,&nlnk,lnk,lnk_lvl,lnkbt,j));
24435f44c854SHong Zhang         nzk  += nlnk;
24445f44c854SHong Zhang 
24455f44c854SHong Zhang         /* update il and jl for prow */
24465f44c854SHong Zhang         if (jmin < jmax) {
24475f44c854SHong Zhang           il[prow] = jmin;
24485f44c854SHong Zhang           j        = *cols; jl[prow] = jl[j]; jl[j] = prow;
24495f44c854SHong Zhang         }
24505f44c854SHong Zhang         prow = nextprow;
24515f44c854SHong Zhang       }
24525f44c854SHong Zhang 
24535f44c854SHong Zhang       /* if free space is not available, make more free space */
24545f44c854SHong Zhang       if (current_space->local_remaining<nzk) {
24555f44c854SHong Zhang         i    = am - k + 1; /* num of unfactored rows */
2456f91af8c7SBarry Smith         i    = PetscIntMultTruncate(i,PetscMin(nzk, i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
24575f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscFreeSpaceGet(i,&current_space));
24585f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscFreeSpaceGet(i,&current_space_lvl));
24595f44c854SHong Zhang         reallocs++;
24605f44c854SHong Zhang       }
24615f44c854SHong Zhang 
24625f44c854SHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
24632c71b3e2SJacob Faibussowitsch       PetscCheckFalse(nzk == 0,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %" PetscInt_FMT " in ICC matrix factor",k);
24645f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt));
24655f44c854SHong Zhang 
24665f44c854SHong Zhang       /* add the k-th row into il and jl */
24675f44c854SHong Zhang       if (nzk > 1) {
24685f44c854SHong Zhang         i     = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
24695f44c854SHong Zhang         jl[k] = jl[i]; jl[i] = k;
24705f44c854SHong Zhang         il[k] = ui[k] + 1;
24715f44c854SHong Zhang       }
24725f44c854SHong Zhang       uj_ptr[k]     = current_space->array;
24735f44c854SHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
24745f44c854SHong Zhang 
24755f44c854SHong Zhang       current_space->array           += nzk;
24765f44c854SHong Zhang       current_space->local_used      += nzk;
24775f44c854SHong Zhang       current_space->local_remaining -= nzk;
24785f44c854SHong Zhang 
24795f44c854SHong Zhang       current_space_lvl->array           += nzk;
24805f44c854SHong Zhang       current_space_lvl->local_used      += nzk;
24815f44c854SHong Zhang       current_space_lvl->local_remaining -= nzk;
24825f44c854SHong Zhang 
24835f44c854SHong Zhang       ui[k+1] = ui[k] + nzk;
24845f44c854SHong Zhang     }
24855f44c854SHong Zhang 
24865f80ce2aSJacob Faibussowitsch     CHKERRQ(ISRestoreIndices(perm,&rip));
24875f80ce2aSJacob Faibussowitsch     CHKERRQ(ISRestoreIndices(iperm,&riip));
24885f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscFree4(uj_ptr,uj_lvl_ptr,jl,il));
24895f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscFree(ajtmp));
24905f44c854SHong Zhang 
24919263d837SHong Zhang     /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */
24925f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscMalloc1(ui[am]+1,&uj));
24935f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag)); /* store matrix factor  */
24945f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscIncompleteLLDestroy(lnk,lnkbt));
24955f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscFreeSpaceDestroy(free_space_lvl));
24965f44c854SHong Zhang 
24975f44c854SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
24985f44c854SHong Zhang 
24995f44c854SHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
25005f44c854SHong Zhang   b               = (Mat_SeqSBAIJ*)(fact)->data;
25015f44c854SHong Zhang   b->singlemalloc = PETSC_FALSE;
25022205254eSKarl Rupp 
25035f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(ui[am]+1,&b->a));
25042205254eSKarl Rupp 
25055f44c854SHong Zhang   b->j             = uj;
25065f44c854SHong Zhang   b->i             = ui;
25075f44c854SHong Zhang   b->diag          = udiag;
25087f53bb6cSHong Zhang   b->free_diag     = PETSC_TRUE;
2509f4259b30SLisandro Dalcin   b->ilen          = NULL;
2510f4259b30SLisandro Dalcin   b->imax          = NULL;
25115f44c854SHong Zhang   b->row           = perm;
25125f44c854SHong Zhang   b->col           = perm;
25135f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectReference((PetscObject)perm));
25145f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectReference((PetscObject)perm));
25155f44c854SHong Zhang   b->icol          = iperm;
25165f44c854SHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
25172205254eSKarl Rupp 
25185f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(am+1,&b->solve_work));
25195f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogObjectMemory((PetscObject)fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar))));
25202205254eSKarl Rupp 
25215f44c854SHong Zhang   b->maxnz   = b->nz = ui[am];
25225f44c854SHong Zhang   b->free_a  = PETSC_TRUE;
25235f44c854SHong Zhang   b->free_ij = PETSC_TRUE;
25245f44c854SHong Zhang 
2525f284f12aSHong Zhang   fact->info.factor_mallocs   = reallocs;
2526f284f12aSHong Zhang   fact->info.fill_ratio_given = fill;
25275f44c854SHong Zhang   if (ai[am] != 0) {
25286a69fef8SHong Zhang     /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */
252995b5ac22SHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am);
25305f44c854SHong Zhang   } else {
2531f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
25325f44c854SHong Zhang   }
25339263d837SHong Zhang #if defined(PETSC_USE_INFO)
25349263d837SHong Zhang   if (ai[am] != 0) {
25359263d837SHong Zhang     PetscReal af = fact->info.fill_ratio_needed;
25365f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"Reallocs %" PetscInt_FMT " Fill ratio:given %g needed %g\n",reallocs,(double)fill,(double)af));
25375f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"Run with -pc_factor_fill %g or use \n",(double)af));
25385f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"PCFactorSetFill(pc,%g) for best performance.\n",(double)af));
25399263d837SHong Zhang   } else {
25405f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"Empty matrix\n"));
25419263d837SHong Zhang   }
25429263d837SHong Zhang #endif
254335233d99SShri Abhyankar   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
25445f44c854SHong Zhang   PetscFunctionReturn(0);
25455f44c854SHong Zhang }
25465f44c854SHong Zhang 
2547ad04f41aSHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
25485f44c854SHong Zhang {
25495f44c854SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
25505f44c854SHong Zhang   Mat_SeqSBAIJ       *b;
2551ace3abfcSBarry Smith   PetscBool          perm_identity,missing;
25525f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
25535f44c854SHong Zhang   const PetscInt     *rip,*riip;
25545f44c854SHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
25550298fd71SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=NULL,d;
25565f44c854SHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
25575f44c854SHong Zhang   PetscReal          fill          =info->fill,levels=info->levels;
25580298fd71SBarry Smith   PetscFreeSpaceList free_space    =NULL,current_space=NULL;
25590298fd71SBarry Smith   PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL;
25605f44c854SHong Zhang   PetscBT            lnkbt;
25615f44c854SHong Zhang   IS                 iperm;
25625f44c854SHong Zhang 
25635f44c854SHong Zhang   PetscFunctionBegin;
25642c71b3e2SJacob Faibussowitsch   PetscCheckFalse(A->rmap->n != A->cmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %" PetscInt_FMT " columns %" PetscInt_FMT,A->rmap->n,A->cmap->n);
25655f80ce2aSJacob Faibussowitsch   CHKERRQ(MatMissingDiagonal(A,&missing,&d));
2566*28b400f6SJacob Faibussowitsch   PetscCheck(!missing,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %" PetscInt_FMT,d);
25675f80ce2aSJacob Faibussowitsch   CHKERRQ(ISIdentity(perm,&perm_identity));
25685f80ce2aSJacob Faibussowitsch   CHKERRQ(ISInvertPermutation(perm,PETSC_DECIDE,&iperm));
25695f44c854SHong Zhang 
25705f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(am+1,&ui));
25715f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(am+1,&udiag));
25725f44c854SHong Zhang   ui[0] = 0;
25735f44c854SHong Zhang 
25745f44c854SHong Zhang   /* ICC(0) without matrix ordering: simply copies fill pattern */
25755f44c854SHong Zhang   if (!levels && perm_identity) {
25765f44c854SHong Zhang 
25775f44c854SHong Zhang     for (i=0; i<am; i++) {
25785f44c854SHong Zhang       ui[i+1]  = ui[i] + ai[i+1] - a->diag[i];
25795f44c854SHong Zhang       udiag[i] = ui[i];
2580ed59401aSHong Zhang     }
25815f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscMalloc1(ui[am]+1,&uj));
258239e3d630SHong Zhang     cols = uj;
2583ed59401aSHong Zhang     for (i=0; i<am; i++) {
2584ed59401aSHong Zhang       aj    = a->j + a->diag[i];
258539e3d630SHong Zhang       ncols = ui[i+1] - ui[i];
258639e3d630SHong Zhang       for (j=0; j<ncols; j++) *cols++ = *aj++;
2587ed59401aSHong Zhang     }
258839e3d630SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
25895f80ce2aSJacob Faibussowitsch     CHKERRQ(ISGetIndices(iperm,&riip));
25905f80ce2aSJacob Faibussowitsch     CHKERRQ(ISGetIndices(perm,&rip));
2591b635d36bSHong Zhang 
25927a48dd6fSHong Zhang     /* initialization */
25935f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscMalloc1(am+1,&ajtmp));
25947a48dd6fSHong Zhang 
25957a48dd6fSHong Zhang     /* jl: linked list for storing indices of the pivot rows
25967a48dd6fSHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
25975f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscMalloc4(am,&uj_ptr,am,&uj_lvl_ptr,am,&jl,am,&il));
25987a48dd6fSHong Zhang     for (i=0; i<am; i++) {
25997a48dd6fSHong Zhang       jl[i] = am; il[i] = 0;
26007a48dd6fSHong Zhang     }
26017a48dd6fSHong Zhang 
26027a48dd6fSHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
26037a48dd6fSHong Zhang     nlnk = am + 1;
26045f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt));
26057a48dd6fSHong Zhang 
26067a48dd6fSHong Zhang     /* initial FreeSpace size is fill*(ai[am]+1) */
26075f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,ai[am]+1),&free_space));
26087a48dd6fSHong Zhang     current_space     = free_space;
26095f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,ai[am]+1),&free_space_lvl));
26107a48dd6fSHong Zhang     current_space_lvl = free_space_lvl;
26117a48dd6fSHong Zhang 
26127a48dd6fSHong Zhang     for (k=0; k<am; k++) {  /* for each active row k */
26137a48dd6fSHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
26147a48dd6fSHong Zhang       nzk   = 0;
2615622e2028SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
2616*28b400f6SJacob Faibussowitsch       PetscCheck(ncols,PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %" PetscInt_FMT " in permuted ordering %" PetscInt_FMT,rip[k],k);
2617622e2028SHong Zhang       ncols_upper = 0;
2618622e2028SHong Zhang       for (j=0; j<ncols; j++) {
2619b635d36bSHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
2620b635d36bSHong Zhang         if (riip[i] >= k) { /* only take upper triangular entry */
26215a8e39fbSHong Zhang           ajtmp[ncols_upper] = i;
2622622e2028SHong Zhang           ncols_upper++;
2623622e2028SHong Zhang         }
2624622e2028SHong Zhang       }
262594bad497SJacob Faibussowitsch       CHKERRQ(PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,&nlnk,lnk,lnk_lvl,lnkbt));
26267a48dd6fSHong Zhang       nzk += nlnk;
26277a48dd6fSHong Zhang 
26287a48dd6fSHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
26297a48dd6fSHong Zhang       prow = jl[k]; /* 1st pivot row */
26307a48dd6fSHong Zhang 
26317a48dd6fSHong Zhang       while (prow < k) {
26327a48dd6fSHong Zhang         nextprow = jl[prow];
26337a48dd6fSHong Zhang 
26347a48dd6fSHong Zhang         /* merge prow into k-th row */
26357a48dd6fSHong Zhang         jmin  = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */
26367a48dd6fSHong Zhang         jmax  = ui[prow+1];
26377a48dd6fSHong Zhang         ncols = jmax-jmin;
2638ed59401aSHong Zhang         i     = jmin - ui[prow];
2639ed59401aSHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
26405a8e39fbSHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
26415a8e39fbSHong Zhang         j     = *(uj - 1);
264294bad497SJacob Faibussowitsch         CHKERRQ(PetscICCLLAddSorted(ncols,cols,levels,uj,am,&nlnk,lnk,lnk_lvl,lnkbt,j));
26437a48dd6fSHong Zhang         nzk  += nlnk;
26447a48dd6fSHong Zhang 
26457a48dd6fSHong Zhang         /* update il and jl for prow */
26467a48dd6fSHong Zhang         if (jmin < jmax) {
26477a48dd6fSHong Zhang           il[prow] = jmin;
26487a48dd6fSHong Zhang           j        = *cols; jl[prow] = jl[j]; jl[j] = prow;
26497a48dd6fSHong Zhang         }
26507a48dd6fSHong Zhang         prow = nextprow;
26517a48dd6fSHong Zhang       }
26527a48dd6fSHong Zhang 
26537a48dd6fSHong Zhang       /* if free space is not available, make more free space */
26547a48dd6fSHong Zhang       if (current_space->local_remaining<nzk) {
26557a48dd6fSHong Zhang         i    = am - k + 1; /* num of unfactored rows */
2656f91af8c7SBarry Smith         i    = PetscIntMultTruncate(i,PetscMin(nzk, i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
26575f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscFreeSpaceGet(i,&current_space));
26585f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscFreeSpaceGet(i,&current_space_lvl));
26597a48dd6fSHong Zhang         reallocs++;
26607a48dd6fSHong Zhang       }
26617a48dd6fSHong Zhang 
26627a48dd6fSHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
2663*28b400f6SJacob Faibussowitsch       PetscCheck(nzk,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %" PetscInt_FMT " in ICC matrix factor",k);
26645f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt));
26657a48dd6fSHong Zhang 
26667a48dd6fSHong Zhang       /* add the k-th row into il and jl */
266739e3d630SHong Zhang       if (nzk > 1) {
26687a48dd6fSHong Zhang         i     = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
26697a48dd6fSHong Zhang         jl[k] = jl[i]; jl[i] = k;
26707a48dd6fSHong Zhang         il[k] = ui[k] + 1;
26717a48dd6fSHong Zhang       }
26727a48dd6fSHong Zhang       uj_ptr[k]     = current_space->array;
26737a48dd6fSHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
26747a48dd6fSHong Zhang 
26757a48dd6fSHong Zhang       current_space->array           += nzk;
26767a48dd6fSHong Zhang       current_space->local_used      += nzk;
26777a48dd6fSHong Zhang       current_space->local_remaining -= nzk;
26787a48dd6fSHong Zhang 
26797a48dd6fSHong Zhang       current_space_lvl->array           += nzk;
26807a48dd6fSHong Zhang       current_space_lvl->local_used      += nzk;
26817a48dd6fSHong Zhang       current_space_lvl->local_remaining -= nzk;
26827a48dd6fSHong Zhang 
26837a48dd6fSHong Zhang       ui[k+1] = ui[k] + nzk;
26847a48dd6fSHong Zhang     }
26857a48dd6fSHong Zhang 
26866cf91177SBarry Smith #if defined(PETSC_USE_INFO)
26877a48dd6fSHong Zhang     if (ai[am] != 0) {
268839e3d630SHong Zhang       PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]);
26895f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscInfo(A,"Reallocs %" PetscInt_FMT " Fill ratio:given %g needed %g\n",reallocs,(double)fill,(double)af));
26905f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscInfo(A,"Run with -pc_factor_fill %g or use \n",(double)af));
26915f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscInfo(A,"PCFactorSetFill(pc,%g) for best performance.\n",(double)af));
26927a48dd6fSHong Zhang     } else {
26935f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscInfo(A,"Empty matrix\n"));
26947a48dd6fSHong Zhang     }
269563ba0a88SBarry Smith #endif
26967a48dd6fSHong Zhang 
26975f80ce2aSJacob Faibussowitsch     CHKERRQ(ISRestoreIndices(perm,&rip));
26985f80ce2aSJacob Faibussowitsch     CHKERRQ(ISRestoreIndices(iperm,&riip));
26995f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscFree4(uj_ptr,uj_lvl_ptr,jl,il));
27005f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscFree(ajtmp));
27017a48dd6fSHong Zhang 
27027a48dd6fSHong Zhang     /* destroy list of free space and other temporary array(s) */
27035f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscMalloc1(ui[am]+1,&uj));
27045f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscFreeSpaceContiguous(&free_space,uj));
27055f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscIncompleteLLDestroy(lnk,lnkbt));
27065f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscFreeSpaceDestroy(free_space_lvl));
27077a48dd6fSHong Zhang 
270839e3d630SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
270939e3d630SHong Zhang 
27107a48dd6fSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
27117a48dd6fSHong Zhang 
2712f284f12aSHong Zhang   b               = (Mat_SeqSBAIJ*)fact->data;
27137a48dd6fSHong Zhang   b->singlemalloc = PETSC_FALSE;
27142205254eSKarl Rupp 
27155f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(ui[am]+1,&b->a));
27162205254eSKarl Rupp 
27177a48dd6fSHong Zhang   b->j         = uj;
27187a48dd6fSHong Zhang   b->i         = ui;
27195f44c854SHong Zhang   b->diag      = udiag;
27207f53bb6cSHong Zhang   b->free_diag = PETSC_TRUE;
2721f4259b30SLisandro Dalcin   b->ilen      = NULL;
2722f4259b30SLisandro Dalcin   b->imax      = NULL;
27237a48dd6fSHong Zhang   b->row       = perm;
2724b635d36bSHong Zhang   b->col       = perm;
27252205254eSKarl Rupp 
27265f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectReference((PetscObject)perm));
27275f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectReference((PetscObject)perm));
27282205254eSKarl Rupp 
2729b635d36bSHong Zhang   b->icol          = iperm;
27307a48dd6fSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
27315f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(am+1,&b->solve_work));
27325f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogObjectMemory((PetscObject)fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar))));
27337a48dd6fSHong Zhang   b->maxnz         = b->nz = ui[am];
2734e6b907acSBarry Smith   b->free_a        = PETSC_TRUE;
2735e6b907acSBarry Smith   b->free_ij       = PETSC_TRUE;
27367a48dd6fSHong Zhang 
2737f284f12aSHong Zhang   fact->info.factor_mallocs   = reallocs;
2738f284f12aSHong Zhang   fact->info.fill_ratio_given = fill;
27397a48dd6fSHong Zhang   if (ai[am] != 0) {
2740f284f12aSHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
27417a48dd6fSHong Zhang   } else {
2742f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
27437a48dd6fSHong Zhang   }
27440a3c351aSHong Zhang   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace;
2745a6175056SHong Zhang   PetscFunctionReturn(0);
2746a6175056SHong Zhang }
2747d5d45c9bSBarry Smith 
274835233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
27490be760fbSHong Zhang {
27500be760fbSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
27510be760fbSHong Zhang   Mat_SeqSBAIJ       *b;
27529186b105SHong Zhang   PetscBool          perm_identity,missing;
27530be760fbSHong Zhang   PetscReal          fill = info->fill;
27540be760fbSHong Zhang   const PetscInt     *rip,*riip;
27550be760fbSHong Zhang   PetscInt           i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
27560be760fbSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
27570be760fbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr,*udiag;
27580298fd71SBarry Smith   PetscFreeSpaceList free_space=NULL,current_space=NULL;
27590be760fbSHong Zhang   PetscBT            lnkbt;
27600be760fbSHong Zhang   IS                 iperm;
27610be760fbSHong Zhang 
27620be760fbSHong Zhang   PetscFunctionBegin;
27632c71b3e2SJacob Faibussowitsch   PetscCheckFalse(A->rmap->n != A->cmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %" PetscInt_FMT " columns %" PetscInt_FMT,A->rmap->n,A->cmap->n);
27645f80ce2aSJacob Faibussowitsch   CHKERRQ(MatMissingDiagonal(A,&missing,&i));
2765*28b400f6SJacob Faibussowitsch   PetscCheck(!missing,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %" PetscInt_FMT,i);
27669186b105SHong Zhang 
27670be760fbSHong Zhang   /* check whether perm is the identity mapping */
27685f80ce2aSJacob Faibussowitsch   CHKERRQ(ISIdentity(perm,&perm_identity));
27695f80ce2aSJacob Faibussowitsch   CHKERRQ(ISInvertPermutation(perm,PETSC_DECIDE,&iperm));
27705f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(iperm,&riip));
27715f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(perm,&rip));
27720be760fbSHong Zhang 
27730be760fbSHong Zhang   /* initialization */
27745f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(am+1,&ui));
27755f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(am+1,&udiag));
27760be760fbSHong Zhang   ui[0] = 0;
27770be760fbSHong Zhang 
27780be760fbSHong Zhang   /* jl: linked list for storing indices of the pivot rows
27790be760fbSHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
27805f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc4(am,&ui_ptr,am,&jl,am,&il,am,&cols));
27810be760fbSHong Zhang   for (i=0; i<am; i++) {
27820be760fbSHong Zhang     jl[i] = am; il[i] = 0;
27830be760fbSHong Zhang   }
27840be760fbSHong Zhang 
27850be760fbSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
27860be760fbSHong Zhang   nlnk = am + 1;
27875f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLLCreate(am,am,nlnk,lnk,lnkbt));
27880be760fbSHong Zhang 
278995b5ac22SHong Zhang   /* initial FreeSpace size is fill*(ai[am]+am)/2 */
27905f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,(ai[am]+am)/2),&free_space));
27910be760fbSHong Zhang   current_space = free_space;
27920be760fbSHong Zhang 
27930be760fbSHong Zhang   for (k=0; k<am; k++) {  /* for each active row k */
27940be760fbSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
27950be760fbSHong Zhang     nzk   = 0;
27960be760fbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
2797*28b400f6SJacob Faibussowitsch     PetscCheck(ncols,PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %" PetscInt_FMT " in permuted ordering %" PetscInt_FMT,rip[k],k);
27980be760fbSHong Zhang     ncols_upper = 0;
27990be760fbSHong Zhang     for (j=0; j<ncols; j++) {
28000be760fbSHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
28010be760fbSHong Zhang       if (i >= k) { /* only take upper triangular entry */
28020be760fbSHong Zhang         cols[ncols_upper] = i;
28030be760fbSHong Zhang         ncols_upper++;
28040be760fbSHong Zhang       }
28050be760fbSHong Zhang     }
280694bad497SJacob Faibussowitsch     CHKERRQ(PetscLLAdd(ncols_upper,cols,am,&nlnk,lnk,lnkbt));
28070be760fbSHong Zhang     nzk += nlnk;
28080be760fbSHong Zhang 
28090be760fbSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
28100be760fbSHong Zhang     prow = jl[k]; /* 1st pivot row */
28110be760fbSHong Zhang 
28120be760fbSHong Zhang     while (prow < k) {
28130be760fbSHong Zhang       nextprow = jl[prow];
28140be760fbSHong Zhang       /* merge prow into k-th row */
28150be760fbSHong Zhang       jmin   = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */
28160be760fbSHong Zhang       jmax   = ui[prow+1];
28170be760fbSHong Zhang       ncols  = jmax-jmin;
28180be760fbSHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
281994bad497SJacob Faibussowitsch       CHKERRQ(PetscLLAddSorted(ncols,uj_ptr,am,&nlnk,lnk,lnkbt));
28200be760fbSHong Zhang       nzk   += nlnk;
28210be760fbSHong Zhang 
28220be760fbSHong Zhang       /* update il and jl for prow */
28230be760fbSHong Zhang       if (jmin < jmax) {
28240be760fbSHong Zhang         il[prow] = jmin;
28252205254eSKarl Rupp         j        = *uj_ptr;
28262205254eSKarl Rupp         jl[prow] = jl[j];
28272205254eSKarl Rupp         jl[j]    = prow;
28280be760fbSHong Zhang       }
28290be760fbSHong Zhang       prow = nextprow;
28300be760fbSHong Zhang     }
28310be760fbSHong Zhang 
28320be760fbSHong Zhang     /* if free space is not available, make more free space */
28330be760fbSHong Zhang     if (current_space->local_remaining<nzk) {
28340be760fbSHong Zhang       i    = am - k + 1; /* num of unfactored rows */
2835f91af8c7SBarry Smith       i    = PetscIntMultTruncate(i,PetscMin(nzk,i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
28365f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscFreeSpaceGet(i,&current_space));
28370be760fbSHong Zhang       reallocs++;
28380be760fbSHong Zhang     }
28390be760fbSHong Zhang 
28400be760fbSHong Zhang     /* copy data into free space, then initialize lnk */
28415f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt));
28420be760fbSHong Zhang 
28430be760fbSHong Zhang     /* add the k-th row into il and jl */
28447b056e98SHong Zhang     if (nzk > 1) {
28450be760fbSHong Zhang       i     = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
28460be760fbSHong Zhang       jl[k] = jl[i]; jl[i] = k;
28470be760fbSHong Zhang       il[k] = ui[k] + 1;
28480be760fbSHong Zhang     }
28490be760fbSHong Zhang     ui_ptr[k] = current_space->array;
28502205254eSKarl Rupp 
28510be760fbSHong Zhang     current_space->array           += nzk;
28520be760fbSHong Zhang     current_space->local_used      += nzk;
28530be760fbSHong Zhang     current_space->local_remaining -= nzk;
28540be760fbSHong Zhang 
28550be760fbSHong Zhang     ui[k+1] = ui[k] + nzk;
28560be760fbSHong Zhang   }
28570be760fbSHong Zhang 
28585f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(perm,&rip));
28595f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(iperm,&riip));
28605f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree4(ui_ptr,jl,il,cols));
28610be760fbSHong Zhang 
28629263d837SHong Zhang   /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */
28635f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(ui[am]+1,&uj));
28645f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag)); /* store matrix factor */
28655f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLLDestroy(lnk,lnkbt));
28660be760fbSHong Zhang 
28670be760fbSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
28680be760fbSHong Zhang 
2869f284f12aSHong Zhang   b               = (Mat_SeqSBAIJ*)fact->data;
28700be760fbSHong Zhang   b->singlemalloc = PETSC_FALSE;
28710be760fbSHong Zhang   b->free_a       = PETSC_TRUE;
28720be760fbSHong Zhang   b->free_ij      = PETSC_TRUE;
28732205254eSKarl Rupp 
28745f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(ui[am]+1,&b->a));
28752205254eSKarl Rupp 
28760be760fbSHong Zhang   b->j         = uj;
28770be760fbSHong Zhang   b->i         = ui;
28780be760fbSHong Zhang   b->diag      = udiag;
28790be760fbSHong Zhang   b->free_diag = PETSC_TRUE;
2880f4259b30SLisandro Dalcin   b->ilen      = NULL;
2881f4259b30SLisandro Dalcin   b->imax      = NULL;
28820be760fbSHong Zhang   b->row       = perm;
28830be760fbSHong Zhang   b->col       = perm;
288426fbe8dcSKarl Rupp 
28855f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectReference((PetscObject)perm));
28865f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectReference((PetscObject)perm));
28872205254eSKarl Rupp 
28880be760fbSHong Zhang   b->icol          = iperm;
28890be760fbSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
289026fbe8dcSKarl Rupp 
28915f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(am+1,&b->solve_work));
28925f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogObjectMemory((PetscObject)fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar))));
28932205254eSKarl Rupp 
28940be760fbSHong Zhang   b->maxnz = b->nz = ui[am];
28950be760fbSHong Zhang 
2896f284f12aSHong Zhang   fact->info.factor_mallocs   = reallocs;
2897f284f12aSHong Zhang   fact->info.fill_ratio_given = fill;
28980be760fbSHong Zhang   if (ai[am] != 0) {
28996a69fef8SHong Zhang     /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */
290095b5ac22SHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am);
29010be760fbSHong Zhang   } else {
2902f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
29030be760fbSHong Zhang   }
29049263d837SHong Zhang #if defined(PETSC_USE_INFO)
29059263d837SHong Zhang   if (ai[am] != 0) {
29069263d837SHong Zhang     PetscReal af = fact->info.fill_ratio_needed;
29075f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"Reallocs %" PetscInt_FMT " Fill ratio:given %g needed %g\n",reallocs,(double)fill,(double)af));
29085f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"Run with -pc_factor_fill %g or use \n",(double)af));
29095f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"PCFactorSetFill(pc,%g) for best performance.\n",(double)af));
29109263d837SHong Zhang   } else {
29115f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"Empty matrix\n"));
29129263d837SHong Zhang   }
29139263d837SHong Zhang #endif
291435233d99SShri Abhyankar   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
29150be760fbSHong Zhang   PetscFunctionReturn(0);
29160be760fbSHong Zhang }
29170be760fbSHong Zhang 
2918ad04f41aSHong Zhang PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2919f76d2b81SHong Zhang {
2920f76d2b81SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
292110c27e3dSHong Zhang   Mat_SeqSBAIJ       *b;
29229186b105SHong Zhang   PetscBool          perm_identity,missing;
292310c27e3dSHong Zhang   PetscReal          fill = info->fill;
29245d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
29255d0c19d7SBarry Smith   PetscInt           i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
292610c27e3dSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
29272ed133dbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr;
29280298fd71SBarry Smith   PetscFreeSpaceList free_space=NULL,current_space=NULL;
292910c27e3dSHong Zhang   PetscBT            lnkbt;
29302ed133dbSHong Zhang   IS                 iperm;
2931f76d2b81SHong Zhang 
2932f76d2b81SHong Zhang   PetscFunctionBegin;
29332c71b3e2SJacob Faibussowitsch   PetscCheckFalse(A->rmap->n != A->cmap->n,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %" PetscInt_FMT " columns %" PetscInt_FMT,A->rmap->n,A->cmap->n);
29345f80ce2aSJacob Faibussowitsch   CHKERRQ(MatMissingDiagonal(A,&missing,&i));
2935*28b400f6SJacob Faibussowitsch   PetscCheck(!missing,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %" PetscInt_FMT,i);
29369186b105SHong Zhang 
29372ed133dbSHong Zhang   /* check whether perm is the identity mapping */
29385f80ce2aSJacob Faibussowitsch   CHKERRQ(ISIdentity(perm,&perm_identity));
29395f80ce2aSJacob Faibussowitsch   CHKERRQ(ISInvertPermutation(perm,PETSC_DECIDE,&iperm));
29405f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(iperm,&riip));
29415f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(perm,&rip));
294210c27e3dSHong Zhang 
294310c27e3dSHong Zhang   /* initialization */
29445f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(am+1,&ui));
294510c27e3dSHong Zhang   ui[0] = 0;
294610c27e3dSHong Zhang 
294710c27e3dSHong Zhang   /* jl: linked list for storing indices of the pivot rows
29481393bc97SHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
29495f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc4(am,&ui_ptr,am,&jl,am,&il,am,&cols));
29501393bc97SHong Zhang   for (i=0; i<am; i++) {
29511393bc97SHong Zhang     jl[i] = am; il[i] = 0;
2952f76d2b81SHong Zhang   }
295310c27e3dSHong Zhang 
295410c27e3dSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
29551393bc97SHong Zhang   nlnk = am + 1;
29565f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLLCreate(am,am,nlnk,lnk,lnkbt));
295710c27e3dSHong Zhang 
29581393bc97SHong Zhang   /* initial FreeSpace size is fill*(ai[am]+1) */
29595f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,ai[am]+1),&free_space));
296010c27e3dSHong Zhang   current_space = free_space;
296110c27e3dSHong Zhang 
29621393bc97SHong Zhang   for (k=0; k<am; k++) {  /* for each active row k */
296310c27e3dSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
296410c27e3dSHong Zhang     nzk   = 0;
29652ed133dbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
296694bad497SJacob Faibussowitsch     PetscCheck(ncols,PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %" PetscInt_FMT " in permuted ordering %" PetscInt_FMT,rip[k],k);
29672ed133dbSHong Zhang     ncols_upper = 0;
29682ed133dbSHong Zhang     for (j=0; j<ncols; j++) {
29699bfd6278SHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
29702ed133dbSHong Zhang       if (i >= k) { /* only take upper triangular entry */
29712ed133dbSHong Zhang         cols[ncols_upper] = i;
29722ed133dbSHong Zhang         ncols_upper++;
29732ed133dbSHong Zhang       }
29742ed133dbSHong Zhang     }
297594bad497SJacob Faibussowitsch     CHKERRQ(PetscLLAdd(ncols_upper,cols,am,&nlnk,lnk,lnkbt));
297610c27e3dSHong Zhang     nzk += nlnk;
297710c27e3dSHong Zhang 
297810c27e3dSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
297910c27e3dSHong Zhang     prow = jl[k]; /* 1st pivot row */
298010c27e3dSHong Zhang 
298110c27e3dSHong Zhang     while (prow < k) {
298210c27e3dSHong Zhang       nextprow = jl[prow];
298310c27e3dSHong Zhang       /* merge prow into k-th row */
29841393bc97SHong Zhang       jmin   = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */
298510c27e3dSHong Zhang       jmax   = ui[prow+1];
298610c27e3dSHong Zhang       ncols  = jmax-jmin;
29871393bc97SHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
298894bad497SJacob Faibussowitsch       CHKERRQ(PetscLLAddSorted(ncols,uj_ptr,am,&nlnk,lnk,lnkbt));
298910c27e3dSHong Zhang       nzk   += nlnk;
299010c27e3dSHong Zhang 
299110c27e3dSHong Zhang       /* update il and jl for prow */
299210c27e3dSHong Zhang       if (jmin < jmax) {
299310c27e3dSHong Zhang         il[prow] = jmin;
29942ed133dbSHong Zhang         j        = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow;
299510c27e3dSHong Zhang       }
299610c27e3dSHong Zhang       prow = nextprow;
299710c27e3dSHong Zhang     }
299810c27e3dSHong Zhang 
299910c27e3dSHong Zhang     /* if free space is not available, make more free space */
300010c27e3dSHong Zhang     if (current_space->local_remaining<nzk) {
30011393bc97SHong Zhang       i    = am - k + 1; /* num of unfactored rows */
300210c27e3dSHong Zhang       i    = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
30035f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscFreeSpaceGet(i,&current_space));
300410c27e3dSHong Zhang       reallocs++;
300510c27e3dSHong Zhang     }
300610c27e3dSHong Zhang 
300710c27e3dSHong Zhang     /* copy data into free space, then initialize lnk */
30085f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt));
300910c27e3dSHong Zhang 
301010c27e3dSHong Zhang     /* add the k-th row into il and jl */
301110c27e3dSHong Zhang     if (nzk-1 > 0) {
30121393bc97SHong Zhang       i     = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
301310c27e3dSHong Zhang       jl[k] = jl[i]; jl[i] = k;
301410c27e3dSHong Zhang       il[k] = ui[k] + 1;
301510c27e3dSHong Zhang     }
30162ed133dbSHong Zhang     ui_ptr[k] = current_space->array;
30172205254eSKarl Rupp 
301810c27e3dSHong Zhang     current_space->array           += nzk;
301910c27e3dSHong Zhang     current_space->local_used      += nzk;
302010c27e3dSHong Zhang     current_space->local_remaining -= nzk;
302110c27e3dSHong Zhang 
302210c27e3dSHong Zhang     ui[k+1] = ui[k] + nzk;
302310c27e3dSHong Zhang   }
302410c27e3dSHong Zhang 
30256cf91177SBarry Smith #if defined(PETSC_USE_INFO)
30261393bc97SHong Zhang   if (ai[am] != 0) {
30271393bc97SHong Zhang     PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]);
30285f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"Reallocs %" PetscInt_FMT " Fill ratio:given %g needed %g\n",reallocs,(double)fill,(double)af));
30295f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"Run with -pc_factor_fill %g or use \n",(double)af));
30305f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"PCFactorSetFill(pc,%g) for best performance.\n",(double)af));
303110c27e3dSHong Zhang   } else {
30325f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(A,"Empty matrix\n"));
303310c27e3dSHong Zhang   }
303463ba0a88SBarry Smith #endif
303510c27e3dSHong Zhang 
30365f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(perm,&rip));
30375f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(iperm,&riip));
30385f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree4(ui_ptr,jl,il,cols));
303910c27e3dSHong Zhang 
304010c27e3dSHong Zhang   /* destroy list of free space and other temporary array(s) */
30415f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(ui[am]+1,&uj));
30425f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFreeSpaceContiguous(&free_space,uj));
30435f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLLDestroy(lnk,lnkbt));
304410c27e3dSHong Zhang 
304510c27e3dSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
304610c27e3dSHong Zhang 
3047f284f12aSHong Zhang   b               = (Mat_SeqSBAIJ*)fact->data;
304810c27e3dSHong Zhang   b->singlemalloc = PETSC_FALSE;
3049e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
3050e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
30512205254eSKarl Rupp 
30525f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(ui[am]+1,&b->a));
30532205254eSKarl Rupp 
305410c27e3dSHong Zhang   b->j    = uj;
305510c27e3dSHong Zhang   b->i    = ui;
3056f4259b30SLisandro Dalcin   b->diag = NULL;
3057f4259b30SLisandro Dalcin   b->ilen = NULL;
3058f4259b30SLisandro Dalcin   b->imax = NULL;
305910c27e3dSHong Zhang   b->row  = perm;
30609bfd6278SHong Zhang   b->col  = perm;
30612205254eSKarl Rupp 
30625f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectReference((PetscObject)perm));
30635f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectReference((PetscObject)perm));
30642205254eSKarl Rupp 
30659bfd6278SHong Zhang   b->icol          = iperm;
306610c27e3dSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
30672205254eSKarl Rupp 
30685f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(am+1,&b->solve_work));
30695f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogObjectMemory((PetscObject)fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar))));
30701393bc97SHong Zhang   b->maxnz = b->nz = ui[am];
307110c27e3dSHong Zhang 
3072f284f12aSHong Zhang   fact->info.factor_mallocs   = reallocs;
3073f284f12aSHong Zhang   fact->info.fill_ratio_given = fill;
30741393bc97SHong Zhang   if (ai[am] != 0) {
3075f284f12aSHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
307610c27e3dSHong Zhang   } else {
3077f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
307810c27e3dSHong Zhang   }
30790a3c351aSHong Zhang   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace;
3080f76d2b81SHong Zhang   PetscFunctionReturn(0);
3081f76d2b81SHong Zhang }
3082599ef60dSHong Zhang 
308335233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx)
3084813a1f2bSShri Abhyankar {
3085813a1f2bSShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
3086813a1f2bSShri Abhyankar   PetscInt          n   = A->rmap->n;
3087813a1f2bSShri Abhyankar   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag,*vi;
3088813a1f2bSShri Abhyankar   PetscScalar       *x,sum;
3089813a1f2bSShri Abhyankar   const PetscScalar *b;
3090813a1f2bSShri Abhyankar   const MatScalar   *aa = a->a,*v;
3091813a1f2bSShri Abhyankar   PetscInt          i,nz;
3092813a1f2bSShri Abhyankar 
3093813a1f2bSShri Abhyankar   PetscFunctionBegin;
3094813a1f2bSShri Abhyankar   if (!n) PetscFunctionReturn(0);
3095813a1f2bSShri Abhyankar 
30965f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayRead(bb,&b));
30975f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayWrite(xx,&x));
3098813a1f2bSShri Abhyankar 
3099813a1f2bSShri Abhyankar   /* forward solve the lower triangular */
3100813a1f2bSShri Abhyankar   x[0] = b[0];
3101813a1f2bSShri Abhyankar   v    = aa;
3102813a1f2bSShri Abhyankar   vi   = aj;
3103813a1f2bSShri Abhyankar   for (i=1; i<n; i++) {
3104813a1f2bSShri Abhyankar     nz  = ai[i+1] - ai[i];
3105813a1f2bSShri Abhyankar     sum = b[i];
3106813a1f2bSShri Abhyankar     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
3107813a1f2bSShri Abhyankar     v   += nz;
3108813a1f2bSShri Abhyankar     vi  += nz;
3109813a1f2bSShri Abhyankar     x[i] = sum;
3110813a1f2bSShri Abhyankar   }
3111813a1f2bSShri Abhyankar 
3112813a1f2bSShri Abhyankar   /* backward solve the upper triangular */
311362fbd6afSShri Abhyankar   for (i=n-1; i>=0; i--) {
31143c0119dfSShri Abhyankar     v   = aa + adiag[i+1] + 1;
31153c0119dfSShri Abhyankar     vi  = aj + adiag[i+1] + 1;
3116813a1f2bSShri Abhyankar     nz  = adiag[i] - adiag[i+1]-1;
3117813a1f2bSShri Abhyankar     sum = x[i];
3118813a1f2bSShri Abhyankar     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
31193c0119dfSShri Abhyankar     x[i] = sum*v[nz]; /* x[i]=aa[adiag[i]]*sum; v++; */
3120813a1f2bSShri Abhyankar   }
3121813a1f2bSShri Abhyankar 
31225f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(2.0*a->nz - A->cmap->n));
31235f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayRead(bb,&b));
31245f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayWrite(xx,&x));
3125813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
3126813a1f2bSShri Abhyankar }
3127813a1f2bSShri Abhyankar 
312835233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx)
3129f268cba8SShri Abhyankar {
3130f268cba8SShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
3131f268cba8SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
3132f268cba8SShri Abhyankar   PetscInt          i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,*adiag = a->diag,nz;
3133f268cba8SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c;
3134f268cba8SShri Abhyankar   PetscScalar       *x,*tmp,sum;
3135f268cba8SShri Abhyankar   const PetscScalar *b;
3136f268cba8SShri Abhyankar   const MatScalar   *aa = a->a,*v;
3137f268cba8SShri Abhyankar 
3138f268cba8SShri Abhyankar   PetscFunctionBegin;
3139f268cba8SShri Abhyankar   if (!n) PetscFunctionReturn(0);
3140f268cba8SShri Abhyankar 
31415f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayRead(bb,&b));
31425f80ce2aSJacob Faibussowitsch   CHKERRQ(VecGetArrayWrite(xx,&x));
3143f268cba8SShri Abhyankar   tmp  = a->solve_work;
3144f268cba8SShri Abhyankar 
31455f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isrow,&rout)); r = rout;
31465f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(iscol,&cout)); c = cout;
3147f268cba8SShri Abhyankar 
3148f268cba8SShri Abhyankar   /* forward solve the lower triangular */
3149f268cba8SShri Abhyankar   tmp[0] = b[r[0]];
3150f268cba8SShri Abhyankar   v      = aa;
3151f268cba8SShri Abhyankar   vi     = aj;
3152f268cba8SShri Abhyankar   for (i=1; i<n; i++) {
3153f268cba8SShri Abhyankar     nz  = ai[i+1] - ai[i];
3154f268cba8SShri Abhyankar     sum = b[r[i]];
3155f268cba8SShri Abhyankar     PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
3156f268cba8SShri Abhyankar     tmp[i] = sum;
3157f268cba8SShri Abhyankar     v     += nz; vi += nz;
3158f268cba8SShri Abhyankar   }
3159f268cba8SShri Abhyankar 
3160f268cba8SShri Abhyankar   /* backward solve the upper triangular */
3161f268cba8SShri Abhyankar   for (i=n-1; i>=0; i--) {
31623c0119dfSShri Abhyankar     v   = aa + adiag[i+1]+1;
31633c0119dfSShri Abhyankar     vi  = aj + adiag[i+1]+1;
3164f268cba8SShri Abhyankar     nz  = adiag[i]-adiag[i+1]-1;
3165f268cba8SShri Abhyankar     sum = tmp[i];
3166f268cba8SShri Abhyankar     PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
3167f268cba8SShri Abhyankar     x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */
3168f268cba8SShri Abhyankar   }
3169f268cba8SShri Abhyankar 
31705f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isrow,&rout));
31715f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(iscol,&cout));
31725f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayRead(bb,&b));
31735f80ce2aSJacob Faibussowitsch   CHKERRQ(VecRestoreArrayWrite(xx,&x));
31745f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(2.0*a->nz - A->cmap->n));
3175f268cba8SShri Abhyankar   PetscFunctionReturn(0);
3176f268cba8SShri Abhyankar }
3177f268cba8SShri Abhyankar 
3178fe97e370SBarry Smith /*
3179bebe2cf6SSatish 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
3180fe97e370SBarry Smith */
31811d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact)
31821d098868SHong Zhang {
31831d098868SHong Zhang   Mat            B = *fact;
3184599ef60dSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*b;
3185599ef60dSHong Zhang   IS             isicol;
31861d098868SHong Zhang   const PetscInt *r,*ic;
31871d098868SHong Zhang   PetscInt       i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag;
3188dc3a2fd3SHong Zhang   PetscInt       *bi,*bj,*bdiag,*bdiag_rev;
3189f61a948aSLisandro Dalcin   PetscInt       row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au;
31901d098868SHong Zhang   PetscInt       nlnk,*lnk;
31911d098868SHong Zhang   PetscBT        lnkbt;
3192ace3abfcSBarry Smith   PetscBool      row_identity,icol_identity;
31938aee2decSHong Zhang   MatScalar      *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp;
31941d098868SHong Zhang   const PetscInt *ics;
31951d098868SHong Zhang   PetscInt       j,nz,*pj,*bjtmp,k,ncut,*jtmp;
3196a2ea699eSBarry Smith   PetscReal      dt     =info->dt,shift=info->shiftamount;
3197f61a948aSLisandro Dalcin   PetscInt       dtcount=(PetscInt)info->dtcount,nnz_max;
3198ace3abfcSBarry Smith   PetscBool      missing;
3199599ef60dSHong Zhang 
3200599ef60dSHong Zhang   PetscFunctionBegin;
3201f61a948aSLisandro Dalcin   if (dt      == PETSC_DEFAULT) dt = 0.005;
3202f61a948aSLisandro Dalcin   if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax);
3203f61a948aSLisandro Dalcin 
32041d098868SHong Zhang   /* ------- symbolic factorization, can be reused ---------*/
32055f80ce2aSJacob Faibussowitsch   CHKERRQ(MatMissingDiagonal(A,&missing,&i));
3206*28b400f6SJacob Faibussowitsch   PetscCheck(!missing,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %" PetscInt_FMT,i);
32071d098868SHong Zhang   adiag=a->diag;
3208599ef60dSHong Zhang 
32095f80ce2aSJacob Faibussowitsch   CHKERRQ(ISInvertPermutation(iscol,PETSC_DECIDE,&isicol));
3210599ef60dSHong Zhang 
3211599ef60dSHong Zhang   /* bdiag is location of diagonal in factor */
32125f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(n+1,&bdiag));     /* becomes b->diag */
32135f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(n+1,&bdiag_rev)); /* temporary */
3214599ef60dSHong Zhang 
32151d098868SHong Zhang   /* allocate row pointers bi */
32165f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(2*n+2,&bi));
32171d098868SHong Zhang 
32181d098868SHong Zhang   /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */
3219393d3a68SHong Zhang   if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */
32201d098868SHong Zhang   nnz_max = ai[n]+2*n*dtcount+2;
32218fc3a347SHong Zhang 
32225f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(nnz_max+1,&bj));
32235f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(nnz_max+1,&ba));
3224599ef60dSHong Zhang 
3225599ef60dSHong Zhang   /* put together the new matrix */
32265f80ce2aSJacob Faibussowitsch   CHKERRQ(MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,NULL));
32275f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogObjectParent((PetscObject)B,(PetscObject)isicol));
3228f284f12aSHong Zhang   b    = (Mat_SeqAIJ*)B->data;
32292205254eSKarl Rupp 
3230599ef60dSHong Zhang   b->free_a       = PETSC_TRUE;
3231599ef60dSHong Zhang   b->free_ij      = PETSC_TRUE;
3232599ef60dSHong Zhang   b->singlemalloc = PETSC_FALSE;
32332205254eSKarl Rupp 
3234599ef60dSHong Zhang   b->a    = ba;
3235599ef60dSHong Zhang   b->j    = bj;
3236599ef60dSHong Zhang   b->i    = bi;
3237599ef60dSHong Zhang   b->diag = bdiag;
3238f4259b30SLisandro Dalcin   b->ilen = NULL;
3239f4259b30SLisandro Dalcin   b->imax = NULL;
3240599ef60dSHong Zhang   b->row  = isrow;
3241599ef60dSHong Zhang   b->col  = iscol;
32425f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectReference((PetscObject)isrow));
32435f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscObjectReference((PetscObject)iscol));
3244599ef60dSHong Zhang   b->icol = isicol;
3245599ef60dSHong Zhang 
32465f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(n+1,&b->solve_work));
32475f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogObjectMemory((PetscObject)B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar))));
32481d098868SHong Zhang   b->maxnz = nnz_max;
3249599ef60dSHong Zhang 
3250d5f3da31SBarry Smith   B->factortype            = MAT_FACTOR_ILUDT;
3251f284f12aSHong Zhang   B->info.factor_mallocs   = 0;
3252f284f12aSHong Zhang   B->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]);
32531d098868SHong Zhang   /* ------- end of symbolic factorization ---------*/
3254599ef60dSHong Zhang 
32555f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isrow,&r));
32565f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isicol,&ic));
3257599ef60dSHong Zhang   ics  = ic;
3258599ef60dSHong Zhang 
3259599ef60dSHong Zhang   /* linked list for storing column indices of the active row */
3260599ef60dSHong Zhang   nlnk = n + 1;
32615f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLLCreate(n,n,nlnk,lnk,lnkbt));
32621d098868SHong Zhang 
32631d098868SHong Zhang   /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */
32645f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc2(n,&im,n,&jtmp));
32651d098868SHong Zhang   /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */
32665f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc2(n,&rtmp,n,&vtmp));
32675f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscArrayzero(rtmp,n));
3268599ef60dSHong Zhang 
3269599ef60dSHong Zhang   bi[0]        = 0;
32708fc3a347SHong Zhang   bdiag[0]     = nnz_max-1; /* location of diag[0] in factor B */
3271dc3a2fd3SHong Zhang   bdiag_rev[n] = bdiag[0];
32728fc3a347SHong Zhang   bi[2*n+1]    = bdiag[0]+1; /* endof bj and ba array */
3273599ef60dSHong Zhang   for (i=0; i<n; i++) {
3274599ef60dSHong Zhang     /* copy initial fill into linked list */
32758369b28dSHong Zhang     nzi = ai[r[i]+1] - ai[r[i]];
327694bad497SJacob Faibussowitsch     PetscCheck(nzi,PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %" PetscInt_FMT " in permuted ordering %" PetscInt_FMT,r[i],i);
32778369b28dSHong Zhang     nzi_al = adiag[r[i]] - ai[r[i]];
32788369b28dSHong Zhang     nzi_au = ai[r[i]+1] - adiag[r[i]] -1;
3279599ef60dSHong Zhang     ajtmp  = aj + ai[r[i]];
328094bad497SJacob Faibussowitsch     CHKERRQ(PetscLLAddPerm(nzi,ajtmp,ic,n,&nlnk,lnk,lnkbt));
3281599ef60dSHong Zhang 
3282599ef60dSHong Zhang     /* load in initial (unfactored row) */
32831d098868SHong Zhang     aatmp = a->a + ai[r[i]];
32848369b28dSHong Zhang     for (j=0; j<nzi; j++) {
32851d098868SHong Zhang       rtmp[ics[*ajtmp++]] = *aatmp++;
3286599ef60dSHong Zhang     }
3287599ef60dSHong Zhang 
3288599ef60dSHong Zhang     /* add pivot rows into linked list */
3289599ef60dSHong Zhang     row = lnk[n];
3290599ef60dSHong Zhang     while (row < i) {
32911d098868SHong Zhang       nzi_bl = bi[row+1] - bi[row] + 1;
32921d098868SHong Zhang       bjtmp  = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */
329394bad497SJacob Faibussowitsch       CHKERRQ(PetscLLAddSortedLU(bjtmp,row,&nlnk,lnk,lnkbt,i,nzi_bl,im));
3294599ef60dSHong Zhang       nzi   += nlnk;
3295599ef60dSHong Zhang       row    = lnk[row];
3296599ef60dSHong Zhang     }
3297599ef60dSHong Zhang 
32988369b28dSHong Zhang     /* copy data from lnk into jtmp, then initialize lnk */
32995f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt));
3300599ef60dSHong Zhang 
3301599ef60dSHong Zhang     /* numerical factorization */
33028369b28dSHong Zhang     bjtmp = jtmp;
3303599ef60dSHong Zhang     row   = *bjtmp++; /* 1st pivot row */
3304599ef60dSHong Zhang     while (row < i) {
3305599ef60dSHong Zhang       pc         = rtmp + row;
33063c2a7987SHong Zhang       pv         = ba + bdiag[row]; /* 1./(diag of the pivot row) */
33073c2a7987SHong Zhang       multiplier = (*pc) * (*pv);
3308599ef60dSHong Zhang       *pc        = multiplier;
33093c2a7987SHong Zhang       if (PetscAbsScalar(*pc) > dt) { /* apply tolerance dropping rule */
33101d098868SHong Zhang         pj = bj + bdiag[row+1] + 1;         /* point to 1st entry of U(row,:) */
33111d098868SHong Zhang         pv = ba + bdiag[row+1] + 1;
33121d098868SHong Zhang         nz = bdiag[row] - bdiag[row+1] - 1;         /* num of entries in U(row,:), excluding diagonal */
33131d098868SHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
33145f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscLogFlops(1+2.0*nz));
3315599ef60dSHong Zhang       }
3316599ef60dSHong Zhang       row = *bjtmp++;
3317599ef60dSHong Zhang     }
3318599ef60dSHong Zhang 
33198369b28dSHong Zhang     /* copy sparse rtmp into contiguous vtmp; separate L and U part */
33208aee2decSHong Zhang     diag_tmp = rtmp[i];  /* save diagonal value - may not needed?? */
33218369b28dSHong Zhang     nzi_bl   = 0; j = 0;
33228369b28dSHong Zhang     while (jtmp[j] < i) { /* Note: jtmp is sorted */
33238aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
33248369b28dSHong Zhang       nzi_bl++; j++;
33258369b28dSHong Zhang     }
33268369b28dSHong Zhang     nzi_bu = nzi - nzi_bl -1;
33278369b28dSHong Zhang     while (j < nzi) {
33288aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
33298369b28dSHong Zhang       j++;
33308369b28dSHong Zhang     }
33318369b28dSHong Zhang 
3332599ef60dSHong Zhang     bjtmp = bj + bi[i];
3333599ef60dSHong Zhang     batmp = ba + bi[i];
33348369b28dSHong Zhang     /* apply level dropping rule to L part */
33358369b28dSHong Zhang     ncut = nzi_al + dtcount;
33368369b28dSHong Zhang     if (ncut < nzi_bl) {
33375f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscSortSplit(ncut,nzi_bl,vtmp,jtmp));
33385f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscSortIntWithScalarArray(ncut,jtmp,vtmp));
3339599ef60dSHong Zhang     } else {
33408369b28dSHong Zhang       ncut = nzi_bl;
33418369b28dSHong Zhang     }
33428369b28dSHong Zhang     for (j=0; j<ncut; j++) {
33438369b28dSHong Zhang       bjtmp[j] = jtmp[j];
33448369b28dSHong Zhang       batmp[j] = vtmp[j];
33458369b28dSHong Zhang     }
33461d098868SHong Zhang     bi[i+1] = bi[i] + ncut;
33478369b28dSHong Zhang     nzi     = ncut + 1;
33488369b28dSHong Zhang 
33498369b28dSHong Zhang     /* apply level dropping rule to U part */
33508369b28dSHong Zhang     ncut = nzi_au + dtcount;
33518369b28dSHong Zhang     if (ncut < nzi_bu) {
33525f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1));
33535f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1));
33548369b28dSHong Zhang     } else {
33558369b28dSHong Zhang       ncut = nzi_bu;
33568369b28dSHong Zhang     }
33578369b28dSHong Zhang     nzi += ncut;
33581d098868SHong Zhang 
33591d098868SHong Zhang     /* mark bdiagonal */
33601d098868SHong Zhang     bdiag[i+1]       = bdiag[i] - (ncut + 1);
3361dc3a2fd3SHong Zhang     bdiag_rev[n-i-1] = bdiag[i+1];
33628fc3a347SHong Zhang     bi[2*n - i]      = bi[2*n - i +1] - (ncut + 1);
33631d098868SHong Zhang     bjtmp            = bj + bdiag[i];
33641d098868SHong Zhang     batmp            = ba + bdiag[i];
33651d098868SHong Zhang     *bjtmp           = i;
33668aee2decSHong Zhang     *batmp           = diag_tmp; /* rtmp[i]; */
3367c9c72f8fSHong Zhang     if (*batmp == 0.0) {
3368c9c72f8fSHong Zhang       *batmp = dt+shift;
3369c9c72f8fSHong Zhang     }
3370a5b23f4aSJose E. Roman     *batmp = 1.0/(*batmp); /* invert diagonal entries for simpler triangular solves */
33711d098868SHong Zhang 
33721d098868SHong Zhang     bjtmp = bj + bdiag[i+1]+1;
33731d098868SHong Zhang     batmp = ba + bdiag[i+1]+1;
33748369b28dSHong Zhang     for (k=0; k<ncut; k++) {
33751d098868SHong Zhang       bjtmp[k] = jtmp[nzi_bl+1+k];
33761d098868SHong Zhang       batmp[k] = vtmp[nzi_bl+1+k];
3377599ef60dSHong Zhang     }
3378599ef60dSHong Zhang 
3379599ef60dSHong Zhang     im[i] = nzi;   /* used by PetscLLAddSortedLU() */
33808369b28dSHong Zhang   } /* for (i=0; i<n; i++) */
33812c71b3e2SJacob Faibussowitsch   PetscCheckFalse(bi[n] >= bdiag[n],PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"end of L array %d cannot >= the beginning of U array %d",bi[n],bdiag[n]);
3382599ef60dSHong Zhang 
33835f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isrow,&r));
33845f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isicol,&ic));
3385599ef60dSHong Zhang 
33865f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLLDestroy(lnk,lnkbt));
33875f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree2(im,jtmp));
33885f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree2(rtmp,vtmp));
33895f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree(bdiag_rev));
3390599ef60dSHong Zhang 
33915f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(B->cmap->n));
33921d098868SHong Zhang   b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n];
3393599ef60dSHong Zhang 
33945f80ce2aSJacob Faibussowitsch   CHKERRQ(ISIdentity(isrow,&row_identity));
33955f80ce2aSJacob Faibussowitsch   CHKERRQ(ISIdentity(isicol,&icol_identity));
3396599ef60dSHong Zhang   if (row_identity && icol_identity) {
339735233d99SShri Abhyankar     B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
3398599ef60dSHong Zhang   } else {
339935233d99SShri Abhyankar     B->ops->solve = MatSolve_SeqAIJ;
3400599ef60dSHong Zhang   }
3401599ef60dSHong Zhang 
3402f4259b30SLisandro Dalcin   B->ops->solveadd          = NULL;
3403f4259b30SLisandro Dalcin   B->ops->solvetranspose    = NULL;
3404f4259b30SLisandro Dalcin   B->ops->solvetransposeadd = NULL;
3405f4259b30SLisandro Dalcin   B->ops->matsolve          = NULL;
3406599ef60dSHong Zhang   B->assembled              = PETSC_TRUE;
3407599ef60dSHong Zhang   B->preallocated           = PETSC_TRUE;
3408599ef60dSHong Zhang   PetscFunctionReturn(0);
3409599ef60dSHong Zhang }
3410599ef60dSHong Zhang 
34113c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */
3412fe97e370SBarry Smith /*
3413bebe2cf6SSatish 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
3414fe97e370SBarry Smith */
3415fe97e370SBarry Smith 
34167087cfbeSBarry Smith PetscErrorCode  MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info)
34173c2a7987SHong Zhang {
34183c2a7987SHong Zhang   PetscFunctionBegin;
34195f80ce2aSJacob Faibussowitsch   CHKERRQ(MatILUDTFactor_SeqAIJ(A,row,col,info,&fact));
34203c2a7987SHong Zhang   PetscFunctionReturn(0);
34213c2a7987SHong Zhang }
34223c2a7987SHong Zhang 
34233c2a7987SHong Zhang /*
34243c2a7987SHong Zhang    same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors
34253c2a7987SHong Zhang    - intend to replace existing MatLUFactorNumeric_SeqAIJ()
34263c2a7987SHong Zhang */
3427fe97e370SBarry Smith /*
3428bebe2cf6SSatish 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
3429fe97e370SBarry Smith */
3430fe97e370SBarry Smith 
34317087cfbeSBarry Smith PetscErrorCode  MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info)
34323c2a7987SHong Zhang {
34333c2a7987SHong Zhang   Mat            C     =fact;
34343c2a7987SHong Zhang   Mat_SeqAIJ     *a    =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data;
34353c2a7987SHong Zhang   IS             isrow = b->row,isicol = b->icol;
34363c2a7987SHong Zhang   const PetscInt *r,*ic,*ics;
3437b78a477dSHong Zhang   PetscInt       i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
3438b78a477dSHong Zhang   PetscInt       *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj;
34393c2a7987SHong Zhang   MatScalar      *rtmp,*pc,multiplier,*v,*pv,*aa=a->a;
3440182b8fbaSHong Zhang   PetscReal      dt=info->dt,shift=info->shiftamount;
3441ace3abfcSBarry Smith   PetscBool      row_identity, col_identity;
34423c2a7987SHong Zhang 
34433c2a7987SHong Zhang   PetscFunctionBegin;
34445f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isrow,&r));
34455f80ce2aSJacob Faibussowitsch   CHKERRQ(ISGetIndices(isicol,&ic));
34465f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(n+1,&rtmp));
34473c2a7987SHong Zhang   ics  = ic;
34483c2a7987SHong Zhang 
34493c2a7987SHong Zhang   for (i=0; i<n; i++) {
3450b78a477dSHong Zhang     /* initialize rtmp array */
3451b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i];       /* num of nozeros in L(i,:) */
34523c2a7987SHong Zhang     bjtmp = bj + bi[i];
3453b78a477dSHong Zhang     for  (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0;
3454b78a477dSHong Zhang     rtmp[i] = 0.0;
3455b78a477dSHong Zhang     nzu     = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */
3456b78a477dSHong Zhang     bjtmp   = bj + bdiag[i+1] + 1;
3457b78a477dSHong Zhang     for  (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0;
34583c2a7987SHong Zhang 
3459b78a477dSHong Zhang     /* load in initial unfactored row of A */
34603c2a7987SHong Zhang     nz    = ai[r[i]+1] - ai[r[i]];
34613c2a7987SHong Zhang     ajtmp = aj + ai[r[i]];
34623c2a7987SHong Zhang     v     = aa + ai[r[i]];
34633c2a7987SHong Zhang     for (j=0; j<nz; j++) {
3464b78a477dSHong Zhang       rtmp[ics[*ajtmp++]] = v[j];
34653c2a7987SHong Zhang     }
34663c2a7987SHong Zhang 
3467b78a477dSHong Zhang     /* numerical factorization */
3468b78a477dSHong Zhang     bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */
3469b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i]; /* num of entries in L(i,:) */
3470b78a477dSHong Zhang     k     = 0;
3471b78a477dSHong Zhang     while (k < nzl) {
34723c2a7987SHong Zhang       row        = *bjtmp++;
34733c2a7987SHong Zhang       pc         = rtmp + row;
3474b78a477dSHong Zhang       pv         = b->a + bdiag[row]; /* 1./(diag of the pivot row) */
3475b78a477dSHong Zhang       multiplier = (*pc) * (*pv);
34763c2a7987SHong Zhang       *pc        = multiplier;
3477b78a477dSHong Zhang       if (PetscAbsScalar(multiplier) > dt) {
3478b78a477dSHong Zhang         pj = bj + bdiag[row+1] + 1;         /* point to 1st entry of U(row,:) */
3479b78a477dSHong Zhang         pv = b->a + bdiag[row+1] + 1;
3480b78a477dSHong Zhang         nz = bdiag[row] - bdiag[row+1] - 1;         /* num of entries in U(row,:), excluding diagonal */
3481b78a477dSHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
34825f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscLogFlops(1+2.0*nz));
34833c2a7987SHong Zhang       }
3484b78a477dSHong Zhang       k++;
34853c2a7987SHong Zhang     }
34863c2a7987SHong Zhang 
3487b78a477dSHong Zhang     /* finished row so stick it into b->a */
3488b78a477dSHong Zhang     /* L-part */
3489b78a477dSHong Zhang     pv  = b->a + bi[i];
3490b78a477dSHong Zhang     pj  = bj + bi[i];
3491b78a477dSHong Zhang     nzl = bi[i+1] - bi[i];
3492b78a477dSHong Zhang     for (j=0; j<nzl; j++) {
3493b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
34943c2a7987SHong Zhang     }
3495b78a477dSHong Zhang 
3496a5b23f4aSJose E. Roman     /* diagonal: invert diagonal entries for simpler triangular solves */
3497b78a477dSHong Zhang     if (rtmp[i] == 0.0) rtmp[i] = dt+shift;
3498b78a477dSHong Zhang     b->a[bdiag[i]] = 1.0/rtmp[i];
3499b78a477dSHong Zhang 
3500b78a477dSHong Zhang     /* U-part */
3501b78a477dSHong Zhang     pv  = b->a + bdiag[i+1] + 1;
3502b78a477dSHong Zhang     pj  = bj + bdiag[i+1] + 1;
3503b78a477dSHong Zhang     nzu = bdiag[i] - bdiag[i+1] - 1;
3504b78a477dSHong Zhang     for (j=0; j<nzu; j++) {
3505b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
3506b78a477dSHong Zhang     }
3507b78a477dSHong Zhang   }
3508b78a477dSHong Zhang 
35095f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree(rtmp));
35105f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isicol,&ic));
35115f80ce2aSJacob Faibussowitsch   CHKERRQ(ISRestoreIndices(isrow,&r));
3512b78a477dSHong Zhang 
35135f80ce2aSJacob Faibussowitsch   CHKERRQ(ISIdentity(isrow,&row_identity));
35145f80ce2aSJacob Faibussowitsch   CHKERRQ(ISIdentity(isicol,&col_identity));
35153c2a7987SHong Zhang   if (row_identity && col_identity) {
351635233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
35173c2a7987SHong Zhang   } else {
351835233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ;
35193c2a7987SHong Zhang   }
3520f4259b30SLisandro Dalcin   C->ops->solveadd          = NULL;
3521f4259b30SLisandro Dalcin   C->ops->solvetranspose    = NULL;
3522f4259b30SLisandro Dalcin   C->ops->solvetransposeadd = NULL;
3523f4259b30SLisandro Dalcin   C->ops->matsolve          = NULL;
35243c2a7987SHong Zhang   C->assembled              = PETSC_TRUE;
35253c2a7987SHong Zhang   C->preallocated           = PETSC_TRUE;
35262205254eSKarl Rupp 
35275f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogFlops(C->cmap->n));
35283c2a7987SHong Zhang   PetscFunctionReturn(0);
35293c2a7987SHong Zhang }
3530