1be1d678aSKris Buschelman #define PETSCMAT_DLL 2289bc588SBarry Smith 3b377110cSBarry Smith 47c4f633dSBarry Smith #include "../src/mat/impls/aij/seq/aij.h" 5783ef271SHong Zhang #include "../src/mat/impls/sbaij/seq/sbaij.h" 61393bc97SHong Zhang #include "petscbt.h" 77c4f633dSBarry Smith #include "../src/mat/utils/freespace.h" 83b2fbd54SBarry Smith 9f6a26d55SMatthew Knepley EXTERN_C_BEGIN 104a2ae208SSatish Balay #undef __FUNCT__ 11a7a54a73SBarry Smith #define __FUNCT__ "MatGetOrdering_Flow_SeqAIJ" 128fa24674SBarry Smith /* 138fa24674SBarry Smith Computes an ordering to get most of the large numerical values in the lower triangular part of the matrix 14*312e372aSBarry Smith 15*312e372aSBarry Smith This code does not work and is not called anywhere. It would be registered with MatOrderingRegisterAll() 168fa24674SBarry Smith */ 17a7a54a73SBarry Smith PetscErrorCode MatGetOrdering_Flow_SeqAIJ(Mat mat,const MatOrderingType type,IS *irow,IS *icol) 183b2fbd54SBarry Smith { 198fa24674SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)mat->data; 208fa24674SBarry Smith PetscErrorCode ierr; 21889a8dedSBarry Smith PetscInt i,j,jj,k, kk,n = mat->rmap->n, current = 0, newcurrent = 0,*order; 228fa24674SBarry Smith const PetscInt *ai = a->i, *aj = a->j; 238fa24674SBarry Smith const PetscScalar *aa = a->a; 248fa24674SBarry Smith PetscTruth *done; 25889a8dedSBarry Smith PetscReal best,past = 0,future; 263a40ed3dSBarry Smith 278fa24674SBarry Smith PetscFunctionBegin; 288fa24674SBarry Smith /* pick initial row */ 298fa24674SBarry Smith best = -1; 308fa24674SBarry Smith for (i=0; i<n; i++) { 3175567043SBarry Smith future = 0.0; 328fa24674SBarry Smith for (j=ai[i]; j<ai[i+1]; j++) { 338fa24674SBarry Smith if (aj[j] != i) future += PetscAbsScalar(aa[j]); else past = PetscAbsScalar(aa[j]); 348fa24674SBarry Smith } 358fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 368fa24674SBarry Smith if (past/future > best) { 378fa24674SBarry Smith best = past/future; 388fa24674SBarry Smith current = i; 398fa24674SBarry Smith } 408fa24674SBarry Smith } 418fa24674SBarry Smith 428fa24674SBarry Smith ierr = PetscMalloc(n*sizeof(PetscTruth),&done);CHKERRQ(ierr); 438fa24674SBarry Smith ierr = PetscMemzero(done,n*sizeof(PetscTruth));CHKERRQ(ierr); 440e83c824SBarry Smith ierr = PetscMalloc(n*sizeof(PetscInt),&order);CHKERRQ(ierr); 458fa24674SBarry Smith order[0] = current; 468fa24674SBarry Smith for (i=0; i<n-1; i++) { 478fa24674SBarry Smith done[current] = PETSC_TRUE; 488fa24674SBarry Smith best = -1; 498fa24674SBarry Smith /* loop over all neighbors of current pivot */ 508fa24674SBarry Smith for (j=ai[current]; j<ai[current+1]; j++) { 518fa24674SBarry Smith jj = aj[j]; 528fa24674SBarry Smith if (done[jj]) continue; 538fa24674SBarry Smith /* loop over columns of potential next row computing weights for below and above diagonal */ 548fa24674SBarry Smith past = future = 0.0; 558fa24674SBarry Smith for (k=ai[jj]; k<ai[jj+1]; k++) { 568fa24674SBarry Smith kk = aj[k]; 578fa24674SBarry Smith if (done[kk]) past += PetscAbsScalar(aa[k]); 588fa24674SBarry Smith else if (kk != jj) future += PetscAbsScalar(aa[k]); 598fa24674SBarry Smith } 608fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 618fa24674SBarry Smith if (past/future > best) { 628fa24674SBarry Smith best = past/future; 638fa24674SBarry Smith newcurrent = jj; 648fa24674SBarry Smith } 658fa24674SBarry Smith } 668fa24674SBarry Smith if (best == -1) { /* no neighbors to select from so select best of all that remain */ 678fa24674SBarry Smith best = -1; 688fa24674SBarry Smith for (k=0; k<n; k++) { 698fa24674SBarry Smith if (done[k]) continue; 7075567043SBarry Smith future = 0.0; 7175567043SBarry Smith past = 0.0; 728fa24674SBarry Smith for (j=ai[k]; j<ai[k+1]; j++) { 738fa24674SBarry Smith kk = aj[j]; 748fa24674SBarry Smith if (done[kk]) past += PetscAbsScalar(aa[j]); 758fa24674SBarry Smith else if (kk != k) future += PetscAbsScalar(aa[j]); 768fa24674SBarry Smith } 778fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 788fa24674SBarry Smith if (past/future > best) { 798fa24674SBarry Smith best = past/future; 808fa24674SBarry Smith newcurrent = k; 818fa24674SBarry Smith } 828fa24674SBarry Smith } 838fa24674SBarry Smith } 84e32f2f54SBarry Smith if (current == newcurrent) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"newcurrent cannot be current"); 858fa24674SBarry Smith current = newcurrent; 868fa24674SBarry Smith order[i+1] = current; 878fa24674SBarry Smith } 888fa24674SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,n,order,irow);CHKERRQ(ierr); 898fa24674SBarry Smith *icol = *irow; 908fa24674SBarry Smith ierr = PetscObjectReference((PetscObject)*irow);CHKERRQ(ierr); 918fa24674SBarry Smith ierr = PetscFree(done);CHKERRQ(ierr); 928fa24674SBarry Smith ierr = PetscFree(order);CHKERRQ(ierr); 93d64ed03dSBarry Smith PetscFunctionReturn(0); 943b2fbd54SBarry Smith } 95f6a26d55SMatthew Knepley EXTERN_C_END 963b2fbd54SBarry Smith 97e631078cSBarry Smith EXTERN_C_BEGIN 984a2ae208SSatish Balay #undef __FUNCT__ 99db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqaij_petsc" 100db4efbfdSBarry Smith PetscErrorCode MatGetFactorAvailable_seqaij_petsc(Mat A,MatFactorType ftype,PetscTruth *flg) 101db4efbfdSBarry Smith { 102db4efbfdSBarry Smith PetscFunctionBegin; 103db4efbfdSBarry Smith *flg = PETSC_TRUE; 104db4efbfdSBarry Smith PetscFunctionReturn(0); 105db4efbfdSBarry Smith } 106db4efbfdSBarry Smith EXTERN_C_END 107db4efbfdSBarry Smith 108db4efbfdSBarry Smith EXTERN_C_BEGIN 109db4efbfdSBarry Smith #undef __FUNCT__ 110b24902e0SBarry Smith #define __FUNCT__ "MatGetFactor_seqaij_petsc" 111b24902e0SBarry Smith PetscErrorCode MatGetFactor_seqaij_petsc(Mat A,MatFactorType ftype,Mat *B) 112b24902e0SBarry Smith { 113d0f46423SBarry Smith PetscInt n = A->rmap->n; 114b24902e0SBarry Smith PetscErrorCode ierr; 115b24902e0SBarry Smith 116b24902e0SBarry Smith PetscFunctionBegin; 117b24902e0SBarry Smith ierr = MatCreate(((PetscObject)A)->comm,B);CHKERRQ(ierr); 118b24902e0SBarry Smith ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr); 119599ef60dSHong Zhang if (ftype == MAT_FACTOR_LU || ftype == MAT_FACTOR_ILU || ftype == MAT_FACTOR_ILUDT){ 120b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQAIJ);CHKERRQ(ierr); 12135233d99SShri Abhyankar (*B)->ops->ilufactorsymbolic = MatILUFactorSymbolic_SeqAIJ; 12235233d99SShri Abhyankar (*B)->ops->lufactorsymbolic = MatLUFactorSymbolic_SeqAIJ; 123b24902e0SBarry Smith } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 124b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr); 125b24902e0SBarry Smith ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 12635233d99SShri Abhyankar (*B)->ops->iccfactorsymbolic = MatICCFactorSymbolic_SeqAIJ; 12735233d99SShri Abhyankar (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ; 128e32f2f54SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported"); 129d5f3da31SBarry Smith (*B)->factortype = ftype; 130b24902e0SBarry Smith PetscFunctionReturn(0); 131b24902e0SBarry Smith } 132e631078cSBarry Smith EXTERN_C_END 133b24902e0SBarry Smith 134b24902e0SBarry Smith #undef __FUNCT__ 135ad04f41aSHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ_inplace" 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; 1406849ba73SBarry Smith PetscErrorCode ierr; 1415d0c19d7SBarry Smith const PetscInt *r,*ic; 1425d0c19d7SBarry Smith PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 1431393bc97SHong Zhang PetscInt *bi,*bj,*ajtmp; 1441393bc97SHong Zhang PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 1459e046878SBarry Smith PetscReal f; 1464875ba38SHong Zhang PetscInt nlnk,*lnk,k,**bi_ptr; 147a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1481393bc97SHong Zhang PetscBT lnkbt; 149289bc588SBarry Smith 1503a40ed3dSBarry Smith PetscFunctionBegin; 151e32f2f54SBarry Smith if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square"); 1524c49b128SBarry Smith ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 153ac121b3dSBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 154ac121b3dSBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 155289bc588SBarry Smith 156289bc588SBarry Smith /* get new row pointers */ 1571393bc97SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 1581393bc97SHong Zhang bi[0] = 0; 1591393bc97SHong Zhang 1601393bc97SHong Zhang /* bdiag is location of diagonal in factor */ 1611393bc97SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 1621393bc97SHong Zhang bdiag[0] = 0; 1631393bc97SHong Zhang 1641393bc97SHong Zhang /* linked list for storing column indices of the active row */ 1651393bc97SHong Zhang nlnk = n + 1; 1661393bc97SHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1671393bc97SHong Zhang 16835baf27eSSatish Balay ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 1691393bc97SHong Zhang 1701393bc97SHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 171d3d32019SBarry Smith f = info->fill; 172a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 1731393bc97SHong Zhang current_space = free_space; 174289bc588SBarry Smith 175289bc588SBarry Smith for (i=0; i<n; i++) { 1761393bc97SHong Zhang /* copy previous fill into linked list */ 177289bc588SBarry Smith nzi = 0; 1781393bc97SHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 179e32f2f54SBarry Smith if (!nnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 1801393bc97SHong Zhang ajtmp = aj + ai[r[i]]; 181afcc9904SHong Zhang ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1821393bc97SHong Zhang nzi += nlnk; 1831393bc97SHong Zhang 1841393bc97SHong Zhang /* add pivot rows into linked list */ 1851393bc97SHong Zhang row = lnk[n]; 1861393bc97SHong Zhang while (row < i) { 187d1170560SHong Zhang nzbd = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */ 188d1170560SHong Zhang ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 189d1170560SHong Zhang ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 1901393bc97SHong Zhang nzi += nlnk; 1911393bc97SHong Zhang row = lnk[row]; 192289bc588SBarry Smith } 1931393bc97SHong Zhang bi[i+1] = bi[i] + nzi; 1941393bc97SHong Zhang im[i] = nzi; 1951393bc97SHong Zhang 1961393bc97SHong Zhang /* mark bdiag */ 1971393bc97SHong Zhang nzbd = 0; 1981393bc97SHong Zhang nnz = nzi; 1991393bc97SHong Zhang k = lnk[n]; 2001393bc97SHong Zhang while (nnz-- && k < i){ 2011393bc97SHong Zhang nzbd++; 2021393bc97SHong Zhang k = lnk[k]; 2031393bc97SHong Zhang } 2041393bc97SHong Zhang bdiag[i] = bi[i] + nzbd; 2051393bc97SHong Zhang 2061393bc97SHong Zhang /* if free space is not available, make more free space */ 2071393bc97SHong Zhang if (current_space->local_remaining<nzi) { 2081393bc97SHong Zhang nnz = (n - i)*nzi; /* estimated and max additional space needed */ 209a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 2101393bc97SHong Zhang reallocs++; 2111393bc97SHong Zhang } 2121393bc97SHong Zhang 2131393bc97SHong Zhang /* copy data into free space, then initialize lnk */ 2141393bc97SHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 2151393bc97SHong Zhang bi_ptr[i] = current_space->array; 2161393bc97SHong Zhang current_space->array += nzi; 2171393bc97SHong Zhang current_space->local_used += nzi; 2181393bc97SHong Zhang current_space->local_remaining -= nzi; 219289bc588SBarry Smith } 2206cf91177SBarry Smith #if defined(PETSC_USE_INFO) 221464e8d44SSatish Balay if (ai[n] != 0) { 2221393bc97SHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 223ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 224ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 225ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 226ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 22705bf559cSSatish Balay } else { 228ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 22905bf559cSSatish Balay } 23063ba0a88SBarry Smith #endif 2312fb3ed76SBarry Smith 232898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 233898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 2341987afe7SBarry Smith 2351393bc97SHong Zhang /* destroy list of free space and other temporary array(s) */ 2361393bc97SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 237a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); 2381393bc97SHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 23935baf27eSSatish Balay ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 240289bc588SBarry Smith 241289bc588SBarry Smith /* put together the new matrix */ 242719d5645SBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 243719d5645SBarry Smith ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 244719d5645SBarry Smith b = (Mat_SeqAIJ*)(B)->data; 245e6b907acSBarry Smith b->free_a = PETSC_TRUE; 246e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 2477c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 2481393bc97SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 2491393bc97SHong Zhang b->j = bj; 2501393bc97SHong Zhang b->i = bi; 2511393bc97SHong Zhang b->diag = bdiag; 252416022c9SBarry Smith b->ilen = 0; 253416022c9SBarry Smith b->imax = 0; 254416022c9SBarry Smith b->row = isrow; 255416022c9SBarry Smith b->col = iscol; 256c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 257c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 25882bf6240SBarry Smith b->icol = isicol; 25987828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2601393bc97SHong Zhang 2611393bc97SHong Zhang /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 262719d5645SBarry Smith ierr = PetscLogObjectMemory(B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 2631393bc97SHong Zhang b->maxnz = b->nz = bi[n] ; 2647fd98bd6SLois Curfman McInnes 265d5f3da31SBarry Smith (B)->factortype = MAT_FACTOR_LU; 266719d5645SBarry Smith (B)->info.factor_mallocs = reallocs; 267719d5645SBarry Smith (B)->info.fill_ratio_given = f; 268703deb49SSatish Balay 2699f7d0b68SHong Zhang if (ai[n]) { 270719d5645SBarry Smith (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 271eea2913aSSatish Balay } else { 272719d5645SBarry Smith (B)->info.fill_ratio_needed = 0.0; 273eea2913aSSatish Balay } 274ad04f41aSHong Zhang (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace; 275f2344125SBarry Smith if (a->inode.size) { 276f2344125SBarry Smith (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace; 277f2344125SBarry Smith } 2783a40ed3dSBarry Smith PetscFunctionReturn(0); 279289bc588SBarry Smith } 2801393bc97SHong Zhang 281ce3d78c0SShri Abhyankar #undef __FUNCT__ 28235233d99SShri Abhyankar #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ" 28335233d99SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 284ce3d78c0SShri Abhyankar { 285ce3d78c0SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 286ce3d78c0SShri Abhyankar IS isicol; 287ce3d78c0SShri Abhyankar PetscErrorCode ierr; 288ce3d78c0SShri Abhyankar const PetscInt *r,*ic; 289ce3d78c0SShri Abhyankar PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 290ce3d78c0SShri Abhyankar PetscInt *bi,*bj,*ajtmp; 291ce3d78c0SShri Abhyankar PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 292ce3d78c0SShri Abhyankar PetscReal f; 293ce3d78c0SShri Abhyankar PetscInt nlnk,*lnk,k,**bi_ptr; 294ce3d78c0SShri Abhyankar PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 295ce3d78c0SShri Abhyankar PetscBT lnkbt; 2963898ab1dSSatish Balay PetscTruth olddatastruct=PETSC_FALSE; 297ce3d78c0SShri Abhyankar 298ce3d78c0SShri Abhyankar PetscFunctionBegin; 2993898ab1dSSatish Balay /* Uncomment the oldatastruct part only while testing new data structure for MatSolve() */ 300280e8513SShri Abhyankar ierr = PetscOptionsGetTruth(PETSC_NULL,"-lu_old",&olddatastruct,PETSC_NULL);CHKERRQ(ierr); 301628f99d7SShri Abhyankar if(olddatastruct){ 302628f99d7SShri Abhyankar ierr = MatLUFactorSymbolic_SeqAIJ_inplace(B,A,isrow,iscol,info);CHKERRQ(ierr); 303628f99d7SShri Abhyankar PetscFunctionReturn(0); 304628f99d7SShri Abhyankar } 305628f99d7SShri Abhyankar 306e32f2f54SBarry Smith if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square"); 307ce3d78c0SShri Abhyankar ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 308ce3d78c0SShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 309ce3d78c0SShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 310ce3d78c0SShri Abhyankar 3111bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 3121bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 3131bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 314a6df321fSShri Abhyankar bi[0] = bdiag[0] = 0; 3159b48462bSShri Abhyankar 3169b48462bSShri Abhyankar /* linked list for storing column indices of the active row */ 3179b48462bSShri Abhyankar nlnk = n + 1; 3189b48462bSShri Abhyankar ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3199b48462bSShri Abhyankar 3209b48462bSShri Abhyankar ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 3219b48462bSShri Abhyankar 3229b48462bSShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 3239b48462bSShri Abhyankar f = info->fill; 3249b48462bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 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]]; 331e32f2f54SBarry Smith if (!nnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 3329b48462bSShri Abhyankar ajtmp = aj + ai[r[i]]; 3339b48462bSShri Abhyankar ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3349b48462bSShri Abhyankar nzi += nlnk; 3359b48462bSShri Abhyankar 3369b48462bSShri Abhyankar /* add pivot rows into linked list */ 3379b48462bSShri Abhyankar row = lnk[n]; 3389b48462bSShri Abhyankar while (row < i){ 3399b48462bSShri Abhyankar nzbd = bdiag[row] + 1; /* num of entries in the row with column index <= row */ 3409b48462bSShri Abhyankar ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 3419b48462bSShri Abhyankar ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 3429b48462bSShri Abhyankar nzi += nlnk; 3439b48462bSShri Abhyankar row = lnk[row]; 3449b48462bSShri Abhyankar } 3459b48462bSShri Abhyankar bi[i+1] = bi[i] + nzi; 3469b48462bSShri Abhyankar im[i] = nzi; 3479b48462bSShri Abhyankar 3489b48462bSShri Abhyankar /* mark bdiag */ 3499b48462bSShri Abhyankar nzbd = 0; 3509b48462bSShri Abhyankar nnz = nzi; 3519b48462bSShri Abhyankar k = lnk[n]; 3529b48462bSShri Abhyankar while (nnz-- && k < i){ 3539b48462bSShri Abhyankar nzbd++; 3549b48462bSShri Abhyankar k = lnk[k]; 3559b48462bSShri Abhyankar } 3569b48462bSShri Abhyankar bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */ 3579b48462bSShri Abhyankar 3589b48462bSShri Abhyankar /* if free space is not available, make more free space */ 3599b48462bSShri Abhyankar if (current_space->local_remaining<nzi) { 3609b48462bSShri Abhyankar nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */ 3619b48462bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 3629b48462bSShri Abhyankar reallocs++; 3639b48462bSShri Abhyankar } 3649b48462bSShri Abhyankar 3659b48462bSShri Abhyankar /* copy data into free space, then initialize lnk */ 3669b48462bSShri Abhyankar ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 3679b48462bSShri Abhyankar bi_ptr[i] = current_space->array; 3689b48462bSShri Abhyankar current_space->array += nzi; 3699b48462bSShri Abhyankar current_space->local_used += nzi; 3709b48462bSShri Abhyankar current_space->local_remaining -= nzi; 3719b48462bSShri Abhyankar } 3729b48462bSShri Abhyankar #if defined(PETSC_USE_INFO) 3739b48462bSShri Abhyankar if (ai[n] != 0) { 37417454e89SShri Abhyankar PetscReal af = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 3759b48462bSShri Abhyankar ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 3769b48462bSShri Abhyankar ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 3779b48462bSShri Abhyankar ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 3789b48462bSShri Abhyankar ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 3799b48462bSShri Abhyankar } else { 3809b48462bSShri Abhyankar ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 3819b48462bSShri Abhyankar } 3829b48462bSShri Abhyankar #endif 3839b48462bSShri Abhyankar 3849b48462bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3859b48462bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3869b48462bSShri Abhyankar 3879b48462bSShri Abhyankar /* destroy list of free space and other temporary array(s) */ 3889b48462bSShri Abhyankar ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 3892ce24eb6SHong Zhang ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 3909b48462bSShri Abhyankar ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 3919b48462bSShri Abhyankar ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 3929b48462bSShri Abhyankar 3939b48462bSShri Abhyankar /* put together the new matrix */ 3949b48462bSShri Abhyankar ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 3959b48462bSShri Abhyankar ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 3969b48462bSShri Abhyankar b = (Mat_SeqAIJ*)(B)->data; 3979b48462bSShri Abhyankar b->free_a = PETSC_TRUE; 3989b48462bSShri Abhyankar b->free_ij = PETSC_TRUE; 3999b48462bSShri Abhyankar b->singlemalloc = PETSC_FALSE; 4009b48462bSShri Abhyankar ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 4019b48462bSShri Abhyankar b->j = bj; 4029b48462bSShri Abhyankar b->i = bi; 4039b48462bSShri Abhyankar b->diag = bdiag; 4049b48462bSShri Abhyankar b->ilen = 0; 4059b48462bSShri Abhyankar b->imax = 0; 4069b48462bSShri Abhyankar b->row = isrow; 4079b48462bSShri Abhyankar b->col = iscol; 4089b48462bSShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 4099b48462bSShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 4109b48462bSShri Abhyankar b->icol = isicol; 4119b48462bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 4129b48462bSShri Abhyankar 4139b48462bSShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 4149b48462bSShri Abhyankar ierr = PetscLogObjectMemory(B,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 4159b48462bSShri Abhyankar b->maxnz = b->nz = bdiag[0]+1; 416d5f3da31SBarry Smith B->factortype = MAT_FACTOR_LU; 417f284f12aSHong Zhang B->info.factor_mallocs = reallocs; 418f284f12aSHong Zhang B->info.fill_ratio_given = f; 4199b48462bSShri Abhyankar 4209b48462bSShri Abhyankar if (ai[n]) { 421f284f12aSHong Zhang B->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 4229b48462bSShri Abhyankar } else { 423f284f12aSHong Zhang B->info.fill_ratio_needed = 0.0; 4249b48462bSShri Abhyankar } 42535233d99SShri Abhyankar B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 426f2344125SBarry Smith if (a->inode.size) { 427d3ac4fa3SBarry Smith B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode; 428d3ac4fa3SBarry Smith } 4299b48462bSShri Abhyankar PetscFunctionReturn(0); 4309b48462bSShri Abhyankar } 4319b48462bSShri Abhyankar 4325b5bc046SBarry Smith /* 4335b5bc046SBarry Smith Trouble in factorization, should we dump the original matrix? 4345b5bc046SBarry Smith */ 4355b5bc046SBarry Smith #undef __FUNCT__ 4365b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix" 4375b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A) 4385b5bc046SBarry Smith { 4395b5bc046SBarry Smith PetscErrorCode ierr; 44090d69ab7SBarry Smith PetscTruth flg = PETSC_FALSE; 4415b5bc046SBarry Smith 4425b5bc046SBarry Smith PetscFunctionBegin; 44390d69ab7SBarry Smith ierr = PetscOptionsGetTruth(PETSC_NULL,"-mat_factor_dump_on_error",&flg,PETSC_NULL);CHKERRQ(ierr); 4445b5bc046SBarry Smith if (flg) { 4455b5bc046SBarry Smith PetscViewer viewer; 4465b5bc046SBarry Smith char filename[PETSC_MAX_PATH_LEN]; 4475b5bc046SBarry Smith 4485b5bc046SBarry Smith ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr); 4497adad957SLisandro Dalcin ierr = PetscViewerBinaryOpen(((PetscObject)A)->comm,filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr); 4505b5bc046SBarry Smith ierr = MatView(A,viewer);CHKERRQ(ierr); 4515b5bc046SBarry Smith ierr = PetscViewerDestroy(viewer);CHKERRQ(ierr); 4525b5bc046SBarry Smith } 4535b5bc046SBarry Smith PetscFunctionReturn(0); 4545b5bc046SBarry Smith } 4555b5bc046SBarry Smith 456c9c72f8fSHong Zhang #undef __FUNCT__ 45735233d99SShri Abhyankar #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ" 45835233d99SShri Abhyankar PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 459c9c72f8fSHong Zhang { 460c9c72f8fSHong Zhang Mat C=B; 461c9c72f8fSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 462c9c72f8fSHong Zhang IS isrow = b->row,isicol = b->icol; 463c9c72f8fSHong Zhang PetscErrorCode ierr; 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; 470c9c72f8fSHong Zhang PetscTruth 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 */ 478d90ac83dSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 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]; 489d4ad06f7SHong Zhang for (j=0; j<nz; j++) 490d4ad06f7SHong Zhang rs += PetscAbsScalar(v[j]); 491d4ad06f7SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 492d4ad06f7SHong Zhang } 493d4ad06f7SHong Zhang sctx.shift_top *= 1.1; 494d4ad06f7SHong Zhang sctx.nshift_max = 5; 495d4ad06f7SHong Zhang sctx.shift_lo = 0.; 496d4ad06f7SHong Zhang sctx.shift_hi = 1.; 497d4ad06f7SHong Zhang } 498d4ad06f7SHong Zhang 499c9c72f8fSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 500c9c72f8fSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 501c9c72f8fSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 502c9c72f8fSHong Zhang ics = ic; 503c9c72f8fSHong Zhang 504d4ad06f7SHong Zhang do { 50507b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 506c9c72f8fSHong Zhang for (i=0; i<n; i++){ 507c9c72f8fSHong Zhang /* zero rtmp */ 508c9c72f8fSHong Zhang /* L part */ 509c9c72f8fSHong Zhang nz = bi[i+1] - bi[i]; 510c9c72f8fSHong Zhang bjtmp = bj + bi[i]; 511c9c72f8fSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 512c9c72f8fSHong Zhang 513c9c72f8fSHong Zhang /* U part */ 51462fbd6afSShri Abhyankar nz = bdiag[i]-bdiag[i+1]; 51562fbd6afSShri Abhyankar bjtmp = bj + bdiag[i+1]+1; 516813a1f2bSShri Abhyankar for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 517813a1f2bSShri Abhyankar 518813a1f2bSShri Abhyankar /* load in initial (unfactored row) */ 519813a1f2bSShri Abhyankar nz = ai[r[i]+1] - ai[r[i]]; 520813a1f2bSShri Abhyankar ajtmp = aj + ai[r[i]]; 521813a1f2bSShri Abhyankar v = aa + ai[r[i]]; 522813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 523813a1f2bSShri Abhyankar rtmp[ics[ajtmp[j]]] = v[j]; 524813a1f2bSShri Abhyankar } 525813a1f2bSShri Abhyankar /* ZeropivotApply() */ 52698a93161SHong Zhang rtmp[i] += sctx.shift_amount; /* shift the diagonal of the matrix */ 527813a1f2bSShri Abhyankar 528813a1f2bSShri Abhyankar /* elimination */ 529813a1f2bSShri Abhyankar bjtmp = bj + bi[i]; 530813a1f2bSShri Abhyankar row = *bjtmp++; 531813a1f2bSShri Abhyankar nzL = bi[i+1] - bi[i]; 532f268cba8SShri Abhyankar for(k=0; k < nzL;k++) { 533813a1f2bSShri Abhyankar pc = rtmp + row; 534813a1f2bSShri Abhyankar if (*pc != 0.0) { 535813a1f2bSShri Abhyankar pv = b->a + bdiag[row]; 536813a1f2bSShri Abhyankar multiplier = *pc * (*pv); 537813a1f2bSShri Abhyankar *pc = multiplier; 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 */ 541813a1f2bSShri Abhyankar for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 542813a1f2bSShri Abhyankar ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 543813a1f2bSShri Abhyankar } 544f268cba8SShri Abhyankar row = *bjtmp++; 545813a1f2bSShri Abhyankar } 546813a1f2bSShri Abhyankar 547813a1f2bSShri Abhyankar /* finished row so stick it into b->a */ 548813a1f2bSShri Abhyankar rs = 0.0; 549813a1f2bSShri Abhyankar /* L part */ 550813a1f2bSShri Abhyankar pv = b->a + bi[i] ; 551813a1f2bSShri Abhyankar pj = b->j + bi[i] ; 552813a1f2bSShri Abhyankar nz = bi[i+1] - bi[i]; 553813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 554813a1f2bSShri Abhyankar pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 555813a1f2bSShri Abhyankar } 556813a1f2bSShri Abhyankar 557813a1f2bSShri Abhyankar /* U part */ 55862fbd6afSShri Abhyankar pv = b->a + bdiag[i+1]+1; 55962fbd6afSShri Abhyankar pj = b->j + bdiag[i+1]+1; 56062fbd6afSShri Abhyankar nz = bdiag[i] - bdiag[i+1]-1; 561813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 562813a1f2bSShri Abhyankar pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 563813a1f2bSShri Abhyankar } 564813a1f2bSShri Abhyankar 565813a1f2bSShri Abhyankar sctx.rs = rs; 566813a1f2bSShri Abhyankar sctx.pv = rtmp[i]; 56707b50cabSHong Zhang ierr = MatPivotCheck(info,&sctx,i);CHKERRQ(ierr); 56807b50cabSHong Zhang if(sctx.newshift) break; /* break for-loop */ 56907b50cabSHong Zhang rtmp[i] = sctx.pv; /* sctx.pv might be updated in the case of MAT_SHIFT_INBLOCKS */ 570813a1f2bSShri Abhyankar 571813a1f2bSShri Abhyankar /* Mark diagonal and invert diagonal for simplier triangular solves */ 572813a1f2bSShri Abhyankar pv = b->a + bdiag[i]; 573813a1f2bSShri Abhyankar *pv = 1.0/rtmp[i]; 574813a1f2bSShri Abhyankar 575813a1f2bSShri Abhyankar } /* endof for (i=0; i<n; i++){ */ 576813a1f2bSShri Abhyankar 5778ff23777SHong Zhang /* MatPivotRefine() */ 57807b50cabSHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max){ 579813a1f2bSShri Abhyankar /* 580813a1f2bSShri Abhyankar * if no shift in this attempt & shifting & started shifting & can refine, 581813a1f2bSShri Abhyankar * then try lower shift 582813a1f2bSShri Abhyankar */ 583813a1f2bSShri Abhyankar sctx.shift_hi = sctx.shift_fraction; 584813a1f2bSShri Abhyankar sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 585813a1f2bSShri Abhyankar sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 58607b50cabSHong Zhang sctx.newshift = PETSC_TRUE; 587813a1f2bSShri Abhyankar sctx.nshift++; 588813a1f2bSShri Abhyankar } 58907b50cabSHong Zhang } while (sctx.newshift); 590813a1f2bSShri Abhyankar 591813a1f2bSShri Abhyankar ierr = PetscFree(rtmp);CHKERRQ(ierr); 592813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 593813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 594d3ac4fa3SBarry Smith 595813a1f2bSShri Abhyankar ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 596813a1f2bSShri Abhyankar ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 597813a1f2bSShri Abhyankar if (row_identity && col_identity) { 59835233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 599813a1f2bSShri Abhyankar } else { 60035233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ; 601813a1f2bSShri Abhyankar } 60235233d99SShri Abhyankar C->ops->solveadd = MatSolveAdd_SeqAIJ; 60335233d99SShri Abhyankar C->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 60435233d99SShri Abhyankar C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 60535233d99SShri Abhyankar C->ops->matsolve = MatMatSolve_SeqAIJ; 606813a1f2bSShri Abhyankar C->assembled = PETSC_TRUE; 607813a1f2bSShri Abhyankar C->preallocated = PETSC_TRUE; 608813a1f2bSShri Abhyankar ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 609813a1f2bSShri Abhyankar 61014d2772eSHong Zhang /* MatShiftView(A,info,&sctx) */ 611813a1f2bSShri Abhyankar if (sctx.nshift){ 612f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 613813a1f2bSShri Abhyankar ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %G, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,sctx.shift_amount,sctx.shift_fraction,sctx.shift_top);CHKERRQ(ierr); 614f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 615813a1f2bSShri Abhyankar ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 616f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS){ 617915743fcSHong Zhang ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr); 618813a1f2bSShri Abhyankar } 619813a1f2bSShri Abhyankar } 620d3ac4fa3SBarry Smith ierr = Mat_CheckInode_FactorLU(C,PETSC_FALSE);CHKERRQ(ierr); 621813a1f2bSShri Abhyankar PetscFunctionReturn(0); 622813a1f2bSShri Abhyankar } 623813a1f2bSShri Abhyankar 6244a2ae208SSatish Balay #undef __FUNCT__ 625ad04f41aSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_inplace" 626ad04f41aSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info) 627289bc588SBarry Smith { 628719d5645SBarry Smith Mat C=B; 629416022c9SBarry Smith Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 63082bf6240SBarry Smith IS isrow = b->row,isicol = b->icol; 6316849ba73SBarry Smith PetscErrorCode ierr; 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; 641d3ac4fa3SBarry Smith PetscTruth row_identity, col_identity; 642289bc588SBarry Smith 6433a40ed3dSBarry Smith PetscFunctionBegin; 644504a3fadSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 645504a3fadSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 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]; 656e24cfe64SHong Zhang for (j=0; j<nz; j++) 6571a890ab1SHong Zhang rs += PetscAbsScalar(v[j]); 6581a890ab1SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 6596cc28720Svictorle } 660118f5789SBarry Smith sctx.shift_top *= 1.1; 661d2276718SHong Zhang sctx.nshift_max = 5; 662d2276718SHong Zhang sctx.shift_lo = 0.; 663d2276718SHong Zhang sctx.shift_hi = 1.; 664a0a356efSHong Zhang } 665a0a356efSHong Zhang 666504a3fadSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 667504a3fadSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 668504a3fadSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 669504a3fadSHong Zhang ics = ic; 670504a3fadSHong Zhang 671085256b3SBarry Smith do { 67207b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 673289bc588SBarry Smith for (i=0; i<n; i++){ 674b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 675b3bf805bSHong Zhang bjtmp = bj + bi[i]; 6769f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 677289bc588SBarry Smith 678289bc588SBarry Smith /* load in initial (unfactored row) */ 6799f7d0b68SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 6809f7d0b68SHong Zhang ajtmp = aj + ai[r[i]]; 6819f7d0b68SHong Zhang v = aa + ai[r[i]]; 682085256b3SBarry Smith for (j=0; j<nz; j++) { 683b3bf805bSHong Zhang rtmp[ics[ajtmp[j]]] = v[j]; 684085256b3SBarry Smith } 685d2276718SHong Zhang rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 686289bc588SBarry Smith 687b3bf805bSHong Zhang row = *bjtmp++; 688f2582111SSatish Balay while (row < i) { 6898c37ef55SBarry Smith pc = rtmp + row; 6908c37ef55SBarry Smith if (*pc != 0.0) { 691010a20caSHong Zhang pv = b->a + diag_offset[row]; 692010a20caSHong Zhang pj = b->j + diag_offset[row] + 1; 69335aab85fSBarry Smith multiplier = *pc / *pv++; 6948c37ef55SBarry Smith *pc = multiplier; 695b3bf805bSHong Zhang nz = bi[row+1] - diag_offset[row] - 1; 6969f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 697dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 698289bc588SBarry Smith } 699b3bf805bSHong Zhang row = *bjtmp++; 700289bc588SBarry Smith } 701416022c9SBarry Smith /* finished row so stick it into b->a */ 702b3bf805bSHong Zhang pv = b->a + bi[i] ; 703b3bf805bSHong Zhang pj = b->j + bi[i] ; 704b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 705b3bf805bSHong Zhang diag = diag_offset[i] - bi[i]; 706e57ff13aSBarry Smith rs = 0.0; 707d3d32019SBarry Smith for (j=0; j<nz; j++) { 7089f7d0b68SHong Zhang pv[j] = rtmp[pj[j]]; 7099f7d0b68SHong Zhang rs += PetscAbsScalar(pv[j]); 710d3d32019SBarry Smith } 711e57ff13aSBarry Smith rs -= PetscAbsScalar(pv[diag]); 712d2276718SHong Zhang 713d2276718SHong Zhang sctx.rs = rs; 714d2276718SHong Zhang sctx.pv = pv[diag]; 71507b50cabSHong Zhang ierr = MatPivotCheck(info,&sctx,i);CHKERRQ(ierr); 71607b50cabSHong Zhang if (sctx.newshift) break; 717504a3fadSHong Zhang pv[diag] = sctx.pv; 71835aab85fSBarry Smith } 719d2276718SHong Zhang 72007b50cabSHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 7216cc28720Svictorle /* 7226c037d1bSvictorle * if no shift in this attempt & shifting & started shifting & can refine, 7236cc28720Svictorle * then try lower shift 7246cc28720Svictorle */ 7250481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 7260481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 7270481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 72807b50cabSHong Zhang sctx.newshift = PETSC_TRUE; 729d2276718SHong Zhang sctx.nshift++; 7306cc28720Svictorle } 73107b50cabSHong Zhang } while (sctx.newshift); 7320f11f9aeSSatish Balay 73335aab85fSBarry Smith /* invert diagonal entries for simplier triangular solves */ 73435aab85fSBarry Smith for (i=0; i<n; i++) { 735010a20caSHong Zhang b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]]; 73635aab85fSBarry Smith } 737606d414cSSatish Balay ierr = PetscFree(rtmp);CHKERRQ(ierr); 73878b31e54SBarry Smith ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 73978b31e54SBarry Smith ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 740d3ac4fa3SBarry Smith 7418d8c7c43SBarry Smith ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 7428d8c7c43SBarry Smith ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 7438d8c7c43SBarry Smith if (row_identity && col_identity) { 744ad04f41aSHong Zhang C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_inplace; 7458d8c7c43SBarry Smith } else { 746ad04f41aSHong Zhang C->ops->solve = MatSolve_SeqAIJ_inplace; 747db4efbfdSBarry Smith } 748ad04f41aSHong Zhang C->ops->solveadd = MatSolveAdd_SeqAIJ_inplace; 749ad04f41aSHong Zhang C->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 750ad04f41aSHong Zhang C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace; 751ad04f41aSHong Zhang C->ops->matsolve = MatMatSolve_SeqAIJ_inplace; 752c456f294SBarry Smith C->assembled = PETSC_TRUE; 7535c9eb25fSBarry Smith C->preallocated = PETSC_TRUE; 754d0f46423SBarry Smith ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 755d2276718SHong Zhang if (sctx.nshift){ 756f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 7570481f469SBarry Smith ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %G, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,sctx.shift_amount,sctx.shift_fraction,sctx.shift_top);CHKERRQ(ierr); 758f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 759e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 7606cc28720Svictorle } 7610697b6caSHong Zhang } 762d3ac4fa3SBarry Smith (C)->ops->solve = MatSolve_SeqAIJ_inplace; 763d3ac4fa3SBarry Smith (C)->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 764d3ac4fa3SBarry Smith ierr = Mat_CheckInode(C,PETSC_FALSE);CHKERRQ(ierr); 7653a40ed3dSBarry Smith PetscFunctionReturn(0); 766289bc588SBarry Smith } 7670889b5dcSvictorle 768137fb511SHong Zhang /* 769137fb511SHong Zhang This routine implements inplace ILU(0) with row or/and column permutations. 770137fb511SHong Zhang Input: 771137fb511SHong Zhang A - original matrix 772137fb511SHong Zhang Output; 773137fb511SHong Zhang A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i] 774137fb511SHong Zhang a->j (col index) is permuted by the inverse of colperm, then sorted 775137fb511SHong Zhang a->a reordered accordingly with a->j 776137fb511SHong Zhang a->diag (ptr to diagonal elements) is updated. 777137fb511SHong Zhang */ 778137fb511SHong Zhang #undef __FUNCT__ 779137fb511SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_InplaceWithPerm" 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; 784137fb511SHong Zhang PetscErrorCode ierr; 7855d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 7865d0c19d7SBarry Smith PetscInt i,j,n=A->rmap->n,*ai=a->i,*aj=a->j; 7875d0c19d7SBarry Smith PetscInt *ajtmp,nz,row; 788b051339dSHong Zhang PetscInt *diag = a->diag,nbdiag,*pj; 789a77337e4SBarry Smith PetscScalar *rtmp,*pc,multiplier,d; 790504a3fadSHong Zhang MatScalar *pv,*v; 791137fb511SHong Zhang PetscReal rs; 792d90ac83dSHong Zhang FactorShiftCtx sctx; 793504a3fadSHong Zhang const MatScalar *aa=a->a,*vtmp; 794137fb511SHong Zhang 795137fb511SHong Zhang PetscFunctionBegin; 796e32f2f54SBarry Smith if (A != B) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address"); 797504a3fadSHong Zhang 798504a3fadSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 799504a3fadSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 800504a3fadSHong Zhang 801504a3fadSHong Zhang if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 802504a3fadSHong Zhang const PetscInt *ddiag = a->diag; 803504a3fadSHong Zhang sctx.shift_top = info->zeropivot; 804504a3fadSHong Zhang for (i=0; i<n; i++) { 805504a3fadSHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 806504a3fadSHong Zhang d = (aa)[ddiag[i]]; 807504a3fadSHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 808504a3fadSHong Zhang vtmp = aa+ai[i]; 809504a3fadSHong Zhang nz = ai[i+1] - ai[i]; 810504a3fadSHong Zhang for (j=0; j<nz; j++) 811504a3fadSHong Zhang rs += PetscAbsScalar(vtmp[j]); 812504a3fadSHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 813504a3fadSHong Zhang } 814504a3fadSHong Zhang sctx.shift_top *= 1.1; 815504a3fadSHong Zhang sctx.nshift_max = 5; 816504a3fadSHong Zhang sctx.shift_lo = 0.; 817504a3fadSHong Zhang sctx.shift_hi = 1.; 818504a3fadSHong Zhang } 819504a3fadSHong Zhang 820137fb511SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 821137fb511SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 822137fb511SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr); 823137fb511SHong Zhang ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 824b051339dSHong Zhang ics = ic; 825137fb511SHong Zhang 826504a3fadSHong Zhang #if defined(MV) 82775567043SBarry Smith sctx.shift_top = 0.; 828e60cf9a0SBarry Smith sctx.nshift_max = 0; 82975567043SBarry Smith sctx.shift_lo = 0.; 83075567043SBarry Smith sctx.shift_hi = 0.; 83175567043SBarry Smith sctx.shift_fraction = 0.; 832137fb511SHong Zhang 833f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 83475567043SBarry Smith sctx.shift_top = 0.; 835137fb511SHong Zhang for (i=0; i<n; i++) { 836137fb511SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 837b051339dSHong Zhang d = (a->a)[diag[i]]; 838137fb511SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 839b051339dSHong Zhang v = a->a+ai[i]; 840b051339dSHong Zhang nz = ai[i+1] - ai[i]; 841137fb511SHong Zhang for (j=0; j<nz; j++) 842137fb511SHong Zhang rs += PetscAbsScalar(v[j]); 843137fb511SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 844137fb511SHong Zhang } 845137fb511SHong Zhang if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot; 846137fb511SHong Zhang sctx.shift_top *= 1.1; 847137fb511SHong Zhang sctx.nshift_max = 5; 848137fb511SHong Zhang sctx.shift_lo = 0.; 849137fb511SHong Zhang sctx.shift_hi = 1.; 850137fb511SHong Zhang } 851137fb511SHong Zhang 85275567043SBarry Smith sctx.shift_amount = 0.; 853137fb511SHong Zhang sctx.nshift = 0; 854504a3fadSHong Zhang #endif 855504a3fadSHong Zhang 856137fb511SHong Zhang do { 85707b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 858137fb511SHong Zhang for (i=0; i<n; i++){ 859b051339dSHong Zhang /* load in initial unfactored row */ 860b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 861b051339dSHong Zhang ajtmp = aj + ai[r[i]]; 862b051339dSHong Zhang v = a->a + ai[r[i]]; 863137fb511SHong Zhang /* sort permuted ajtmp and values v accordingly */ 864b051339dSHong Zhang for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]]; 865137fb511SHong Zhang ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr); 866137fb511SHong Zhang 867b051339dSHong Zhang diag[r[i]] = ai[r[i]]; 868137fb511SHong Zhang for (j=0; j<nz; j++) { 869137fb511SHong Zhang rtmp[ajtmp[j]] = v[j]; 870b051339dSHong Zhang if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */ 871137fb511SHong Zhang } 872137fb511SHong Zhang rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 873137fb511SHong Zhang 874b051339dSHong Zhang row = *ajtmp++; 875137fb511SHong Zhang while (row < i) { 876137fb511SHong Zhang pc = rtmp + row; 877137fb511SHong Zhang if (*pc != 0.0) { 878b051339dSHong Zhang pv = a->a + diag[r[row]]; 879b051339dSHong Zhang pj = aj + diag[r[row]] + 1; 880137fb511SHong Zhang 881137fb511SHong Zhang multiplier = *pc / *pv++; 882137fb511SHong Zhang *pc = multiplier; 883b051339dSHong Zhang nz = ai[r[row]+1] - diag[r[row]] - 1; 884b051339dSHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 885dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 886137fb511SHong Zhang } 887b051339dSHong Zhang row = *ajtmp++; 888137fb511SHong Zhang } 889b051339dSHong Zhang /* finished row so overwrite it onto a->a */ 890b051339dSHong Zhang pv = a->a + ai[r[i]] ; 891b051339dSHong Zhang pj = aj + ai[r[i]] ; 892b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 893b051339dSHong Zhang nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */ 894137fb511SHong Zhang 895137fb511SHong Zhang rs = 0.0; 896137fb511SHong Zhang for (j=0; j<nz; j++) { 897b051339dSHong Zhang pv[j] = rtmp[pj[j]]; 898b051339dSHong Zhang if (j != nbdiag) rs += PetscAbsScalar(pv[j]); 899137fb511SHong Zhang } 900137fb511SHong Zhang 901137fb511SHong Zhang sctx.rs = rs; 902b051339dSHong Zhang sctx.pv = pv[nbdiag]; 90307b50cabSHong Zhang ierr = MatPivotCheck(info,&sctx,i);CHKERRQ(ierr); 90407b50cabSHong Zhang if (sctx.newshift) break; 905504a3fadSHong Zhang pv[nbdiag] = sctx.pv; 906137fb511SHong Zhang } 907137fb511SHong Zhang 90807b50cabSHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 909137fb511SHong Zhang /* 910137fb511SHong Zhang * if no shift in this attempt & shifting & started shifting & can refine, 911137fb511SHong Zhang * then try lower shift 912137fb511SHong Zhang */ 9130481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 9140481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 9150481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 91607b50cabSHong Zhang sctx.newshift = PETSC_TRUE; 917137fb511SHong Zhang sctx.nshift++; 918137fb511SHong Zhang } 91907b50cabSHong Zhang } while (sctx.newshift); 920137fb511SHong Zhang 921137fb511SHong Zhang /* invert diagonal entries for simplier triangular solves */ 922137fb511SHong Zhang for (i=0; i<n; i++) { 923b051339dSHong Zhang a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]]; 924137fb511SHong Zhang } 925137fb511SHong Zhang 926137fb511SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 927137fb511SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 928137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 929b051339dSHong Zhang A->ops->solve = MatSolve_SeqAIJ_InplaceWithPerm; 930ad04f41aSHong Zhang A->ops->solveadd = MatSolveAdd_SeqAIJ_inplace; 931ad04f41aSHong Zhang A->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 932ad04f41aSHong Zhang A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace; 933b051339dSHong Zhang A->assembled = PETSC_TRUE; 9345c9eb25fSBarry Smith A->preallocated = PETSC_TRUE; 935d0f46423SBarry Smith ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr); 936137fb511SHong Zhang if (sctx.nshift){ 937f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 9380481f469SBarry Smith ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %G, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,sctx.shift_amount,sctx.shift_fraction,sctx.shift_top);CHKERRQ(ierr); 939f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 940e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 941137fb511SHong Zhang } 942137fb511SHong Zhang } 943137fb511SHong Zhang PetscFunctionReturn(0); 944137fb511SHong Zhang } 945137fb511SHong Zhang 9460a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 9474a2ae208SSatish Balay #undef __FUNCT__ 9484a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ" 9490481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info) 950da3a660dSBarry Smith { 951dfbe8321SBarry Smith PetscErrorCode ierr; 952416022c9SBarry Smith Mat C; 953416022c9SBarry Smith 9543a40ed3dSBarry Smith PetscFunctionBegin; 9552692d6eeSBarry Smith ierr = MatGetFactor(A,MATSOLVERPETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr); 956719d5645SBarry Smith ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr); 957719d5645SBarry Smith ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr); 958db4efbfdSBarry Smith A->ops->solve = C->ops->solve; 959db4efbfdSBarry Smith A->ops->solvetranspose = C->ops->solvetranspose; 960eb6b5d47SBarry Smith ierr = MatHeaderMerge(A,C);CHKERRQ(ierr); 96152e6d16bSBarry Smith ierr = PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr); 9623a40ed3dSBarry Smith PetscFunctionReturn(0); 963da3a660dSBarry Smith } 9640a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 9651d098868SHong Zhang 966889a8dedSBarry Smith 9674a2ae208SSatish Balay #undef __FUNCT__ 968ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_inplace" 969ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_inplace(Mat A,Vec bb,Vec xx) 9708c37ef55SBarry Smith { 971416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 972416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 9736849ba73SBarry Smith PetscErrorCode ierr; 9745d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 9755d0c19d7SBarry Smith PetscInt nz; 9765d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 977dd6ea824SBarry Smith PetscScalar *x,*tmp,*tmps,sum; 978d9fead3dSBarry Smith const PetscScalar *b; 979dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 9808c37ef55SBarry Smith 9813a40ed3dSBarry Smith PetscFunctionBegin; 9823a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 98391bf9adeSBarry Smith 9843649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 9851ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 986416022c9SBarry Smith tmp = a->solve_work; 9878c37ef55SBarry Smith 9883b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 9893b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 9908c37ef55SBarry Smith 9918c37ef55SBarry Smith /* forward solve the lower triangular */ 9928c37ef55SBarry Smith tmp[0] = b[*r++]; 993010a20caSHong Zhang tmps = tmp; 9948c37ef55SBarry Smith for (i=1; i<n; i++) { 995010a20caSHong Zhang v = aa + ai[i] ; 996010a20caSHong Zhang vi = aj + ai[i] ; 99753e7425aSBarry Smith nz = a->diag[i] - ai[i]; 99853e7425aSBarry Smith sum = b[*r++]; 999003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10008c37ef55SBarry Smith tmp[i] = sum; 10018c37ef55SBarry Smith } 10028c37ef55SBarry Smith 10038c37ef55SBarry Smith /* backward solve the upper triangular */ 10048c37ef55SBarry Smith for (i=n-1; i>=0; i--){ 1005010a20caSHong Zhang v = aa + a->diag[i] + 1; 1006010a20caSHong Zhang vi = aj + a->diag[i] + 1; 1007416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 10088c37ef55SBarry Smith sum = tmp[i]; 1009003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1010010a20caSHong Zhang x[*c--] = tmp[i] = sum*aa[a->diag[i]]; 10118c37ef55SBarry Smith } 10128c37ef55SBarry Smith 10133b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10143b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 10153649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 10161ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1017dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 10183a40ed3dSBarry Smith PetscFunctionReturn(0); 10198c37ef55SBarry Smith } 1020026e39d0SSatish Balay 10212bebee5dSHong Zhang #undef __FUNCT__ 1022ad04f41aSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ_inplace" 1023ad04f41aSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ_inplace(Mat A,Mat B,Mat X) 10242bebee5dSHong Zhang { 10252bebee5dSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 10262bebee5dSHong Zhang IS iscol = a->col,isrow = a->row; 10272bebee5dSHong Zhang PetscErrorCode ierr; 10285d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 10295d0c19d7SBarry Smith PetscInt nz,neq; 10305d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 1031dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1032dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1033db4efbfdSBarry Smith PetscTruth bisdense,xisdense; 10342bebee5dSHong Zhang 10352bebee5dSHong Zhang PetscFunctionBegin; 10362bebee5dSHong Zhang if (!n) PetscFunctionReturn(0); 10372bebee5dSHong Zhang 1038db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr); 1039e32f2f54SBarry Smith if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix"); 1040db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr); 1041e32f2f54SBarry Smith if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix"); 1042db4efbfdSBarry Smith 10432bebee5dSHong Zhang ierr = MatGetArray(B,&b);CHKERRQ(ierr); 10442bebee5dSHong Zhang ierr = MatGetArray(X,&x);CHKERRQ(ierr); 10452bebee5dSHong Zhang 10462bebee5dSHong Zhang tmp = a->solve_work; 10472bebee5dSHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 10482bebee5dSHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 10492bebee5dSHong Zhang 1050a36b22ccSSatish Balay for (neq=0; neq<B->cmap->n; neq++){ 10512bebee5dSHong Zhang /* forward solve the lower triangular */ 10522bebee5dSHong Zhang tmp[0] = b[r[0]]; 10532bebee5dSHong Zhang tmps = tmp; 10542bebee5dSHong Zhang for (i=1; i<n; i++) { 10552bebee5dSHong Zhang v = aa + ai[i] ; 10562bebee5dSHong Zhang vi = aj + ai[i] ; 10572bebee5dSHong Zhang nz = a->diag[i] - ai[i]; 10582bebee5dSHong Zhang sum = b[r[i]]; 1059003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10602bebee5dSHong Zhang tmp[i] = sum; 10612bebee5dSHong Zhang } 10622bebee5dSHong Zhang /* backward solve the upper triangular */ 10632bebee5dSHong Zhang for (i=n-1; i>=0; i--){ 10642bebee5dSHong Zhang v = aa + a->diag[i] + 1; 10652bebee5dSHong Zhang vi = aj + a->diag[i] + 1; 10662bebee5dSHong Zhang nz = ai[i+1] - a->diag[i] - 1; 10672bebee5dSHong Zhang sum = tmp[i]; 1068003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10692bebee5dSHong Zhang x[c[i]] = tmp[i] = sum*aa[a->diag[i]]; 10702bebee5dSHong Zhang } 10712bebee5dSHong Zhang 10722bebee5dSHong Zhang b += n; 10732bebee5dSHong Zhang x += n; 10742bebee5dSHong Zhang } 10752bebee5dSHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10762bebee5dSHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 10772bebee5dSHong Zhang ierr = MatRestoreArray(B,&b);CHKERRQ(ierr); 10782bebee5dSHong Zhang ierr = MatRestoreArray(X,&x);CHKERRQ(ierr); 1079dc0b31edSSatish Balay ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr); 10802bebee5dSHong Zhang PetscFunctionReturn(0); 10812bebee5dSHong Zhang } 10822bebee5dSHong Zhang 1083137fb511SHong Zhang #undef __FUNCT__ 108435233d99SShri Abhyankar #define __FUNCT__ "MatMatSolve_SeqAIJ" 108535233d99SShri Abhyankar PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X) 10869bd0847aSShri Abhyankar { 10879bd0847aSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 10889bd0847aSShri Abhyankar IS iscol = a->col,isrow = a->row; 10899bd0847aSShri Abhyankar PetscErrorCode ierr; 10909bd0847aSShri Abhyankar PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag; 10919bd0847aSShri Abhyankar PetscInt nz,neq; 10929bd0847aSShri Abhyankar const PetscInt *rout,*cout,*r,*c; 10939bd0847aSShri Abhyankar PetscScalar *x,*b,*tmp,sum; 10949bd0847aSShri Abhyankar const MatScalar *aa = a->a,*v; 10959bd0847aSShri Abhyankar PetscTruth bisdense,xisdense; 10969bd0847aSShri Abhyankar 10979bd0847aSShri Abhyankar PetscFunctionBegin; 10989bd0847aSShri Abhyankar if (!n) PetscFunctionReturn(0); 10999bd0847aSShri Abhyankar 11009bd0847aSShri Abhyankar ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr); 1101e32f2f54SBarry Smith if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix"); 11029bd0847aSShri Abhyankar ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr); 1103e32f2f54SBarry Smith if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix"); 11049bd0847aSShri Abhyankar 11059bd0847aSShri Abhyankar ierr = MatGetArray(B,&b);CHKERRQ(ierr); 11069bd0847aSShri Abhyankar ierr = MatGetArray(X,&x);CHKERRQ(ierr); 11079bd0847aSShri Abhyankar 11089bd0847aSShri Abhyankar tmp = a->solve_work; 11099bd0847aSShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 11109bd0847aSShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 11119bd0847aSShri Abhyankar 11129bd0847aSShri Abhyankar for (neq=0; neq<B->cmap->n; neq++){ 11139bd0847aSShri Abhyankar /* forward solve the lower triangular */ 11149bd0847aSShri Abhyankar tmp[0] = b[r[0]]; 11159bd0847aSShri Abhyankar v = aa; 11169bd0847aSShri Abhyankar vi = aj; 11179bd0847aSShri Abhyankar for (i=1; i<n; i++) { 11189bd0847aSShri Abhyankar nz = ai[i+1] - ai[i]; 11199bd0847aSShri Abhyankar sum = b[r[i]]; 11209bd0847aSShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 11219bd0847aSShri Abhyankar tmp[i] = sum; 11229bd0847aSShri Abhyankar v += nz; vi += nz; 11239bd0847aSShri Abhyankar } 11249bd0847aSShri Abhyankar 11259bd0847aSShri Abhyankar /* backward solve the upper triangular */ 11269bd0847aSShri Abhyankar for (i=n-1; i>=0; i--){ 11279bd0847aSShri Abhyankar v = aa + adiag[i+1]+1; 11289bd0847aSShri Abhyankar vi = aj + adiag[i+1]+1; 11299bd0847aSShri Abhyankar nz = adiag[i]-adiag[i+1]-1; 11309bd0847aSShri Abhyankar sum = tmp[i]; 11319bd0847aSShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 11329bd0847aSShri Abhyankar x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */ 11339bd0847aSShri Abhyankar } 11349bd0847aSShri Abhyankar 11359bd0847aSShri Abhyankar b += n; 11369bd0847aSShri Abhyankar x += n; 11379bd0847aSShri Abhyankar } 11389bd0847aSShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 11399bd0847aSShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 11409bd0847aSShri Abhyankar ierr = MatRestoreArray(B,&b);CHKERRQ(ierr); 11419bd0847aSShri Abhyankar ierr = MatRestoreArray(X,&x);CHKERRQ(ierr); 11429bd0847aSShri Abhyankar ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr); 11439bd0847aSShri Abhyankar PetscFunctionReturn(0); 11449bd0847aSShri Abhyankar } 11459bd0847aSShri Abhyankar 11469bd0847aSShri Abhyankar #undef __FUNCT__ 1147137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm" 1148137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx) 1149137fb511SHong Zhang { 1150137fb511SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1151137fb511SHong Zhang IS iscol = a->col,isrow = a->row; 1152137fb511SHong Zhang PetscErrorCode ierr; 11535d0c19d7SBarry Smith const PetscInt *r,*c,*rout,*cout; 11545d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 11555d0c19d7SBarry Smith PetscInt nz,row; 1156dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1157dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1158137fb511SHong Zhang 1159137fb511SHong Zhang PetscFunctionBegin; 1160137fb511SHong Zhang if (!n) PetscFunctionReturn(0); 1161137fb511SHong Zhang 1162137fb511SHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 1163137fb511SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1164137fb511SHong Zhang tmp = a->solve_work; 1165137fb511SHong Zhang 1166137fb511SHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1167137fb511SHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1168137fb511SHong Zhang 1169137fb511SHong Zhang /* forward solve the lower triangular */ 1170137fb511SHong Zhang tmp[0] = b[*r++]; 1171137fb511SHong Zhang tmps = tmp; 1172137fb511SHong Zhang for (row=1; row<n; row++) { 1173137fb511SHong Zhang i = rout[row]; /* permuted row */ 1174137fb511SHong Zhang v = aa + ai[i] ; 1175137fb511SHong Zhang vi = aj + ai[i] ; 1176137fb511SHong Zhang nz = a->diag[i] - ai[i]; 1177137fb511SHong Zhang sum = b[*r++]; 1178003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1179137fb511SHong Zhang tmp[row] = sum; 1180137fb511SHong Zhang } 1181137fb511SHong Zhang 1182137fb511SHong Zhang /* backward solve the upper triangular */ 1183137fb511SHong Zhang for (row=n-1; row>=0; row--){ 1184137fb511SHong Zhang i = rout[row]; /* permuted row */ 1185137fb511SHong Zhang v = aa + a->diag[i] + 1; 1186137fb511SHong Zhang vi = aj + a->diag[i] + 1; 1187137fb511SHong Zhang nz = ai[i+1] - a->diag[i] - 1; 1188137fb511SHong Zhang sum = tmp[row]; 1189003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1190137fb511SHong Zhang x[*c--] = tmp[row] = sum*aa[a->diag[i]]; 1191137fb511SHong Zhang } 1192137fb511SHong Zhang 1193137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1194137fb511SHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1195137fb511SHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 1196137fb511SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1197dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 1198137fb511SHong Zhang PetscFunctionReturn(0); 1199137fb511SHong Zhang } 1200137fb511SHong Zhang 1201930ae53cSBarry Smith /* ----------------------------------------------------------- */ 1202a4005a5dSBarry Smith #include "../src/mat/impls/aij/seq/ftn-kernels/fsolve.h" 12034a2ae208SSatish Balay #undef __FUNCT__ 1204ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_inplace" 1205ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_inplace(Mat A,Vec bb,Vec xx) 1206930ae53cSBarry Smith { 1207930ae53cSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 12086849ba73SBarry Smith PetscErrorCode ierr; 1209d0f46423SBarry Smith PetscInt n = A->rmap->n; 1210b377110cSBarry Smith const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag; 1211356650c2SBarry Smith PetscScalar *x; 1212356650c2SBarry Smith const PetscScalar *b; 1213dd6ea824SBarry Smith const MatScalar *aa = a->a; 1214aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 1215356650c2SBarry Smith PetscInt adiag_i,i,nz,ai_i; 1216b377110cSBarry Smith const PetscInt *vi; 1217dd6ea824SBarry Smith const MatScalar *v; 1218dd6ea824SBarry Smith PetscScalar sum; 1219d85afc42SSatish Balay #endif 1220930ae53cSBarry Smith 12213a40ed3dSBarry Smith PetscFunctionBegin; 12223a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 1223930ae53cSBarry Smith 12243649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 12251ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1226930ae53cSBarry Smith 1227aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 12281c4feecaSSatish Balay fortransolveaij_(&n,x,ai,aj,adiag,aa,b); 12291c4feecaSSatish Balay #else 1230930ae53cSBarry Smith /* forward solve the lower triangular */ 1231930ae53cSBarry Smith x[0] = b[0]; 1232930ae53cSBarry Smith for (i=1; i<n; i++) { 1233930ae53cSBarry Smith ai_i = ai[i]; 1234930ae53cSBarry Smith v = aa + ai_i; 1235930ae53cSBarry Smith vi = aj + ai_i; 1236930ae53cSBarry Smith nz = adiag[i] - ai_i; 1237930ae53cSBarry Smith sum = b[i]; 1238003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1239930ae53cSBarry Smith x[i] = sum; 1240930ae53cSBarry Smith } 1241930ae53cSBarry Smith 1242930ae53cSBarry Smith /* backward solve the upper triangular */ 1243930ae53cSBarry Smith for (i=n-1; i>=0; i--){ 1244930ae53cSBarry Smith adiag_i = adiag[i]; 1245d708749eSBarry Smith v = aa + adiag_i + 1; 1246d708749eSBarry Smith vi = aj + adiag_i + 1; 1247930ae53cSBarry Smith nz = ai[i+1] - adiag_i - 1; 1248930ae53cSBarry Smith sum = x[i]; 1249003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1250930ae53cSBarry Smith x[i] = sum*aa[adiag_i]; 1251930ae53cSBarry Smith } 12521c4feecaSSatish Balay #endif 1253dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 12543649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 12551ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 12563a40ed3dSBarry Smith PetscFunctionReturn(0); 1257930ae53cSBarry Smith } 1258930ae53cSBarry Smith 12594a2ae208SSatish Balay #undef __FUNCT__ 1260ad04f41aSHong Zhang #define __FUNCT__ "MatSolveAdd_SeqAIJ_inplace" 1261ad04f41aSHong Zhang PetscErrorCode MatSolveAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec yy,Vec xx) 1262da3a660dSBarry Smith { 1263416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1264416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 12656849ba73SBarry Smith PetscErrorCode ierr; 126604fbf559SBarry Smith PetscInt i, n = A->rmap->n,j; 12675d0c19d7SBarry Smith PetscInt nz; 126804fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j; 126904fbf559SBarry Smith PetscScalar *x,*tmp,sum; 127004fbf559SBarry Smith const PetscScalar *b; 1271dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1272da3a660dSBarry Smith 12733a40ed3dSBarry Smith PetscFunctionBegin; 127478b31e54SBarry Smith if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 1275da3a660dSBarry Smith 12763649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 12771ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1278416022c9SBarry Smith tmp = a->solve_work; 1279da3a660dSBarry Smith 12803b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 12813b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1282da3a660dSBarry Smith 1283da3a660dSBarry Smith /* forward solve the lower triangular */ 1284da3a660dSBarry Smith tmp[0] = b[*r++]; 1285da3a660dSBarry Smith for (i=1; i<n; i++) { 1286010a20caSHong Zhang v = aa + ai[i] ; 1287010a20caSHong Zhang vi = aj + ai[i] ; 1288416022c9SBarry Smith nz = a->diag[i] - ai[i]; 1289da3a660dSBarry Smith sum = b[*r++]; 129004fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1291da3a660dSBarry Smith tmp[i] = sum; 1292da3a660dSBarry Smith } 1293da3a660dSBarry Smith 1294da3a660dSBarry Smith /* backward solve the upper triangular */ 1295da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1296010a20caSHong Zhang v = aa + a->diag[i] + 1; 1297010a20caSHong Zhang vi = aj + a->diag[i] + 1; 1298416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 1299da3a660dSBarry Smith sum = tmp[i]; 130004fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1301010a20caSHong Zhang tmp[i] = sum*aa[a->diag[i]]; 1302da3a660dSBarry Smith x[*c--] += tmp[i]; 1303da3a660dSBarry Smith } 1304da3a660dSBarry Smith 13053b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 13063b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 13073649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 13081ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1309dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 1310898183ecSLois Curfman McInnes 13113a40ed3dSBarry Smith PetscFunctionReturn(0); 1312da3a660dSBarry Smith } 131304fbf559SBarry Smith 13144a2ae208SSatish Balay #undef __FUNCT__ 131535233d99SShri Abhyankar #define __FUNCT__ "MatSolveAdd_SeqAIJ" 131635233d99SShri Abhyankar PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx) 13173c0119dfSShri Abhyankar { 13183c0119dfSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 13193c0119dfSShri Abhyankar IS iscol = a->col,isrow = a->row; 13203c0119dfSShri Abhyankar PetscErrorCode ierr; 13213c0119dfSShri Abhyankar PetscInt i, n = A->rmap->n,j; 13223c0119dfSShri Abhyankar PetscInt nz; 13233c0119dfSShri Abhyankar const PetscInt *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag; 13243c0119dfSShri Abhyankar PetscScalar *x,*tmp,sum; 13253c0119dfSShri Abhyankar const PetscScalar *b; 13263c0119dfSShri Abhyankar const MatScalar *aa = a->a,*v; 13273c0119dfSShri Abhyankar 13283c0119dfSShri Abhyankar PetscFunctionBegin; 13293c0119dfSShri Abhyankar if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 13303c0119dfSShri Abhyankar 13313649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 13323c0119dfSShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 13333c0119dfSShri Abhyankar tmp = a->solve_work; 13343c0119dfSShri Abhyankar 13353c0119dfSShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 13363c0119dfSShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 13373c0119dfSShri Abhyankar 13383c0119dfSShri Abhyankar /* forward solve the lower triangular */ 13393c0119dfSShri Abhyankar tmp[0] = b[r[0]]; 13403c0119dfSShri Abhyankar v = aa; 13413c0119dfSShri Abhyankar vi = aj; 13423c0119dfSShri Abhyankar for (i=1; i<n; i++) { 13433c0119dfSShri Abhyankar nz = ai[i+1] - ai[i]; 13443c0119dfSShri Abhyankar sum = b[r[i]]; 13453c0119dfSShri Abhyankar for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 13463c0119dfSShri Abhyankar tmp[i] = sum; 13473c0119dfSShri Abhyankar v += nz; vi += nz; 13483c0119dfSShri Abhyankar } 13493c0119dfSShri Abhyankar 13503c0119dfSShri Abhyankar /* backward solve the upper triangular */ 13513c0119dfSShri Abhyankar v = aa + adiag[n-1]; 13523c0119dfSShri Abhyankar vi = aj + adiag[n-1]; 13533c0119dfSShri Abhyankar for (i=n-1; i>=0; i--){ 13543c0119dfSShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 13553c0119dfSShri Abhyankar sum = tmp[i]; 13563c0119dfSShri Abhyankar for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 13573c0119dfSShri Abhyankar tmp[i] = sum*v[nz]; 13583c0119dfSShri Abhyankar x[c[i]] += tmp[i]; 13593c0119dfSShri Abhyankar v += nz+1; vi += nz+1; 13603c0119dfSShri Abhyankar } 13613c0119dfSShri Abhyankar 13623c0119dfSShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 13633c0119dfSShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 13643649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 13653c0119dfSShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 13663c0119dfSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 13673c0119dfSShri Abhyankar 13683c0119dfSShri Abhyankar PetscFunctionReturn(0); 13693c0119dfSShri Abhyankar } 13703c0119dfSShri Abhyankar 13713c0119dfSShri Abhyankar #undef __FUNCT__ 1372ad04f41aSHong Zhang #define __FUNCT__ "MatSolveTranspose_SeqAIJ_inplace" 1373ad04f41aSHong Zhang PetscErrorCode MatSolveTranspose_SeqAIJ_inplace(Mat A,Vec bb,Vec xx) 1374da3a660dSBarry Smith { 1375416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 13762235e667SBarry Smith IS iscol = a->col,isrow = a->row; 13776849ba73SBarry Smith PetscErrorCode ierr; 137804fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 137904fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 138004fbf559SBarry Smith PetscInt nz; 138104fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1382dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 138304fbf559SBarry Smith const PetscScalar *b; 1384da3a660dSBarry Smith 13853a40ed3dSBarry Smith PetscFunctionBegin; 13863649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 13871ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1388416022c9SBarry Smith tmp = a->solve_work; 1389da3a660dSBarry Smith 13902235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 13912235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1392da3a660dSBarry Smith 1393da3a660dSBarry Smith /* copy the b into temp work space according to permutation */ 13942235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1395da3a660dSBarry Smith 1396da3a660dSBarry Smith /* forward solve the U^T */ 1397da3a660dSBarry Smith for (i=0; i<n; i++) { 1398010a20caSHong Zhang v = aa + diag[i] ; 1399010a20caSHong Zhang vi = aj + diag[i] + 1; 1400f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 1401c38d4ed2SBarry Smith s1 = tmp[i]; 1402ef66eb69SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 140304fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1404c38d4ed2SBarry Smith tmp[i] = s1; 1405da3a660dSBarry Smith } 1406da3a660dSBarry Smith 1407da3a660dSBarry Smith /* backward solve the L^T */ 1408da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1409010a20caSHong Zhang v = aa + diag[i] - 1 ; 1410010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1411f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 1412f1af5d2fSBarry Smith s1 = tmp[i]; 141304fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 1414da3a660dSBarry Smith } 1415da3a660dSBarry Smith 1416da3a660dSBarry Smith /* copy tmp into x according to permutation */ 1417591294e4SBarry Smith for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1418da3a660dSBarry Smith 14192235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 14202235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 14213649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 14221ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1423da3a660dSBarry Smith 1424dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 14253a40ed3dSBarry Smith PetscFunctionReturn(0); 1426da3a660dSBarry Smith } 1427da3a660dSBarry Smith 1428d1fa9404SShri Abhyankar #undef __FUNCT__ 142935233d99SShri Abhyankar #define __FUNCT__ "MatSolveTranspose_SeqAIJ" 143035233d99SShri Abhyankar PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx) 1431d1fa9404SShri Abhyankar { 1432d1fa9404SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1433d1fa9404SShri Abhyankar IS iscol = a->col,isrow = a->row; 1434d1fa9404SShri Abhyankar PetscErrorCode ierr; 1435d1fa9404SShri Abhyankar const PetscInt *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi; 1436d1fa9404SShri Abhyankar PetscInt i,n = A->rmap->n,j; 1437d1fa9404SShri Abhyankar PetscInt nz; 1438d1fa9404SShri Abhyankar PetscScalar *x,*tmp,s1; 1439d1fa9404SShri Abhyankar const MatScalar *aa = a->a,*v; 1440d1fa9404SShri Abhyankar const PetscScalar *b; 1441d1fa9404SShri Abhyankar 1442d1fa9404SShri Abhyankar PetscFunctionBegin; 14433649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 1444d1fa9404SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1445d1fa9404SShri Abhyankar tmp = a->solve_work; 1446d1fa9404SShri Abhyankar 1447d1fa9404SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1448d1fa9404SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1449d1fa9404SShri Abhyankar 1450d1fa9404SShri Abhyankar /* copy the b into temp work space according to permutation */ 1451d1fa9404SShri Abhyankar for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1452d1fa9404SShri Abhyankar 1453d1fa9404SShri Abhyankar /* forward solve the U^T */ 1454d1fa9404SShri Abhyankar for (i=0; i<n; i++) { 1455d1fa9404SShri Abhyankar v = aa + adiag[i+1] + 1; 1456d1fa9404SShri Abhyankar vi = aj + adiag[i+1] + 1; 1457d1fa9404SShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 1458d1fa9404SShri Abhyankar s1 = tmp[i]; 1459d1fa9404SShri Abhyankar s1 *= v[nz]; /* multiply by inverse of diagonal entry */ 1460d1fa9404SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1461d1fa9404SShri Abhyankar tmp[i] = s1; 1462d1fa9404SShri Abhyankar } 1463d1fa9404SShri Abhyankar 1464d1fa9404SShri Abhyankar /* backward solve the L^T */ 1465d1fa9404SShri Abhyankar for (i=n-1; i>=0; i--){ 1466d1fa9404SShri Abhyankar v = aa + ai[i]; 1467d1fa9404SShri Abhyankar vi = aj + ai[i]; 1468d1fa9404SShri Abhyankar nz = ai[i+1] - ai[i]; 1469d1fa9404SShri Abhyankar s1 = tmp[i]; 1470d1fa9404SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1471d1fa9404SShri Abhyankar } 1472d1fa9404SShri Abhyankar 1473d1fa9404SShri Abhyankar /* copy tmp into x according to permutation */ 1474d1fa9404SShri Abhyankar for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1475d1fa9404SShri Abhyankar 1476d1fa9404SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1477d1fa9404SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 14783649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 1479d1fa9404SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1480d1fa9404SShri Abhyankar 1481d1fa9404SShri Abhyankar ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 1482d1fa9404SShri Abhyankar PetscFunctionReturn(0); 1483d1fa9404SShri Abhyankar } 1484d1fa9404SShri Abhyankar 14854a2ae208SSatish Balay #undef __FUNCT__ 1486ad04f41aSHong Zhang #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ_inplace" 1487ad04f41aSHong Zhang PetscErrorCode MatSolveTransposeAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec zz,Vec xx) 1488da3a660dSBarry Smith { 1489416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 14902235e667SBarry Smith IS iscol = a->col,isrow = a->row; 14916849ba73SBarry Smith PetscErrorCode ierr; 149204fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 149304fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 149404fbf559SBarry Smith PetscInt nz; 149504fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1496dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 149704fbf559SBarry Smith const PetscScalar *b; 14986abc6512SBarry Smith 14993a40ed3dSBarry Smith PetscFunctionBegin; 150023bc6035SMatthew Knepley if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 15013649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 15021ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1503416022c9SBarry Smith tmp = a->solve_work; 15046abc6512SBarry Smith 15052235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 15062235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 15076abc6512SBarry Smith 15086abc6512SBarry Smith /* copy the b into temp work space according to permutation */ 15092235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 15106abc6512SBarry Smith 15116abc6512SBarry Smith /* forward solve the U^T */ 15126abc6512SBarry Smith for (i=0; i<n; i++) { 1513010a20caSHong Zhang v = aa + diag[i] ; 1514010a20caSHong Zhang vi = aj + diag[i] + 1; 1515f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 151604fbf559SBarry Smith s1 = tmp[i]; 151704fbf559SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 151804fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 151904fbf559SBarry Smith tmp[i] = s1; 15206abc6512SBarry Smith } 15216abc6512SBarry Smith 15226abc6512SBarry Smith /* backward solve the L^T */ 15236abc6512SBarry Smith for (i=n-1; i>=0; i--){ 1524010a20caSHong Zhang v = aa + diag[i] - 1 ; 1525010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1526f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 152704fbf559SBarry Smith s1 = tmp[i]; 152804fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 15296abc6512SBarry Smith } 15306abc6512SBarry Smith 15316abc6512SBarry Smith /* copy tmp into x according to permutation */ 15326abc6512SBarry Smith for (i=0; i<n; i++) x[r[i]] += tmp[i]; 15336abc6512SBarry Smith 15342235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 15352235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 15363649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 15371ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 15386abc6512SBarry Smith 153904fbf559SBarry Smith ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 15403a40ed3dSBarry Smith PetscFunctionReturn(0); 1541da3a660dSBarry Smith } 154204fbf559SBarry Smith 1543533e6011SShri Abhyankar #undef __FUNCT__ 154435233d99SShri Abhyankar #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ" 154535233d99SShri Abhyankar PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx) 1546533e6011SShri Abhyankar { 1547533e6011SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1548533e6011SShri Abhyankar IS iscol = a->col,isrow = a->row; 1549533e6011SShri Abhyankar PetscErrorCode ierr; 1550533e6011SShri Abhyankar const PetscInt *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi; 1551533e6011SShri Abhyankar PetscInt i,n = A->rmap->n,j; 1552533e6011SShri Abhyankar PetscInt nz; 1553533e6011SShri Abhyankar PetscScalar *x,*tmp,s1; 1554533e6011SShri Abhyankar const MatScalar *aa = a->a,*v; 1555533e6011SShri Abhyankar const PetscScalar *b; 1556533e6011SShri Abhyankar 1557533e6011SShri Abhyankar PetscFunctionBegin; 1558533e6011SShri Abhyankar if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 15593649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 1560533e6011SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1561533e6011SShri Abhyankar tmp = a->solve_work; 1562533e6011SShri Abhyankar 1563533e6011SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1564533e6011SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1565533e6011SShri Abhyankar 1566533e6011SShri Abhyankar /* copy the b into temp work space according to permutation */ 1567533e6011SShri Abhyankar for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1568533e6011SShri Abhyankar 1569533e6011SShri Abhyankar /* forward solve the U^T */ 1570533e6011SShri Abhyankar for (i=0; i<n; i++) { 1571533e6011SShri Abhyankar v = aa + adiag[i+1] + 1; 1572533e6011SShri Abhyankar vi = aj + adiag[i+1] + 1; 1573533e6011SShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 1574533e6011SShri Abhyankar s1 = tmp[i]; 1575533e6011SShri Abhyankar s1 *= v[nz]; /* multiply by inverse of diagonal entry */ 1576533e6011SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1577533e6011SShri Abhyankar tmp[i] = s1; 1578533e6011SShri Abhyankar } 1579533e6011SShri Abhyankar 1580533e6011SShri Abhyankar 1581533e6011SShri Abhyankar /* backward solve the L^T */ 1582533e6011SShri Abhyankar for (i=n-1; i>=0; i--){ 1583533e6011SShri Abhyankar v = aa + ai[i] ; 1584533e6011SShri Abhyankar vi = aj + ai[i]; 1585533e6011SShri Abhyankar nz = ai[i+1] - ai[i]; 1586533e6011SShri Abhyankar s1 = tmp[i]; 1587c5e3b2a3SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1588533e6011SShri Abhyankar } 1589533e6011SShri Abhyankar 1590533e6011SShri Abhyankar /* copy tmp into x according to permutation */ 1591533e6011SShri Abhyankar for (i=0; i<n; i++) x[r[i]] += tmp[i]; 1592533e6011SShri Abhyankar 1593533e6011SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1594533e6011SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 15953649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 1596533e6011SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1597533e6011SShri Abhyankar 1598533e6011SShri Abhyankar ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 1599533e6011SShri Abhyankar PetscFunctionReturn(0); 1600533e6011SShri Abhyankar } 1601533e6011SShri Abhyankar 16029e25ed09SBarry Smith /* ----------------------------------------------------------------*/ 1603b47f5a65SHong Zhang 1604f77e22a1SHong Zhang EXTERN PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,Mat,MatDuplicateOption,PetscTruth); 160508480c60SBarry Smith 16068fc3a347SHong Zhang /* 16078a76ed4fSHong Zhang ilu() under revised new data structure. 1608813a1f2bSShri Abhyankar Factored arrays bj and ba are stored as 1609813a1f2bSShri Abhyankar L(0,:), L(1,:), ...,L(n-1,:), U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:) 1610813a1f2bSShri Abhyankar 1611813a1f2bSShri Abhyankar bi=fact->i is an array of size n+1, in which 1612813a1f2bSShri Abhyankar bi+ 1613baabb860SHong Zhang bi[i]: points to 1st entry of L(i,:),i=0,...,n-1 1614baabb860SHong Zhang bi[n]: points to L(n-1,n-1)+1 1615baabb860SHong Zhang 1616813a1f2bSShri Abhyankar bdiag=fact->diag is an array of size n+1,in which 1617baabb860SHong Zhang bdiag[i]: points to diagonal of U(i,:), i=0,...,n-1 1618a24f213cSHong Zhang bdiag[n]: points to entry of U(n-1,0)-1 1619813a1f2bSShri Abhyankar 1620813a1f2bSShri Abhyankar U(i,:) contains bdiag[i] as its last entry, i.e., 1621813a1f2bSShri Abhyankar U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 1622813a1f2bSShri Abhyankar */ 1623813a1f2bSShri Abhyankar #undef __FUNCT__ 162435233d99SShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0" 162535233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1626813a1f2bSShri Abhyankar { 1627813a1f2bSShri Abhyankar 1628813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1629813a1f2bSShri Abhyankar PetscErrorCode ierr; 1630bbd65245SShri Abhyankar const PetscInt n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag; 1631bbd65245SShri Abhyankar PetscInt i,j,k=0,nz,*bi,*bj,*bdiag; 16321df811f5SHong Zhang PetscTruth missing; 16331df811f5SHong Zhang IS isicol; 1634813a1f2bSShri Abhyankar 1635813a1f2bSShri Abhyankar PetscFunctionBegin; 1636e32f2f54SBarry Smith if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 16371df811f5SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 1638e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 16391df811f5SHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 16401df811f5SHong Zhang 1641813a1f2bSShri Abhyankar ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr); 1642813a1f2bSShri Abhyankar b = (Mat_SeqAIJ*)(fact)->data; 1643813a1f2bSShri Abhyankar 1644813a1f2bSShri Abhyankar /* allocate matrix arrays for new data structure */ 1645813a1f2bSShri Abhyankar ierr = PetscMalloc3(ai[n]+1,PetscScalar,&b->a,ai[n]+1,PetscInt,&b->j,n+1,PetscInt,&b->i);CHKERRQ(ierr); 1646813a1f2bSShri Abhyankar ierr = PetscLogObjectMemory(fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(n+1)*sizeof(PetscInt));CHKERRQ(ierr); 1647813a1f2bSShri Abhyankar b->singlemalloc = PETSC_TRUE; 1648813a1f2bSShri Abhyankar if (!b->diag){ 1649813a1f2bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscInt),&b->diag);CHKERRQ(ierr); 1650813a1f2bSShri Abhyankar ierr = PetscLogObjectMemory(fact,(n+1)*sizeof(PetscInt));CHKERRQ(ierr); 1651813a1f2bSShri Abhyankar } 1652813a1f2bSShri Abhyankar bdiag = b->diag; 1653813a1f2bSShri Abhyankar 1654813a1f2bSShri Abhyankar if (n > 0) { 1655813a1f2bSShri Abhyankar ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr); 1656813a1f2bSShri Abhyankar } 1657813a1f2bSShri Abhyankar 1658813a1f2bSShri Abhyankar /* set bi and bj with new data structure */ 1659813a1f2bSShri Abhyankar bi = b->i; 1660813a1f2bSShri Abhyankar bj = b->j; 1661813a1f2bSShri Abhyankar 1662813a1f2bSShri Abhyankar /* L part */ 1663e60cf9a0SBarry Smith bi[0] = 0; 1664813a1f2bSShri Abhyankar for (i=0; i<n; i++){ 1665813a1f2bSShri Abhyankar nz = adiag[i] - ai[i]; 1666813a1f2bSShri Abhyankar bi[i+1] = bi[i] + nz; 1667813a1f2bSShri Abhyankar aj = a->j + ai[i]; 1668813a1f2bSShri Abhyankar for (j=0; j<nz; j++){ 1669bbd65245SShri Abhyankar /* *bj = aj[j]; bj++; */ 1670bbd65245SShri Abhyankar bj[k++] = aj[j]; 1671813a1f2bSShri Abhyankar } 1672813a1f2bSShri Abhyankar } 1673813a1f2bSShri Abhyankar 1674813a1f2bSShri Abhyankar /* U part */ 167562fbd6afSShri Abhyankar bdiag[n] = bi[n]-1; 1676813a1f2bSShri Abhyankar for (i=n-1; i>=0; i--){ 1677813a1f2bSShri Abhyankar nz = ai[i+1] - adiag[i] - 1; 1678813a1f2bSShri Abhyankar aj = a->j + adiag[i] + 1; 1679813a1f2bSShri Abhyankar for (j=0; j<nz; j++){ 1680bbd65245SShri Abhyankar /* *bj = aj[j]; bj++; */ 1681bbd65245SShri Abhyankar bj[k++] = aj[j]; 1682813a1f2bSShri Abhyankar } 1683813a1f2bSShri Abhyankar /* diag[i] */ 1684bbd65245SShri Abhyankar /* *bj = i; bj++; */ 1685bbd65245SShri Abhyankar bj[k++] = i; 1686a24f213cSHong Zhang bdiag[i] = bdiag[i+1] + nz + 1; 1687813a1f2bSShri Abhyankar } 16881df811f5SHong Zhang 1689d5f3da31SBarry Smith fact->factortype = MAT_FACTOR_ILU; 16901df811f5SHong Zhang fact->info.factor_mallocs = 0; 16911df811f5SHong Zhang fact->info.fill_ratio_given = info->fill; 16921df811f5SHong Zhang fact->info.fill_ratio_needed = 1.0; 169335233d99SShri Abhyankar fact->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 16941df811f5SHong Zhang 16951df811f5SHong Zhang b = (Mat_SeqAIJ*)(fact)->data; 16961df811f5SHong Zhang b->row = isrow; 16971df811f5SHong Zhang b->col = iscol; 16981df811f5SHong Zhang b->icol = isicol; 16991df811f5SHong Zhang ierr = PetscMalloc((fact->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 17001df811f5SHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 17011df811f5SHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1702813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1703813a1f2bSShri Abhyankar } 1704813a1f2bSShri Abhyankar 1705813a1f2bSShri Abhyankar #undef __FUNCT__ 170635233d99SShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ" 170735233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1708813a1f2bSShri Abhyankar { 1709813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1710813a1f2bSShri Abhyankar IS isicol; 1711813a1f2bSShri Abhyankar PetscErrorCode ierr; 1712813a1f2bSShri Abhyankar const PetscInt *r,*ic; 17131df811f5SHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j; 1714813a1f2bSShri Abhyankar PetscInt *bi,*cols,nnz,*cols_lvl; 1715813a1f2bSShri Abhyankar PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 1716813a1f2bSShri Abhyankar PetscInt i,levels,diagonal_fill; 1717813a1f2bSShri Abhyankar PetscTruth col_identity,row_identity; 1718813a1f2bSShri Abhyankar PetscReal f; 1719813a1f2bSShri Abhyankar PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 1720813a1f2bSShri Abhyankar PetscBT lnkbt; 1721813a1f2bSShri Abhyankar PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 1722813a1f2bSShri Abhyankar PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1723813a1f2bSShri Abhyankar PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 1724813a1f2bSShri Abhyankar 1725813a1f2bSShri Abhyankar PetscFunctionBegin; 17263898ab1dSSatish Balay /* Uncomment the old data struct part only while testing new data structure for MatSolve() */ 1727124a3351SHong Zhang /* 1728124a3351SHong Zhang PetscTruth olddatastruct=PETSC_FALSE; 172935233d99SShri Abhyankar ierr = PetscOptionsGetTruth(PETSC_NULL,"-ilu_old",&olddatastruct,PETSC_NULL);CHKERRQ(ierr); 173035233d99SShri Abhyankar if(olddatastruct){ 1731ad04f41aSHong Zhang ierr = MatILUFactorSymbolic_SeqAIJ_inplace(fact,A,isrow,iscol,info);CHKERRQ(ierr); 1732ad04f41aSHong Zhang PetscFunctionReturn(0); 1733ad04f41aSHong Zhang } 1734124a3351SHong Zhang */ 1735ad04f41aSHong Zhang 1736813a1f2bSShri Abhyankar levels = (PetscInt)info->levels; 1737813a1f2bSShri Abhyankar ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 1738813a1f2bSShri Abhyankar ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 1739813a1f2bSShri Abhyankar 1740813a1f2bSShri Abhyankar if (!levels && row_identity && col_identity) { 1741813a1f2bSShri Abhyankar /* special case: ilu(0) with natural ordering */ 174235233d99SShri Abhyankar ierr = MatILUFactorSymbolic_SeqAIJ_ilu0(fact,A,isrow,iscol,info);CHKERRQ(ierr); 1743124a3351SHong Zhang if (a->inode.size) { 1744124a3351SHong Zhang fact->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode; 1745124a3351SHong Zhang } 1746813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1747813a1f2bSShri Abhyankar } 1748813a1f2bSShri Abhyankar 1749e32f2f54SBarry Smith if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 17501df811f5SHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 1751813a1f2bSShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1752813a1f2bSShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 1753813a1f2bSShri Abhyankar 17541bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 17551bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 17561bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 1757e60cf9a0SBarry Smith bi[0] = bdiag[0] = 0; 1758813a1f2bSShri Abhyankar 17590e83c824SBarry Smith ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr); 1760813a1f2bSShri Abhyankar 1761813a1f2bSShri Abhyankar /* create a linked list for storing column indices of the active row */ 1762813a1f2bSShri Abhyankar nlnk = n + 1; 1763813a1f2bSShri Abhyankar ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 1764813a1f2bSShri Abhyankar 1765813a1f2bSShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 17661df811f5SHong Zhang f = info->fill; 17671df811f5SHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 1768813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 1769813a1f2bSShri Abhyankar current_space = free_space; 1770813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 1771813a1f2bSShri Abhyankar current_space_lvl = free_space_lvl; 1772813a1f2bSShri Abhyankar 1773813a1f2bSShri Abhyankar for (i=0; i<n; i++) { 1774813a1f2bSShri Abhyankar nzi = 0; 1775813a1f2bSShri Abhyankar /* copy current row into linked list */ 1776813a1f2bSShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 1777e32f2f54SBarry Smith if (!nnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 1778813a1f2bSShri Abhyankar cols = aj + ai[r[i]]; 1779813a1f2bSShri Abhyankar lnk[i] = -1; /* marker to indicate if diagonal exists */ 1780813a1f2bSShri Abhyankar ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 1781813a1f2bSShri Abhyankar nzi += nlnk; 1782813a1f2bSShri Abhyankar 1783813a1f2bSShri Abhyankar /* make sure diagonal entry is included */ 1784813a1f2bSShri Abhyankar if (diagonal_fill && lnk[i] == -1) { 1785813a1f2bSShri Abhyankar fm = n; 1786813a1f2bSShri Abhyankar while (lnk[fm] < i) fm = lnk[fm]; 1787813a1f2bSShri Abhyankar lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 1788813a1f2bSShri Abhyankar lnk[fm] = i; 1789813a1f2bSShri Abhyankar lnk_lvl[i] = 0; 1790813a1f2bSShri Abhyankar nzi++; dcount++; 1791813a1f2bSShri Abhyankar } 1792813a1f2bSShri Abhyankar 1793813a1f2bSShri Abhyankar /* add pivot rows into the active row */ 1794813a1f2bSShri Abhyankar nzbd = 0; 1795813a1f2bSShri Abhyankar prow = lnk[n]; 1796813a1f2bSShri Abhyankar while (prow < i) { 1797813a1f2bSShri Abhyankar nnz = bdiag[prow]; 1798813a1f2bSShri Abhyankar cols = bj_ptr[prow] + nnz + 1; 1799813a1f2bSShri Abhyankar cols_lvl = bjlvl_ptr[prow] + nnz + 1; 1800813a1f2bSShri Abhyankar nnz = bi[prow+1] - bi[prow] - nnz - 1; 1801813a1f2bSShri Abhyankar ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 1802813a1f2bSShri Abhyankar nzi += nlnk; 1803813a1f2bSShri Abhyankar prow = lnk[prow]; 1804813a1f2bSShri Abhyankar nzbd++; 1805813a1f2bSShri Abhyankar } 1806813a1f2bSShri Abhyankar bdiag[i] = nzbd; 1807813a1f2bSShri Abhyankar bi[i+1] = bi[i] + nzi; 1808813a1f2bSShri Abhyankar 1809813a1f2bSShri Abhyankar /* if free space is not available, make more free space */ 1810813a1f2bSShri Abhyankar if (current_space->local_remaining<nzi) { 1811813a1f2bSShri Abhyankar nnz = 2*nzi*(n - i); /* estimated and max additional space needed */ 1812813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 1813813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 1814813a1f2bSShri Abhyankar reallocs++; 1815813a1f2bSShri Abhyankar } 1816813a1f2bSShri Abhyankar 1817813a1f2bSShri Abhyankar /* copy data into free_space and free_space_lvl, then initialize lnk */ 1818813a1f2bSShri Abhyankar ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 1819813a1f2bSShri Abhyankar bj_ptr[i] = current_space->array; 1820813a1f2bSShri Abhyankar bjlvl_ptr[i] = current_space_lvl->array; 1821813a1f2bSShri Abhyankar 1822813a1f2bSShri Abhyankar /* make sure the active row i has diagonal entry */ 182365e19b50SBarry Smith if (*(bj_ptr[i]+bdiag[i]) != i) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\ntry running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i); 1824813a1f2bSShri Abhyankar 1825813a1f2bSShri Abhyankar current_space->array += nzi; 1826813a1f2bSShri Abhyankar current_space->local_used += nzi; 1827813a1f2bSShri Abhyankar current_space->local_remaining -= nzi; 1828813a1f2bSShri Abhyankar current_space_lvl->array += nzi; 1829813a1f2bSShri Abhyankar current_space_lvl->local_used += nzi; 1830813a1f2bSShri Abhyankar current_space_lvl->local_remaining -= nzi; 1831813a1f2bSShri Abhyankar } 1832813a1f2bSShri Abhyankar 1833813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 1834813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 1835813a1f2bSShri Abhyankar 1836813a1f2bSShri Abhyankar /* destroy list of free space and other temporary arrays */ 1837813a1f2bSShri Abhyankar ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 1838813a1f2bSShri Abhyankar 1839813a1f2bSShri Abhyankar /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */ 18402ce24eb6SHong Zhang ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 1841813a1f2bSShri Abhyankar 1842813a1f2bSShri Abhyankar ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 1843813a1f2bSShri Abhyankar ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 18440e83c824SBarry Smith ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr); 1845813a1f2bSShri Abhyankar 1846813a1f2bSShri Abhyankar #if defined(PETSC_USE_INFO) 1847813a1f2bSShri Abhyankar { 1848aef85c9fSShri Abhyankar PetscReal af = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 1849813a1f2bSShri Abhyankar ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 1850813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 1851813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 1852813a1f2bSShri Abhyankar ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 1853813a1f2bSShri Abhyankar if (diagonal_fill) { 1854813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 1855813a1f2bSShri Abhyankar } 1856813a1f2bSShri Abhyankar } 1857813a1f2bSShri Abhyankar #endif 1858813a1f2bSShri Abhyankar 1859813a1f2bSShri Abhyankar /* put together the new matrix */ 1860813a1f2bSShri Abhyankar ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 1861813a1f2bSShri Abhyankar ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 1862813a1f2bSShri Abhyankar b = (Mat_SeqAIJ*)(fact)->data; 1863813a1f2bSShri Abhyankar b->free_a = PETSC_TRUE; 1864813a1f2bSShri Abhyankar b->free_ij = PETSC_TRUE; 1865813a1f2bSShri Abhyankar b->singlemalloc = PETSC_FALSE; 1866f268cba8SShri Abhyankar ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 1867813a1f2bSShri Abhyankar b->j = bj; 1868813a1f2bSShri Abhyankar b->i = bi; 1869813a1f2bSShri Abhyankar b->diag = bdiag; 1870813a1f2bSShri Abhyankar b->ilen = 0; 1871813a1f2bSShri Abhyankar b->imax = 0; 1872813a1f2bSShri Abhyankar b->row = isrow; 1873813a1f2bSShri Abhyankar b->col = iscol; 1874813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1875813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1876813a1f2bSShri Abhyankar b->icol = isicol; 1877813a1f2bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1878813a1f2bSShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. 1879813a1f2bSShri Abhyankar Allocate bdiag, solve_work, new a, new j */ 1880f268cba8SShri Abhyankar ierr = PetscLogObjectMemory(fact,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 1881f268cba8SShri Abhyankar b->maxnz = b->nz = bdiag[0]+1; 1882813a1f2bSShri Abhyankar (fact)->info.factor_mallocs = reallocs; 1883813a1f2bSShri Abhyankar (fact)->info.fill_ratio_given = f; 1884f268cba8SShri Abhyankar (fact)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 188535233d99SShri Abhyankar (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 1886f2344125SBarry Smith if (a->inode.size) { 1887f2344125SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode; 1888f2344125SBarry Smith } 1889813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1890813a1f2bSShri Abhyankar } 1891813a1f2bSShri Abhyankar 18926516239fSHong Zhang #undef __FUNCT__ 1893ad04f41aSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_inplace" 1894ad04f41aSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 18956516239fSHong Zhang { 18966516239fSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 18976516239fSHong Zhang IS isicol; 18986516239fSHong Zhang PetscErrorCode ierr; 18996516239fSHong Zhang const PetscInt *r,*ic; 19006516239fSHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,d; 19016516239fSHong Zhang PetscInt *bi,*cols,nnz,*cols_lvl; 19026516239fSHong Zhang PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 19036516239fSHong Zhang PetscInt i,levels,diagonal_fill; 19046516239fSHong Zhang PetscTruth col_identity,row_identity; 19056516239fSHong Zhang PetscReal f; 19066516239fSHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 19076516239fSHong Zhang PetscBT lnkbt; 19086516239fSHong Zhang PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 19096516239fSHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 19106516239fSHong Zhang PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 19116516239fSHong Zhang PetscTruth missing; 19126516239fSHong Zhang 19136516239fSHong Zhang PetscFunctionBegin; 1914e32f2f54SBarry Smith if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 19156516239fSHong Zhang f = info->fill; 19166516239fSHong Zhang levels = (PetscInt)info->levels; 19176516239fSHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 19186516239fSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 19196516239fSHong Zhang 19206516239fSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 19216516239fSHong Zhang ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 19226516239fSHong Zhang if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */ 1923f77e22a1SHong Zhang ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr); 1924ad04f41aSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace; 1925f2344125SBarry Smith if (a->inode.size) { 1926f2344125SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace; 1927f2344125SBarry Smith } 1928d5f3da31SBarry Smith fact->factortype = MAT_FACTOR_ILU; 1929719d5645SBarry Smith (fact)->info.factor_mallocs = 0; 1930719d5645SBarry Smith (fact)->info.fill_ratio_given = info->fill; 1931719d5645SBarry Smith (fact)->info.fill_ratio_needed = 1.0; 1932719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 19338ef3462fSBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 1934e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 193508480c60SBarry Smith b->row = isrow; 193608480c60SBarry Smith b->col = iscol; 193782bf6240SBarry Smith b->icol = isicol; 1938719d5645SBarry Smith ierr = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1939c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1940c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 19413a40ed3dSBarry Smith PetscFunctionReturn(0); 194208480c60SBarry Smith } 194308480c60SBarry Smith 1944898183ecSLois Curfman McInnes ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1945898183ecSLois Curfman McInnes ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 19469e25ed09SBarry Smith 19471bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 19481bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 19491bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 1950a6df321fSShri Abhyankar bi[0] = bdiag[0] = 0; 19516cf997b0SBarry Smith 19520e83c824SBarry Smith ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr); 19535a8e39fbSHong Zhang 19545a8e39fbSHong Zhang /* create a linked list for storing column indices of the active row */ 19555a8e39fbSHong Zhang nlnk = n + 1; 19565a8e39fbSHong Zhang ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 19575a8e39fbSHong Zhang 19585a8e39fbSHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 1959a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 19605a8e39fbSHong Zhang current_space = free_space; 1961a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 19625a8e39fbSHong Zhang current_space_lvl = free_space_lvl; 19635a8e39fbSHong Zhang 19645a8e39fbSHong Zhang for (i=0; i<n; i++) { 19655a8e39fbSHong Zhang nzi = 0; 19666cf997b0SBarry Smith /* copy current row into linked list */ 19675a8e39fbSHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 1968e32f2f54SBarry Smith if (!nnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 19695a8e39fbSHong Zhang cols = aj + ai[r[i]]; 19705a8e39fbSHong Zhang lnk[i] = -1; /* marker to indicate if diagonal exists */ 19715a8e39fbSHong Zhang ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 19725a8e39fbSHong Zhang nzi += nlnk; 19736cf997b0SBarry Smith 19746cf997b0SBarry Smith /* make sure diagonal entry is included */ 19755a8e39fbSHong Zhang if (diagonal_fill && lnk[i] == -1) { 19766cf997b0SBarry Smith fm = n; 19775a8e39fbSHong Zhang while (lnk[fm] < i) fm = lnk[fm]; 19785a8e39fbSHong Zhang lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 19795a8e39fbSHong Zhang lnk[fm] = i; 19805a8e39fbSHong Zhang lnk_lvl[i] = 0; 19815a8e39fbSHong Zhang nzi++; dcount++; 19826cf997b0SBarry Smith } 19836cf997b0SBarry Smith 19845a8e39fbSHong Zhang /* add pivot rows into the active row */ 19855a8e39fbSHong Zhang nzbd = 0; 19865a8e39fbSHong Zhang prow = lnk[n]; 19875a8e39fbSHong Zhang while (prow < i) { 19885a8e39fbSHong Zhang nnz = bdiag[prow]; 19895a8e39fbSHong Zhang cols = bj_ptr[prow] + nnz + 1; 19905a8e39fbSHong Zhang cols_lvl = bjlvl_ptr[prow] + nnz + 1; 19915a8e39fbSHong Zhang nnz = bi[prow+1] - bi[prow] - nnz - 1; 19925a8e39fbSHong Zhang ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 19935a8e39fbSHong Zhang nzi += nlnk; 19945a8e39fbSHong Zhang prow = lnk[prow]; 19955a8e39fbSHong Zhang nzbd++; 199656beaf04SBarry Smith } 19975a8e39fbSHong Zhang bdiag[i] = nzbd; 19985a8e39fbSHong Zhang bi[i+1] = bi[i] + nzi; 1999ecf371e4SBarry Smith 20005a8e39fbSHong Zhang /* if free space is not available, make more free space */ 20015a8e39fbSHong Zhang if (current_space->local_remaining<nzi) { 20025a8e39fbSHong Zhang nnz = nzi*(n - i); /* estimated and max additional space needed */ 2003a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 2004a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 20055a8e39fbSHong Zhang reallocs++; 20065a8e39fbSHong Zhang } 2007ecf371e4SBarry Smith 20085a8e39fbSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 20095a8e39fbSHong Zhang ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 20105a8e39fbSHong Zhang bj_ptr[i] = current_space->array; 20115a8e39fbSHong Zhang bjlvl_ptr[i] = current_space_lvl->array; 20125a8e39fbSHong Zhang 20135a8e39fbSHong Zhang /* make sure the active row i has diagonal entry */ 201465e19b50SBarry Smith if (*(bj_ptr[i]+bdiag[i]) != i) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\ntry running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i); 20155a8e39fbSHong Zhang 20165a8e39fbSHong Zhang current_space->array += nzi; 20175a8e39fbSHong Zhang current_space->local_used += nzi; 20185a8e39fbSHong Zhang current_space->local_remaining -= nzi; 20195a8e39fbSHong Zhang current_space_lvl->array += nzi; 20205a8e39fbSHong Zhang current_space_lvl->local_used += nzi; 20215a8e39fbSHong Zhang current_space_lvl->local_remaining -= nzi; 20229e25ed09SBarry Smith } 20235a8e39fbSHong Zhang 2024898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 2025898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 20265a8e39fbSHong Zhang 20275a8e39fbSHong Zhang /* destroy list of free space and other temporary arrays */ 20285a8e39fbSHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 20296516239fSHong Zhang ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */ 20305a8e39fbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2031a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 20320e83c824SBarry Smith ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr); 20339e25ed09SBarry Smith 20346cf91177SBarry Smith #if defined(PETSC_USE_INFO) 2035f64065bbSBarry Smith { 20365a8e39fbSHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 2037ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 2038ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2039ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 2040ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 2041335d9088SBarry Smith if (diagonal_fill) { 2042ae15b995SBarry Smith ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 2043335d9088SBarry Smith } 2044f64065bbSBarry Smith } 204563ba0a88SBarry Smith #endif 2046bbb6d6a8SBarry Smith 20479e25ed09SBarry Smith /* put together the new matrix */ 204807dbf95fSBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 2049719d5645SBarry Smith ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 2050719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 2051e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2052e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 20537c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 20540e83c824SBarry Smith ierr = PetscMalloc(bi[n]*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 20555a8e39fbSHong Zhang b->j = bj; 20565a8e39fbSHong Zhang b->i = bi; 20575a8e39fbSHong Zhang for (i=0; i<n; i++) bdiag[i] += bi[i]; 20585a8e39fbSHong Zhang b->diag = bdiag; 2059416022c9SBarry Smith b->ilen = 0; 2060416022c9SBarry Smith b->imax = 0; 2061416022c9SBarry Smith b->row = isrow; 2062416022c9SBarry Smith b->col = iscol; 2063c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 2064c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 206582bf6240SBarry Smith b->icol = isicol; 206687828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2067416022c9SBarry Smith /* In b structure: Free imax, ilen, old a, old j. 20685a8e39fbSHong Zhang Allocate bdiag, solve_work, new a, new j */ 2069719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 20705a8e39fbSHong Zhang b->maxnz = b->nz = bi[n] ; 2071719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 2072719d5645SBarry Smith (fact)->info.fill_ratio_given = f; 2073719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 2074ad04f41aSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace; 2075f2344125SBarry Smith if (a->inode.size) { 2076f2344125SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace; 2077d3ac4fa3SBarry Smith } 20783a40ed3dSBarry Smith PetscFunctionReturn(0); 20799e25ed09SBarry Smith } 2080d5d45c9bSBarry Smith 2081a6175056SHong Zhang #undef __FUNCT__ 208235233d99SShri Abhyankar #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ" 208335233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 20845f44c854SHong Zhang { 20855f44c854SHong Zhang Mat C = B; 20865f44c854SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 20875f44c854SHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 20885f44c854SHong Zhang IS ip=b->row,iip = b->icol; 20895f44c854SHong Zhang PetscErrorCode ierr; 20905f44c854SHong Zhang const PetscInt *rip,*riip; 20915f44c854SHong Zhang PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp; 20925f44c854SHong Zhang PetscInt *ai=a->i,*aj=a->j; 20935f44c854SHong Zhang PetscInt k,jmin,jmax,*c2r,*il,col,nexti,ili,nz; 20945f44c854SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 20955f44c854SHong Zhang PetscTruth perm_identity; 2096d90ac83dSHong Zhang FactorShiftCtx sctx; 209798a93161SHong Zhang PetscReal rs; 209898a93161SHong Zhang MatScalar d,*v; 209998a93161SHong Zhang 21005f44c854SHong Zhang PetscFunctionBegin; 210198a93161SHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 2102d90ac83dSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 210398a93161SHong Zhang 2104f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 210598a93161SHong Zhang sctx.shift_top = info->zeropivot; 210698a93161SHong Zhang for (i=0; i<mbs; i++) { 210798a93161SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 210898a93161SHong Zhang d = (aa)[a->diag[i]]; 210998a93161SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 211098a93161SHong Zhang v = aa+ai[i]; 211198a93161SHong Zhang nz = ai[i+1] - ai[i]; 211298a93161SHong Zhang for (j=0; j<nz; j++) 211398a93161SHong Zhang rs += PetscAbsScalar(v[j]); 211498a93161SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 211598a93161SHong Zhang } 211698a93161SHong Zhang sctx.shift_top *= 1.1; 211798a93161SHong Zhang sctx.nshift_max = 5; 211898a93161SHong Zhang sctx.shift_lo = 0.; 211998a93161SHong Zhang sctx.shift_hi = 1.; 212098a93161SHong Zhang } 21215f44c854SHong Zhang 21225f44c854SHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 21235f44c854SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 21245f44c854SHong Zhang 21255f44c854SHong Zhang /* allocate working arrays 21265f44c854SHong Zhang c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col 21275f44c854SHong 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 21285f44c854SHong Zhang */ 21290e83c824SBarry Smith ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&c2r);CHKERRQ(ierr); 21305f44c854SHong Zhang 213198a93161SHong Zhang do { 213207b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 213398a93161SHong Zhang 2134c5546cabSHong Zhang for (i=0; i<mbs; i++) c2r[i] = mbs; 21355f44c854SHong Zhang il[0] = 0; 21365f44c854SHong Zhang 21375f44c854SHong Zhang for (k = 0; k<mbs; k++){ 21385f44c854SHong Zhang /* zero rtmp */ 21395f44c854SHong Zhang nz = bi[k+1] - bi[k]; 21405f44c854SHong Zhang bjtmp = bj + bi[k]; 21415f44c854SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 21425f44c854SHong Zhang 21435f44c854SHong Zhang /* load in initial unfactored row */ 21445f44c854SHong Zhang bval = ba + bi[k]; 21455f44c854SHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 21465f44c854SHong Zhang for (j = jmin; j < jmax; j++){ 21475f44c854SHong Zhang col = riip[aj[j]]; 21485f44c854SHong Zhang if (col >= k){ /* only take upper triangular entry */ 21495f44c854SHong Zhang rtmp[col] = aa[j]; 21505f44c854SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 21515f44c854SHong Zhang } 21525f44c854SHong Zhang } 215398a93161SHong Zhang /* shift the diagonal of the matrix: ZeropivotApply() */ 215498a93161SHong Zhang rtmp[k] += sctx.shift_amount; /* shift the diagonal of the matrix */ 21555f44c854SHong Zhang 21565f44c854SHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 21575f44c854SHong Zhang dk = rtmp[k]; 21585f44c854SHong Zhang i = c2r[k]; /* first row to be added to k_th row */ 21595f44c854SHong Zhang 21605f44c854SHong Zhang while (i < k){ 21615f44c854SHong Zhang nexti = c2r[i]; /* next row to be added to k_th row */ 21625f44c854SHong Zhang 21635f44c854SHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 21645f44c854SHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 21655f44c854SHong Zhang uikdi = - ba[ili]*ba[bdiag[i]]; /* diagonal(k) */ 21665f44c854SHong Zhang dk += uikdi*ba[ili]; /* update diag[k] */ 21675f44c854SHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 21685f44c854SHong Zhang 21695f44c854SHong Zhang /* add multiple of row i to k-th row */ 21705f44c854SHong Zhang jmin = ili + 1; jmax = bi[i+1]; 21715f44c854SHong Zhang if (jmin < jmax){ 21725f44c854SHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 21735f44c854SHong Zhang /* update il and c2r for row i */ 21745f44c854SHong Zhang il[i] = jmin; 21755f44c854SHong Zhang j = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i; 21765f44c854SHong Zhang } 21775f44c854SHong Zhang i = nexti; 21785f44c854SHong Zhang } 21795f44c854SHong Zhang 21805f44c854SHong Zhang /* copy data into U(k,:) */ 218198a93161SHong Zhang rs = 0.0; 21825f44c854SHong Zhang jmin = bi[k]; jmax = bi[k+1]-1; 21835f44c854SHong Zhang if (jmin < jmax) { 21845f44c854SHong Zhang for (j=jmin; j<jmax; j++){ 218598a93161SHong Zhang col = bj[j]; ba[j] = rtmp[col]; rs += PetscAbsScalar(ba[j]); 21865f44c854SHong Zhang } 21875f44c854SHong Zhang /* add the k-th row into il and c2r */ 21885f44c854SHong Zhang il[k] = jmin; 21895f44c854SHong Zhang i = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k; 21905f44c854SHong Zhang } 219198a93161SHong Zhang 219298a93161SHong Zhang /* MatPivotCheck() */ 219398a93161SHong Zhang sctx.rs = rs; 219498a93161SHong Zhang sctx.pv = dk; 219507b50cabSHong Zhang ierr = MatPivotCheck(info,&sctx,i);CHKERRQ(ierr); 219607b50cabSHong Zhang if(sctx.newshift) break; 219798a93161SHong Zhang dk = sctx.pv; 219898a93161SHong Zhang 219998a93161SHong Zhang ba[bdiag[k]] = 1.0/dk; /* U(k,k) */ 220098a93161SHong Zhang } 220107b50cabSHong Zhang } while (sctx.newshift); 22025f44c854SHong Zhang 22030e83c824SBarry Smith ierr = PetscFree3(rtmp,il,c2r);CHKERRQ(ierr); 22045f44c854SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 22055f44c854SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 22065f44c854SHong Zhang 22075f44c854SHong Zhang ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 22085f44c854SHong Zhang if (perm_identity){ 220935233d99SShri Abhyankar B->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering; 221035233d99SShri Abhyankar B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering; 22112169352eSHong Zhang B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering; 22122169352eSHong Zhang B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering; 22135f44c854SHong Zhang } else { 221435233d99SShri Abhyankar B->ops->solve = MatSolve_SeqSBAIJ_1; 221535233d99SShri Abhyankar B->ops->solvetranspose = MatSolve_SeqSBAIJ_1; 221635233d99SShri Abhyankar B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1; 221735233d99SShri Abhyankar B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1; 22185f44c854SHong Zhang } 22195f44c854SHong Zhang 22205f44c854SHong Zhang C->assembled = PETSC_TRUE; 22215f44c854SHong Zhang C->preallocated = PETSC_TRUE; 22225f44c854SHong Zhang ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 222398a93161SHong Zhang 222498a93161SHong Zhang /* MatPivotView() */ 222598a93161SHong Zhang if (sctx.nshift){ 2226f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 222798a93161SHong Zhang ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %G, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,sctx.shift_amount,sctx.shift_fraction,sctx.shift_top);CHKERRQ(ierr); 2228f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 222998a93161SHong Zhang ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 2230f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS){ 2231b2a402fcSHong Zhang ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr); 223298a93161SHong Zhang } 223398a93161SHong Zhang } 22345f44c854SHong Zhang PetscFunctionReturn(0); 22355f44c854SHong Zhang } 22365f44c854SHong Zhang 22375f44c854SHong Zhang #undef __FUNCT__ 22380a3c351aSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ_inplace" 22390a3c351aSHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info) 2240a6175056SHong Zhang { 2241719d5645SBarry Smith Mat C = B; 22420968510aSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 22432ed133dbSHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 22449bfd6278SHong Zhang IS ip=b->row,iip = b->icol; 22452ed133dbSHong Zhang PetscErrorCode ierr; 22465d0c19d7SBarry Smith const PetscInt *rip,*riip; 2247c5546cabSHong Zhang PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol,*bjtmp; 22482ed133dbSHong Zhang PetscInt *ai=a->i,*aj=a->j; 22492ed133dbSHong Zhang PetscInt k,jmin,jmax,*jl,*il,col,nexti,ili,nz; 2250622e2028SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 2251db4efbfdSBarry Smith PetscTruth perm_identity; 22520e95ead3SHong Zhang FactorShiftCtx sctx; 22530e95ead3SHong Zhang PetscReal rs; 22540e95ead3SHong Zhang MatScalar d,*v; 2255d5d45c9bSBarry Smith 2256a6175056SHong Zhang PetscFunctionBegin; 22570e95ead3SHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 22580e95ead3SHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 22590e95ead3SHong Zhang 22600e95ead3SHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 22610e95ead3SHong Zhang sctx.shift_top = info->zeropivot; 22620e95ead3SHong Zhang for (i=0; i<mbs; i++) { 22630e95ead3SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 22640e95ead3SHong Zhang d = (aa)[a->diag[i]]; 22650e95ead3SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 22660e95ead3SHong Zhang v = aa+ai[i]; 22670e95ead3SHong Zhang nz = ai[i+1] - ai[i]; 22680e95ead3SHong Zhang for (j=0; j<nz; j++) 22690e95ead3SHong Zhang rs += PetscAbsScalar(v[j]); 22700e95ead3SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 22710e95ead3SHong Zhang } 22720e95ead3SHong Zhang sctx.shift_top *= 1.1; 22730e95ead3SHong Zhang sctx.nshift_max = 5; 22740e95ead3SHong Zhang sctx.shift_lo = 0.; 22750e95ead3SHong Zhang sctx.shift_hi = 1.; 22760e95ead3SHong Zhang } 2277ee45ca4aSHong Zhang 22782ed133dbSHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 22799bfd6278SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 22802ed133dbSHong Zhang 22812ed133dbSHong Zhang /* initialization */ 22820e83c824SBarry Smith ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&jl);CHKERRQ(ierr); 22830e95ead3SHong Zhang 22842ed133dbSHong Zhang do { 228507b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 22860e95ead3SHong Zhang 2287c5546cabSHong Zhang for (i=0; i<mbs; i++) jl[i] = mbs; 2288c5546cabSHong Zhang il[0] = 0; 22892ed133dbSHong Zhang 22902ed133dbSHong Zhang for (k = 0; k<mbs; k++){ 2291c5546cabSHong Zhang /* zero rtmp */ 2292c5546cabSHong Zhang nz = bi[k+1] - bi[k]; 2293c5546cabSHong Zhang bjtmp = bj + bi[k]; 2294c5546cabSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 2295c5546cabSHong Zhang 2296c05c3958SHong Zhang bval = ba + bi[k]; 22972ed133dbSHong Zhang /* initialize k-th row by the perm[k]-th row of A */ 22982ed133dbSHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 22992ed133dbSHong Zhang for (j = jmin; j < jmax; j++){ 23009bfd6278SHong Zhang col = riip[aj[j]]; 23012ed133dbSHong Zhang if (col >= k){ /* only take upper triangular entry */ 23022ed133dbSHong Zhang rtmp[col] = aa[j]; 2303c05c3958SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 23042ed133dbSHong Zhang } 23052ed133dbSHong Zhang } 230639e3d630SHong Zhang /* shift the diagonal of the matrix */ 2307540703acSHong Zhang if (sctx.nshift) rtmp[k] += sctx.shift_amount; 23082ed133dbSHong Zhang 23092ed133dbSHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 23102ed133dbSHong Zhang dk = rtmp[k]; 23112ed133dbSHong Zhang i = jl[k]; /* first row to be added to k_th row */ 23122ed133dbSHong Zhang 23132ed133dbSHong Zhang while (i < k){ 23142ed133dbSHong Zhang nexti = jl[i]; /* next row to be added to k_th row */ 23152ed133dbSHong Zhang 23162ed133dbSHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 23172ed133dbSHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 23182ed133dbSHong Zhang uikdi = - ba[ili]*ba[bi[i]]; /* diagonal(k) */ 23192ed133dbSHong Zhang dk += uikdi*ba[ili]; 23202ed133dbSHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 23212ed133dbSHong Zhang 23222ed133dbSHong Zhang /* add multiple of row i to k-th row */ 23232ed133dbSHong Zhang jmin = ili + 1; jmax = bi[i+1]; 23242ed133dbSHong Zhang if (jmin < jmax){ 23252ed133dbSHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 23262ed133dbSHong Zhang /* update il and jl for row i */ 23272ed133dbSHong Zhang il[i] = jmin; 23282ed133dbSHong Zhang j = bj[jmin]; jl[i] = jl[j]; jl[j] = i; 23292ed133dbSHong Zhang } 23302ed133dbSHong Zhang i = nexti; 23312ed133dbSHong Zhang } 23322ed133dbSHong Zhang 23330697b6caSHong Zhang /* shift the diagonals when zero pivot is detected */ 23340697b6caSHong Zhang /* compute rs=sum of abs(off-diagonal) */ 23350697b6caSHong Zhang rs = 0.0; 23363c9e8598SHong Zhang jmin = bi[k]+1; 23373c9e8598SHong Zhang nz = bi[k+1] - jmin; 23383c9e8598SHong Zhang bcol = bj + jmin; 233904fbf559SBarry Smith for (j=0; j<nz; j++) { 234004fbf559SBarry Smith rs += PetscAbsScalar(rtmp[bcol[j]]); 23413c9e8598SHong Zhang } 2342540703acSHong Zhang 2343540703acSHong Zhang sctx.rs = rs; 2344540703acSHong Zhang sctx.pv = dk; 234507b50cabSHong Zhang ierr = MatPivotCheck(info,&sctx,k);CHKERRQ(ierr); 234607b50cabSHong Zhang if (sctx.newshift) break; 23470e95ead3SHong Zhang dk = sctx.pv; 23483c9e8598SHong Zhang 23493c9e8598SHong Zhang /* copy data into U(k,:) */ 235039e3d630SHong Zhang ba[bi[k]] = 1.0/dk; /* U(k,k) */ 235139e3d630SHong Zhang jmin = bi[k]+1; jmax = bi[k+1]; 235239e3d630SHong Zhang if (jmin < jmax) { 235339e3d630SHong Zhang for (j=jmin; j<jmax; j++){ 2354c6ae9bfcSHong Zhang col = bj[j]; ba[j] = rtmp[col]; 23553c9e8598SHong Zhang } 235639e3d630SHong Zhang /* add the k-th row into il and jl */ 23573c9e8598SHong Zhang il[k] = jmin; 23583c9e8598SHong Zhang i = bj[jmin]; jl[k] = jl[i]; jl[i] = k; 23593c9e8598SHong Zhang } 23603c9e8598SHong Zhang } 236107b50cabSHong Zhang } while (sctx.newshift); 23620e95ead3SHong Zhang 23630e83c824SBarry Smith ierr = PetscFree3(rtmp,il,jl);CHKERRQ(ierr); 236439e3d630SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 23659bfd6278SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 2366db4efbfdSBarry Smith 2367db4efbfdSBarry Smith ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 2368db4efbfdSBarry Smith if (perm_identity){ 23690a3c351aSHong Zhang B->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 23700a3c351aSHong Zhang B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 23710a3c351aSHong Zhang B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 23720a3c351aSHong Zhang B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 2373db4efbfdSBarry Smith } else { 23740a3c351aSHong Zhang B->ops->solve = MatSolve_SeqSBAIJ_1_inplace; 23750a3c351aSHong Zhang B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_inplace; 23760a3c351aSHong Zhang B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_inplace; 23770a3c351aSHong Zhang B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_inplace; 2378db4efbfdSBarry Smith } 2379db4efbfdSBarry Smith 23803c9e8598SHong Zhang C->assembled = PETSC_TRUE; 23813c9e8598SHong Zhang C->preallocated = PETSC_TRUE; 2382d0f46423SBarry Smith ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 2383540703acSHong Zhang if (sctx.nshift){ 2384f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 23851e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 2386f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 23871e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 23880697b6caSHong Zhang } 23893c9e8598SHong Zhang } 23903c9e8598SHong Zhang PetscFunctionReturn(0); 23913c9e8598SHong Zhang } 2392a6175056SHong Zhang 23938a76ed4fSHong Zhang /* 23948a76ed4fSHong Zhang icc() under revised new data structure. 23958a76ed4fSHong Zhang Factored arrays bj and ba are stored as 23968a76ed4fSHong Zhang U(0,:),...,U(i,:),U(n-1,:) 23978a76ed4fSHong Zhang 23988a76ed4fSHong Zhang ui=fact->i is an array of size n+1, in which 23998a76ed4fSHong Zhang ui+ 24008a76ed4fSHong Zhang ui[i]: points to 1st entry of U(i,:),i=0,...,n-1 24018a76ed4fSHong Zhang ui[n]: points to U(n-1,n-1)+1 24028a76ed4fSHong Zhang 24038a76ed4fSHong Zhang udiag=fact->diag is an array of size n,in which 24048a76ed4fSHong Zhang udiag[i]: points to diagonal of U(i,:), i=0,...,n-1 24058a76ed4fSHong Zhang 24068a76ed4fSHong Zhang U(i,:) contains udiag[i] as its last entry, i.e., 24078a76ed4fSHong Zhang U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 24088a76ed4fSHong Zhang */ 24098a76ed4fSHong Zhang 2410a6175056SHong Zhang #undef __FUNCT__ 241135233d99SShri Abhyankar #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ" 241235233d99SShri Abhyankar PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2413a6175056SHong Zhang { 24140968510aSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2415ed59401aSHong Zhang Mat_SeqSBAIJ *b; 2416dfbe8321SBarry Smith PetscErrorCode ierr; 241758ebbce7SBarry Smith PetscTruth perm_identity,missing; 24185f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 24195d0c19d7SBarry Smith const PetscInt *rip,*riip; 2420ed59401aSHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 242158ebbce7SBarry Smith PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 24225a8e39fbSHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 2423ed59401aSHong Zhang PetscReal fill=info->fill,levels=info->levels; 2424a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 2425a1a86e44SBarry Smith PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 24267a48dd6fSHong Zhang PetscBT lnkbt; 2427b635d36bSHong Zhang IS iperm; 2428a6175056SHong Zhang 2429a6175056SHong Zhang PetscFunctionBegin; 2430e32f2f54SBarry Smith if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 243158ebbce7SBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 2432e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 2433653a6975SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 2434b635d36bSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 24357a48dd6fSHong Zhang 243639e3d630SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 24375f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 243839e3d630SHong Zhang ui[0] = 0; 243939e3d630SHong Zhang 24408a76ed4fSHong Zhang /* ICC(0) without matrix ordering: simply rearrange column indices */ 2441622e2028SHong Zhang if (!levels && perm_identity) { 24425f44c854SHong Zhang for (i=0; i<am; i++) { 2443c5546cabSHong Zhang ncols = ai[i+1] - a->diag[i]; 2444c5546cabSHong Zhang ui[i+1] = ui[i] + ncols; 2445c5546cabSHong Zhang udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */ 2446dc1e2a3fSHong Zhang } 2447dc1e2a3fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2448dc1e2a3fSHong Zhang cols = uj; 2449dc1e2a3fSHong Zhang for (i=0; i<am; i++) { 24505f44c854SHong Zhang aj = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */ 2451dc1e2a3fSHong Zhang ncols = ai[i+1] - a->diag[i] -1; 24525f44c854SHong Zhang for (j=0; j<ncols; j++) *cols++ = aj[j]; 24535f44c854SHong Zhang *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */ 24545f44c854SHong Zhang } 24555f44c854SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 24565f44c854SHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 24575f44c854SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 24585f44c854SHong Zhang 24595f44c854SHong Zhang /* initialization */ 24605f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 24615f44c854SHong Zhang 24625f44c854SHong Zhang /* jl: linked list for storing indices of the pivot rows 24635f44c854SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 24640e83c824SBarry Smith ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr); 24655f44c854SHong Zhang for (i=0; i<am; i++){ 24665f44c854SHong Zhang jl[i] = am; il[i] = 0; 24675f44c854SHong Zhang } 24685f44c854SHong Zhang 24695f44c854SHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 24705f44c854SHong Zhang nlnk = am + 1; 24715f44c854SHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 24725f44c854SHong Zhang 24735f44c854SHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 24745f44c854SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 24755f44c854SHong Zhang current_space = free_space; 24765f44c854SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr); 24775f44c854SHong Zhang current_space_lvl = free_space_lvl; 24785f44c854SHong Zhang 24795f44c854SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 24805f44c854SHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 24815f44c854SHong Zhang nzk = 0; 24825f44c854SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 2483e32f2f54SBarry Smith if (!ncols) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k); 24845f44c854SHong Zhang ncols_upper = 0; 24855f44c854SHong Zhang for (j=0; j<ncols; j++){ 24865f44c854SHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 24875f44c854SHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 24885f44c854SHong Zhang ajtmp[ncols_upper] = i; 24895f44c854SHong Zhang ncols_upper++; 24905f44c854SHong Zhang } 24915f44c854SHong Zhang } 24925f44c854SHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 24935f44c854SHong Zhang nzk += nlnk; 24945f44c854SHong Zhang 24955f44c854SHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 24965f44c854SHong Zhang prow = jl[k]; /* 1st pivot row */ 24975f44c854SHong Zhang 24985f44c854SHong Zhang while (prow < k){ 24995f44c854SHong Zhang nextprow = jl[prow]; 25005f44c854SHong Zhang 25015f44c854SHong Zhang /* merge prow into k-th row */ 25025f44c854SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 25035f44c854SHong Zhang jmax = ui[prow+1]; 25045f44c854SHong Zhang ncols = jmax-jmin; 25055f44c854SHong Zhang i = jmin - ui[prow]; 25065f44c854SHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 25075f44c854SHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 25085f44c854SHong Zhang j = *(uj - 1); 25095f44c854SHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 25105f44c854SHong Zhang nzk += nlnk; 25115f44c854SHong Zhang 25125f44c854SHong Zhang /* update il and jl for prow */ 25135f44c854SHong Zhang if (jmin < jmax){ 25145f44c854SHong Zhang il[prow] = jmin; 25155f44c854SHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 25165f44c854SHong Zhang } 25175f44c854SHong Zhang prow = nextprow; 25185f44c854SHong Zhang } 25195f44c854SHong Zhang 25205f44c854SHong Zhang /* if free space is not available, make more free space */ 25215f44c854SHong Zhang if (current_space->local_remaining<nzk) { 25225f44c854SHong Zhang i = am - k + 1; /* num of unfactored rows */ 2523c6ae9bfcSHong Zhang i *= PetscMin(nzk, i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */ 25245f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 25255f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 25265f44c854SHong Zhang reallocs++; 25275f44c854SHong Zhang } 25285f44c854SHong Zhang 25295f44c854SHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 2530e32f2f54SBarry Smith if (nzk == 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 25315f44c854SHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 25325f44c854SHong Zhang 25335f44c854SHong Zhang /* add the k-th row into il and jl */ 25345f44c854SHong Zhang if (nzk > 1){ 25355f44c854SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 25365f44c854SHong Zhang jl[k] = jl[i]; jl[i] = k; 25375f44c854SHong Zhang il[k] = ui[k] + 1; 25385f44c854SHong Zhang } 25395f44c854SHong Zhang uj_ptr[k] = current_space->array; 25405f44c854SHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 25415f44c854SHong Zhang 25425f44c854SHong Zhang current_space->array += nzk; 25435f44c854SHong Zhang current_space->local_used += nzk; 25445f44c854SHong Zhang current_space->local_remaining -= nzk; 25455f44c854SHong Zhang 25465f44c854SHong Zhang current_space_lvl->array += nzk; 25475f44c854SHong Zhang current_space_lvl->local_used += nzk; 25485f44c854SHong Zhang current_space_lvl->local_remaining -= nzk; 25495f44c854SHong Zhang 25505f44c854SHong Zhang ui[k+1] = ui[k] + nzk; 25515f44c854SHong Zhang } 25525f44c854SHong Zhang 25535f44c854SHong Zhang #if defined(PETSC_USE_INFO) 25545f44c854SHong Zhang if (ai[am] != 0) { 25555f44c854SHong Zhang PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]); 25565f44c854SHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 25575f44c854SHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 25585f44c854SHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 25595f44c854SHong Zhang } else { 25605f44c854SHong Zhang ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 25615f44c854SHong Zhang } 25625f44c854SHong Zhang #endif 25635f44c854SHong Zhang 25645f44c854SHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 25655f44c854SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 25660e83c824SBarry Smith ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr); 25675f44c854SHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 25685f44c854SHong Zhang 25695f44c854SHong Zhang /* destroy list of free space and other temporary array(s) */ 25705f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 257135233d99SShri Abhyankar ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor */ 25725f44c854SHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 25735f44c854SHong Zhang ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 25745f44c854SHong Zhang 25755f44c854SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 25765f44c854SHong Zhang 25775f44c854SHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 25785f44c854SHong Zhang b = (Mat_SeqSBAIJ*)(fact)->data; 25795f44c854SHong Zhang b->singlemalloc = PETSC_FALSE; 25805f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 25815f44c854SHong Zhang b->j = uj; 25825f44c854SHong Zhang b->i = ui; 25835f44c854SHong Zhang b->diag = udiag; 25847f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 25855f44c854SHong Zhang b->ilen = 0; 25865f44c854SHong Zhang b->imax = 0; 25875f44c854SHong Zhang b->row = perm; 25885f44c854SHong Zhang b->col = perm; 25895f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 25905f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 25915f44c854SHong Zhang b->icol = iperm; 25925f44c854SHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 25935f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2594d595f711SHong Zhang ierr = PetscLogObjectMemory(fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 25955f44c854SHong Zhang b->maxnz = b->nz = ui[am]; 25965f44c854SHong Zhang b->free_a = PETSC_TRUE; 25975f44c854SHong Zhang b->free_ij = PETSC_TRUE; 25985f44c854SHong Zhang 2599f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 2600f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 26015f44c854SHong Zhang if (ai[am] != 0) { 2602f284f12aSHong Zhang fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 26035f44c854SHong Zhang } else { 2604f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 26055f44c854SHong Zhang } 260635233d99SShri Abhyankar fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 26075f44c854SHong Zhang PetscFunctionReturn(0); 26085f44c854SHong Zhang } 26095f44c854SHong Zhang 26105f44c854SHong Zhang #undef __FUNCT__ 2611ad04f41aSHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ_inplace" 2612ad04f41aSHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 26135f44c854SHong Zhang { 26145f44c854SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 26155f44c854SHong Zhang Mat_SeqSBAIJ *b; 26165f44c854SHong Zhang PetscErrorCode ierr; 26175f44c854SHong Zhang PetscTruth perm_identity,missing; 26185f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 26195f44c854SHong Zhang const PetscInt *rip,*riip; 26205f44c854SHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 26215f44c854SHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 26225f44c854SHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 26235f44c854SHong Zhang PetscReal fill=info->fill,levels=info->levels; 26245f44c854SHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 26255f44c854SHong Zhang PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 26265f44c854SHong Zhang PetscBT lnkbt; 26275f44c854SHong Zhang IS iperm; 26285f44c854SHong Zhang 26295f44c854SHong Zhang PetscFunctionBegin; 2630e32f2f54SBarry Smith if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 26315f44c854SHong Zhang ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 2632e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 26335f44c854SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 26345f44c854SHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 26355f44c854SHong Zhang 26365f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 26375f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 26385f44c854SHong Zhang ui[0] = 0; 26395f44c854SHong Zhang 26405f44c854SHong Zhang /* ICC(0) without matrix ordering: simply copies fill pattern */ 26415f44c854SHong Zhang if (!levels && perm_identity) { 26425f44c854SHong Zhang 26435f44c854SHong Zhang for (i=0; i<am; i++) { 26445f44c854SHong Zhang ui[i+1] = ui[i] + ai[i+1] - a->diag[i]; 26455f44c854SHong Zhang udiag[i] = ui[i]; 2646ed59401aSHong Zhang } 264739e3d630SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 264839e3d630SHong Zhang cols = uj; 2649ed59401aSHong Zhang for (i=0; i<am; i++) { 2650ed59401aSHong Zhang aj = a->j + a->diag[i]; 265139e3d630SHong Zhang ncols = ui[i+1] - ui[i]; 265239e3d630SHong Zhang for (j=0; j<ncols; j++) *cols++ = *aj++; 2653ed59401aSHong Zhang } 265439e3d630SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 2655b635d36bSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 2656b635d36bSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 2657b635d36bSHong Zhang 26587a48dd6fSHong Zhang /* initialization */ 26595a8e39fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 26607a48dd6fSHong Zhang 26617a48dd6fSHong Zhang /* jl: linked list for storing indices of the pivot rows 26627a48dd6fSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 26630e83c824SBarry Smith ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr); 26647a48dd6fSHong Zhang for (i=0; i<am; i++){ 26657a48dd6fSHong Zhang jl[i] = am; il[i] = 0; 26667a48dd6fSHong Zhang } 26677a48dd6fSHong Zhang 26687a48dd6fSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 26697a48dd6fSHong Zhang nlnk = am + 1; 26702ed133dbSHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 26717a48dd6fSHong Zhang 26727a48dd6fSHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 2673a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 26747a48dd6fSHong Zhang current_space = free_space; 2675a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr); 26767a48dd6fSHong Zhang current_space_lvl = free_space_lvl; 26777a48dd6fSHong Zhang 26787a48dd6fSHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 26797a48dd6fSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 26807a48dd6fSHong Zhang nzk = 0; 2681622e2028SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 2682e32f2f54SBarry Smith if (!ncols) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k); 2683622e2028SHong Zhang ncols_upper = 0; 2684622e2028SHong Zhang for (j=0; j<ncols; j++){ 2685b635d36bSHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 2686b635d36bSHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 26875a8e39fbSHong Zhang ajtmp[ncols_upper] = i; 2688622e2028SHong Zhang ncols_upper++; 2689622e2028SHong Zhang } 2690622e2028SHong Zhang } 2691b635d36bSHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 26927a48dd6fSHong Zhang nzk += nlnk; 26937a48dd6fSHong Zhang 26947a48dd6fSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 26957a48dd6fSHong Zhang prow = jl[k]; /* 1st pivot row */ 26967a48dd6fSHong Zhang 26977a48dd6fSHong Zhang while (prow < k){ 26987a48dd6fSHong Zhang nextprow = jl[prow]; 26997a48dd6fSHong Zhang 27007a48dd6fSHong Zhang /* merge prow into k-th row */ 27017a48dd6fSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 27027a48dd6fSHong Zhang jmax = ui[prow+1]; 27037a48dd6fSHong Zhang ncols = jmax-jmin; 2704ed59401aSHong Zhang i = jmin - ui[prow]; 2705ed59401aSHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 27065a8e39fbSHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 27075a8e39fbSHong Zhang j = *(uj - 1); 27085a8e39fbSHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 27097a48dd6fSHong Zhang nzk += nlnk; 27107a48dd6fSHong Zhang 27117a48dd6fSHong Zhang /* update il and jl for prow */ 27127a48dd6fSHong Zhang if (jmin < jmax){ 27137a48dd6fSHong Zhang il[prow] = jmin; 27147a48dd6fSHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 27157a48dd6fSHong Zhang } 27167a48dd6fSHong Zhang prow = nextprow; 27177a48dd6fSHong Zhang } 27187a48dd6fSHong Zhang 27197a48dd6fSHong Zhang /* if free space is not available, make more free space */ 27207a48dd6fSHong Zhang if (current_space->local_remaining<nzk) { 27217a48dd6fSHong Zhang i = am - k + 1; /* num of unfactored rows */ 2722fd3f55cbSMatthew Knepley i *= PetscMin(nzk, (i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 2723a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 2724a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 27257a48dd6fSHong Zhang reallocs++; 27267a48dd6fSHong Zhang } 27277a48dd6fSHong Zhang 27287a48dd6fSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 272965e19b50SBarry Smith if (!nzk) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 27302ed133dbSHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 27317a48dd6fSHong Zhang 27327a48dd6fSHong Zhang /* add the k-th row into il and jl */ 273339e3d630SHong Zhang if (nzk > 1){ 27347a48dd6fSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 27357a48dd6fSHong Zhang jl[k] = jl[i]; jl[i] = k; 27367a48dd6fSHong Zhang il[k] = ui[k] + 1; 27377a48dd6fSHong Zhang } 27387a48dd6fSHong Zhang uj_ptr[k] = current_space->array; 27397a48dd6fSHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 27407a48dd6fSHong Zhang 27417a48dd6fSHong Zhang current_space->array += nzk; 27427a48dd6fSHong Zhang current_space->local_used += nzk; 27437a48dd6fSHong Zhang current_space->local_remaining -= nzk; 27447a48dd6fSHong Zhang 27457a48dd6fSHong Zhang current_space_lvl->array += nzk; 27467a48dd6fSHong Zhang current_space_lvl->local_used += nzk; 27477a48dd6fSHong Zhang current_space_lvl->local_remaining -= nzk; 27487a48dd6fSHong Zhang 27497a48dd6fSHong Zhang ui[k+1] = ui[k] + nzk; 27507a48dd6fSHong Zhang } 27517a48dd6fSHong Zhang 27526cf91177SBarry Smith #if defined(PETSC_USE_INFO) 27537a48dd6fSHong Zhang if (ai[am] != 0) { 275439e3d630SHong Zhang PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]); 2755ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 2756ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2757ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 27587a48dd6fSHong Zhang } else { 2759ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 27607a48dd6fSHong Zhang } 276163ba0a88SBarry Smith #endif 27627a48dd6fSHong Zhang 27637a48dd6fSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 2764b635d36bSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 27650e83c824SBarry Smith ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr); 27665a8e39fbSHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 27677a48dd6fSHong Zhang 27687a48dd6fSHong Zhang /* destroy list of free space and other temporary array(s) */ 27697a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2770a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 27712ed133dbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2772a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 27737a48dd6fSHong Zhang 277439e3d630SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 277539e3d630SHong Zhang 27767a48dd6fSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 27777a48dd6fSHong Zhang 2778f284f12aSHong Zhang b = (Mat_SeqSBAIJ*)fact->data; 27797a48dd6fSHong Zhang b->singlemalloc = PETSC_FALSE; 27807a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 27817a48dd6fSHong Zhang b->j = uj; 27827a48dd6fSHong Zhang b->i = ui; 27835f44c854SHong Zhang b->diag = udiag; 27847f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 27857a48dd6fSHong Zhang b->ilen = 0; 27867a48dd6fSHong Zhang b->imax = 0; 27877a48dd6fSHong Zhang b->row = perm; 2788b635d36bSHong Zhang b->col = perm; 2789b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2790b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2791b635d36bSHong Zhang b->icol = iperm; 27927a48dd6fSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 27937a48dd6fSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2794f284f12aSHong Zhang ierr = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 27957a48dd6fSHong Zhang b->maxnz = b->nz = ui[am]; 2796e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2797e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 27987a48dd6fSHong Zhang 2799f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 2800f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 28017a48dd6fSHong Zhang if (ai[am] != 0) { 2802f284f12aSHong Zhang fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 28037a48dd6fSHong Zhang } else { 2804f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 28057a48dd6fSHong Zhang } 28060a3c351aSHong Zhang fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace; 2807a6175056SHong Zhang PetscFunctionReturn(0); 2808a6175056SHong Zhang } 2809d5d45c9bSBarry Smith 281035233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 28110be760fbSHong Zhang { 28120be760fbSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 28130be760fbSHong Zhang Mat_SeqSBAIJ *b; 28140be760fbSHong Zhang PetscErrorCode ierr; 28150be760fbSHong Zhang PetscTruth perm_identity; 28160be760fbSHong Zhang PetscReal fill = info->fill; 28170be760fbSHong Zhang const PetscInt *rip,*riip; 28180be760fbSHong Zhang PetscInt i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 28190be760fbSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 28200be760fbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr,*udiag; 28210be760fbSHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 28220be760fbSHong Zhang PetscBT lnkbt; 28230be760fbSHong Zhang IS iperm; 28240be760fbSHong Zhang 28250be760fbSHong Zhang PetscFunctionBegin; 2826e32f2f54SBarry Smith if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 28270be760fbSHong Zhang /* check whether perm is the identity mapping */ 28280be760fbSHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 28290be760fbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 28300be760fbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 28310be760fbSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 28320be760fbSHong Zhang 28330be760fbSHong Zhang /* initialization */ 28340be760fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 28350be760fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 28360be760fbSHong Zhang ui[0] = 0; 28370be760fbSHong Zhang 28380be760fbSHong Zhang /* jl: linked list for storing indices of the pivot rows 28390be760fbSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 28400be760fbSHong Zhang ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr); 28410be760fbSHong Zhang for (i=0; i<am; i++){ 28420be760fbSHong Zhang jl[i] = am; il[i] = 0; 28430be760fbSHong Zhang } 28440be760fbSHong Zhang 28450be760fbSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 28460be760fbSHong Zhang nlnk = am + 1; 28470be760fbSHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28480be760fbSHong Zhang 28490be760fbSHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 28500be760fbSHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 28510be760fbSHong Zhang current_space = free_space; 28520be760fbSHong Zhang 28530be760fbSHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 28540be760fbSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 28550be760fbSHong Zhang nzk = 0; 28560be760fbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 2857e32f2f54SBarry Smith if (!ncols) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k); 28580be760fbSHong Zhang ncols_upper = 0; 28590be760fbSHong Zhang for (j=0; j<ncols; j++){ 28600be760fbSHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 28610be760fbSHong Zhang if (i >= k){ /* only take upper triangular entry */ 28620be760fbSHong Zhang cols[ncols_upper] = i; 28630be760fbSHong Zhang ncols_upper++; 28640be760fbSHong Zhang } 28650be760fbSHong Zhang } 28660be760fbSHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28670be760fbSHong Zhang nzk += nlnk; 28680be760fbSHong Zhang 28690be760fbSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 28700be760fbSHong Zhang prow = jl[k]; /* 1st pivot row */ 28710be760fbSHong Zhang 28720be760fbSHong Zhang while (prow < k){ 28730be760fbSHong Zhang nextprow = jl[prow]; 28740be760fbSHong Zhang /* merge prow into k-th row */ 28750be760fbSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 28760be760fbSHong Zhang jmax = ui[prow+1]; 28770be760fbSHong Zhang ncols = jmax-jmin; 28780be760fbSHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 28790be760fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28800be760fbSHong Zhang nzk += nlnk; 28810be760fbSHong Zhang 28820be760fbSHong Zhang /* update il and jl for prow */ 28830be760fbSHong Zhang if (jmin < jmax){ 28840be760fbSHong Zhang il[prow] = jmin; 28850be760fbSHong Zhang j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow; 28860be760fbSHong Zhang } 28870be760fbSHong Zhang prow = nextprow; 28880be760fbSHong Zhang } 28890be760fbSHong Zhang 28900be760fbSHong Zhang /* if free space is not available, make more free space */ 28910be760fbSHong Zhang if (current_space->local_remaining<nzk) { 28920be760fbSHong Zhang i = am - k + 1; /* num of unfactored rows */ 2893c6ae9bfcSHong Zhang i *= PetscMin(nzk,i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */ 28940be760fbSHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 28950be760fbSHong Zhang reallocs++; 28960be760fbSHong Zhang } 28970be760fbSHong Zhang 28980be760fbSHong Zhang /* copy data into free space, then initialize lnk */ 28990be760fbSHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 29000be760fbSHong Zhang 29010be760fbSHong Zhang /* add the k-th row into il and jl */ 29027b056e98SHong Zhang if (nzk > 1){ 29030be760fbSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 29040be760fbSHong Zhang jl[k] = jl[i]; jl[i] = k; 29050be760fbSHong Zhang il[k] = ui[k] + 1; 29060be760fbSHong Zhang } 29070be760fbSHong Zhang ui_ptr[k] = current_space->array; 29080be760fbSHong Zhang current_space->array += nzk; 29090be760fbSHong Zhang current_space->local_used += nzk; 29100be760fbSHong Zhang current_space->local_remaining -= nzk; 29110be760fbSHong Zhang 29120be760fbSHong Zhang ui[k+1] = ui[k] + nzk; 29130be760fbSHong Zhang } 29140be760fbSHong Zhang 29150be760fbSHong Zhang #if defined(PETSC_USE_INFO) 29160be760fbSHong Zhang if (ai[am] != 0) { 29170be760fbSHong Zhang PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]); 29180be760fbSHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 29190be760fbSHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 29200be760fbSHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 29210be760fbSHong Zhang } else { 29220be760fbSHong Zhang ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 29230be760fbSHong Zhang } 29240be760fbSHong Zhang #endif 29250be760fbSHong Zhang 29260be760fbSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 29270be760fbSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 29280be760fbSHong Zhang ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr); 29290be760fbSHong Zhang 29300be760fbSHong Zhang /* destroy list of free space and other temporary array(s) */ 29310be760fbSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 293235233d99SShri Abhyankar ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor */ 29330be760fbSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 29340be760fbSHong Zhang 29350be760fbSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 29360be760fbSHong Zhang 2937f284f12aSHong Zhang b = (Mat_SeqSBAIJ*)fact->data; 29380be760fbSHong Zhang b->singlemalloc = PETSC_FALSE; 29390be760fbSHong Zhang b->free_a = PETSC_TRUE; 29400be760fbSHong Zhang b->free_ij = PETSC_TRUE; 29410be760fbSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 29420be760fbSHong Zhang b->j = uj; 29430be760fbSHong Zhang b->i = ui; 29440be760fbSHong Zhang b->diag = udiag; 29450be760fbSHong Zhang b->free_diag = PETSC_TRUE; 29460be760fbSHong Zhang b->ilen = 0; 29470be760fbSHong Zhang b->imax = 0; 29480be760fbSHong Zhang b->row = perm; 29490be760fbSHong Zhang b->col = perm; 29500be760fbSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 29510be760fbSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 29520be760fbSHong Zhang b->icol = iperm; 29530be760fbSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 29540be760fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 29550860e08cSHong Zhang ierr = PetscLogObjectMemory(fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 29560be760fbSHong Zhang b->maxnz = b->nz = ui[am]; 29570be760fbSHong Zhang 2958f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 2959f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 29600be760fbSHong Zhang if (ai[am] != 0) { 2961f284f12aSHong Zhang fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 29620be760fbSHong Zhang } else { 2963f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 29640be760fbSHong Zhang } 296535233d99SShri Abhyankar fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 29660be760fbSHong Zhang PetscFunctionReturn(0); 29670be760fbSHong Zhang } 29680be760fbSHong Zhang 2969f76d2b81SHong Zhang #undef __FUNCT__ 2970ad04f41aSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ_inplace" 2971ad04f41aSHong Zhang PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2972f76d2b81SHong Zhang { 2973f76d2b81SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 297410c27e3dSHong Zhang Mat_SeqSBAIJ *b; 2975dfbe8321SBarry Smith PetscErrorCode ierr; 2976f76d2b81SHong Zhang PetscTruth perm_identity; 297710c27e3dSHong Zhang PetscReal fill = info->fill; 29785d0c19d7SBarry Smith const PetscInt *rip,*riip; 29795d0c19d7SBarry Smith PetscInt i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 298010c27e3dSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 29812ed133dbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr; 2982a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 298310c27e3dSHong Zhang PetscBT lnkbt; 29842ed133dbSHong Zhang IS iperm; 2985f76d2b81SHong Zhang 2986f76d2b81SHong Zhang PetscFunctionBegin; 2987e32f2f54SBarry Smith if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 29882ed133dbSHong Zhang /* check whether perm is the identity mapping */ 2989f76d2b81SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 29902ed133dbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 29912ed133dbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 29929bfd6278SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 299310c27e3dSHong Zhang 299410c27e3dSHong Zhang /* initialization */ 29951393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 299610c27e3dSHong Zhang ui[0] = 0; 299710c27e3dSHong Zhang 299810c27e3dSHong Zhang /* jl: linked list for storing indices of the pivot rows 29991393bc97SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 30000e83c824SBarry Smith ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr); 30011393bc97SHong Zhang for (i=0; i<am; i++){ 30021393bc97SHong Zhang jl[i] = am; il[i] = 0; 3003f76d2b81SHong Zhang } 300410c27e3dSHong Zhang 300510c27e3dSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 30061393bc97SHong Zhang nlnk = am + 1; 30071393bc97SHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 300810c27e3dSHong Zhang 30091393bc97SHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 3010a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 301110c27e3dSHong Zhang current_space = free_space; 301210c27e3dSHong Zhang 30131393bc97SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 301410c27e3dSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 301510c27e3dSHong Zhang nzk = 0; 30162ed133dbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 3017e32f2f54SBarry Smith if (!ncols) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k); 30182ed133dbSHong Zhang ncols_upper = 0; 30192ed133dbSHong Zhang for (j=0; j<ncols; j++){ 30209bfd6278SHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 30212ed133dbSHong Zhang if (i >= k){ /* only take upper triangular entry */ 30222ed133dbSHong Zhang cols[ncols_upper] = i; 30232ed133dbSHong Zhang ncols_upper++; 30242ed133dbSHong Zhang } 30252ed133dbSHong Zhang } 30261393bc97SHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 302710c27e3dSHong Zhang nzk += nlnk; 302810c27e3dSHong Zhang 302910c27e3dSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 303010c27e3dSHong Zhang prow = jl[k]; /* 1st pivot row */ 303110c27e3dSHong Zhang 303210c27e3dSHong Zhang while (prow < k){ 303310c27e3dSHong Zhang nextprow = jl[prow]; 303410c27e3dSHong Zhang /* merge prow into k-th row */ 30351393bc97SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 303610c27e3dSHong Zhang jmax = ui[prow+1]; 303710c27e3dSHong Zhang ncols = jmax-jmin; 30381393bc97SHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 30395a8e39fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 304010c27e3dSHong Zhang nzk += nlnk; 304110c27e3dSHong Zhang 304210c27e3dSHong Zhang /* update il and jl for prow */ 304310c27e3dSHong Zhang if (jmin < jmax){ 304410c27e3dSHong Zhang il[prow] = jmin; 30452ed133dbSHong Zhang j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow; 304610c27e3dSHong Zhang } 304710c27e3dSHong Zhang prow = nextprow; 304810c27e3dSHong Zhang } 304910c27e3dSHong Zhang 305010c27e3dSHong Zhang /* if free space is not available, make more free space */ 305110c27e3dSHong Zhang if (current_space->local_remaining<nzk) { 30521393bc97SHong Zhang i = am - k + 1; /* num of unfactored rows */ 305310c27e3dSHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 3054a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 305510c27e3dSHong Zhang reallocs++; 305610c27e3dSHong Zhang } 305710c27e3dSHong Zhang 305810c27e3dSHong Zhang /* copy data into free space, then initialize lnk */ 30591393bc97SHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 306010c27e3dSHong Zhang 306110c27e3dSHong Zhang /* add the k-th row into il and jl */ 306210c27e3dSHong Zhang if (nzk-1 > 0){ 30631393bc97SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 306410c27e3dSHong Zhang jl[k] = jl[i]; jl[i] = k; 306510c27e3dSHong Zhang il[k] = ui[k] + 1; 306610c27e3dSHong Zhang } 30672ed133dbSHong Zhang ui_ptr[k] = current_space->array; 306810c27e3dSHong Zhang current_space->array += nzk; 306910c27e3dSHong Zhang current_space->local_used += nzk; 307010c27e3dSHong Zhang current_space->local_remaining -= nzk; 307110c27e3dSHong Zhang 307210c27e3dSHong Zhang ui[k+1] = ui[k] + nzk; 307310c27e3dSHong Zhang } 307410c27e3dSHong Zhang 30756cf91177SBarry Smith #if defined(PETSC_USE_INFO) 30761393bc97SHong Zhang if (ai[am] != 0) { 30771393bc97SHong Zhang PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]); 3078ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 3079ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 3080ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 308110c27e3dSHong Zhang } else { 3082ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 308310c27e3dSHong Zhang } 308463ba0a88SBarry Smith #endif 308510c27e3dSHong Zhang 308610c27e3dSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 30879bfd6278SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 30880e83c824SBarry Smith ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr); 308910c27e3dSHong Zhang 309010c27e3dSHong Zhang /* destroy list of free space and other temporary array(s) */ 30911393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 3092a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 309310c27e3dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 309410c27e3dSHong Zhang 309510c27e3dSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 309610c27e3dSHong Zhang 3097f284f12aSHong Zhang b = (Mat_SeqSBAIJ*)fact->data; 309810c27e3dSHong Zhang b->singlemalloc = PETSC_FALSE; 3099e6b907acSBarry Smith b->free_a = PETSC_TRUE; 3100e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 31011393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 310210c27e3dSHong Zhang b->j = uj; 310310c27e3dSHong Zhang b->i = ui; 310410c27e3dSHong Zhang b->diag = 0; 310510c27e3dSHong Zhang b->ilen = 0; 310610c27e3dSHong Zhang b->imax = 0; 310710c27e3dSHong Zhang b->row = perm; 31089bfd6278SHong Zhang b->col = perm; 31099bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 31109bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 31119bfd6278SHong Zhang b->icol = iperm; 311210c27e3dSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 31131393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 3114719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 31151393bc97SHong Zhang b->maxnz = b->nz = ui[am]; 311610c27e3dSHong Zhang 3117f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 3118f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 31191393bc97SHong Zhang if (ai[am] != 0) { 3120f284f12aSHong Zhang fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 312110c27e3dSHong Zhang } else { 3122f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 312310c27e3dSHong Zhang } 31240a3c351aSHong Zhang fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace; 3125f76d2b81SHong Zhang PetscFunctionReturn(0); 3126f76d2b81SHong Zhang } 3127599ef60dSHong Zhang 31281d098868SHong Zhang #undef __FUNCT__ 312935233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering" 313035233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx) 3131813a1f2bSShri Abhyankar { 3132813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3133813a1f2bSShri Abhyankar PetscErrorCode ierr; 3134813a1f2bSShri Abhyankar PetscInt n = A->rmap->n; 3135813a1f2bSShri Abhyankar const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag,*vi; 3136813a1f2bSShri Abhyankar PetscScalar *x,sum; 3137813a1f2bSShri Abhyankar const PetscScalar *b; 3138813a1f2bSShri Abhyankar const MatScalar *aa = a->a,*v; 3139813a1f2bSShri Abhyankar PetscInt i,nz; 3140813a1f2bSShri Abhyankar 3141813a1f2bSShri Abhyankar PetscFunctionBegin; 3142813a1f2bSShri Abhyankar if (!n) PetscFunctionReturn(0); 3143813a1f2bSShri Abhyankar 31443649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 3145813a1f2bSShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3146813a1f2bSShri Abhyankar 3147813a1f2bSShri Abhyankar /* forward solve the lower triangular */ 3148813a1f2bSShri Abhyankar x[0] = b[0]; 3149813a1f2bSShri Abhyankar v = aa; 3150813a1f2bSShri Abhyankar vi = aj; 3151813a1f2bSShri Abhyankar for (i=1; i<n; i++) { 3152813a1f2bSShri Abhyankar nz = ai[i+1] - ai[i]; 3153813a1f2bSShri Abhyankar sum = b[i]; 3154813a1f2bSShri Abhyankar PetscSparseDenseMinusDot(sum,x,v,vi,nz); 3155813a1f2bSShri Abhyankar v += nz; 3156813a1f2bSShri Abhyankar vi += nz; 3157813a1f2bSShri Abhyankar x[i] = sum; 3158813a1f2bSShri Abhyankar } 3159813a1f2bSShri Abhyankar 3160813a1f2bSShri Abhyankar /* backward solve the upper triangular */ 316162fbd6afSShri Abhyankar for (i=n-1; i>=0; i--){ 31623c0119dfSShri Abhyankar v = aa + adiag[i+1] + 1; 31633c0119dfSShri Abhyankar vi = aj + adiag[i+1] + 1; 3164813a1f2bSShri Abhyankar nz = adiag[i] - adiag[i+1]-1; 3165813a1f2bSShri Abhyankar sum = x[i]; 3166813a1f2bSShri Abhyankar PetscSparseDenseMinusDot(sum,x,v,vi,nz); 31673c0119dfSShri Abhyankar x[i] = sum*v[nz]; /* x[i]=aa[adiag[i]]*sum; v++; */ 3168813a1f2bSShri Abhyankar } 3169813a1f2bSShri Abhyankar 3170813a1f2bSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 31713649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 3172813a1f2bSShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3173813a1f2bSShri Abhyankar PetscFunctionReturn(0); 3174813a1f2bSShri Abhyankar } 3175813a1f2bSShri Abhyankar 31761d098868SHong Zhang #undef __FUNCT__ 317735233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ" 317835233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx) 3179f268cba8SShri Abhyankar { 3180f268cba8SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3181f268cba8SShri Abhyankar IS iscol = a->col,isrow = a->row; 3182f268cba8SShri Abhyankar PetscErrorCode ierr; 3183f268cba8SShri Abhyankar PetscInt i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,*adiag = a->diag,nz; 3184f268cba8SShri Abhyankar const PetscInt *rout,*cout,*r,*c; 3185f268cba8SShri Abhyankar PetscScalar *x,*tmp,sum; 3186f268cba8SShri Abhyankar const PetscScalar *b; 3187f268cba8SShri Abhyankar const MatScalar *aa = a->a,*v; 3188f268cba8SShri Abhyankar 3189f268cba8SShri Abhyankar PetscFunctionBegin; 3190f268cba8SShri Abhyankar if (!n) PetscFunctionReturn(0); 3191f268cba8SShri Abhyankar 31923649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 3193f268cba8SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3194f268cba8SShri Abhyankar tmp = a->solve_work; 3195f268cba8SShri Abhyankar 3196f268cba8SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 3197f268cba8SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 3198f268cba8SShri Abhyankar 3199f268cba8SShri Abhyankar /* forward solve the lower triangular */ 3200f268cba8SShri Abhyankar tmp[0] = b[r[0]]; 3201f268cba8SShri Abhyankar v = aa; 3202f268cba8SShri Abhyankar vi = aj; 3203f268cba8SShri Abhyankar for (i=1; i<n; i++) { 3204f268cba8SShri Abhyankar nz = ai[i+1] - ai[i]; 3205f268cba8SShri Abhyankar sum = b[r[i]]; 3206f268cba8SShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 3207f268cba8SShri Abhyankar tmp[i] = sum; 3208f268cba8SShri Abhyankar v += nz; vi += nz; 3209f268cba8SShri Abhyankar } 3210f268cba8SShri Abhyankar 3211f268cba8SShri Abhyankar /* backward solve the upper triangular */ 3212f268cba8SShri Abhyankar for (i=n-1; i>=0; i--){ 32133c0119dfSShri Abhyankar v = aa + adiag[i+1]+1; 32143c0119dfSShri Abhyankar vi = aj + adiag[i+1]+1; 3215f268cba8SShri Abhyankar nz = adiag[i]-adiag[i+1]-1; 3216f268cba8SShri Abhyankar sum = tmp[i]; 3217f268cba8SShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 3218f268cba8SShri Abhyankar x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */ 3219f268cba8SShri Abhyankar } 3220f268cba8SShri Abhyankar 3221f268cba8SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 3222f268cba8SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 32233649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 3224f268cba8SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3225f268cba8SShri Abhyankar ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr); 3226f268cba8SShri Abhyankar PetscFunctionReturn(0); 3227f268cba8SShri Abhyankar } 3228f268cba8SShri Abhyankar 3229f268cba8SShri Abhyankar #undef __FUNCT__ 32301d098868SHong Zhang #define __FUNCT__ "MatILUDTFactor_SeqAIJ" 3231fe97e370SBarry Smith /* 3232fe97e370SBarry Smith This will get a new name and become a varient of MatILUFactor_SeqAIJ() there is no longer seperate functions in the matrix function table for dt factors 3233fe97e370SBarry Smith */ 32341d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact) 32351d098868SHong Zhang { 32361d098868SHong Zhang Mat B = *fact; 3237599ef60dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b; 3238599ef60dSHong Zhang IS isicol; 3239599ef60dSHong Zhang PetscErrorCode ierr; 32401d098868SHong Zhang const PetscInt *r,*ic; 32411d098868SHong Zhang PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag; 3242dc3a2fd3SHong Zhang PetscInt *bi,*bj,*bdiag,*bdiag_rev; 3243f61a948aSLisandro Dalcin PetscInt row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au; 32441d098868SHong Zhang PetscInt nlnk,*lnk; 32451d098868SHong Zhang PetscBT lnkbt; 32465a586d82SBarry Smith PetscTruth row_identity,icol_identity; 32478aee2decSHong Zhang MatScalar *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp; 32481d098868SHong Zhang const PetscInt *ics; 32491d098868SHong Zhang PetscInt j,nz,*pj,*bjtmp,k,ncut,*jtmp; 3250182b8fbaSHong Zhang PetscReal dt=info->dt,dtcol=info->dtcol,shift=info->shiftamount; 3251f61a948aSLisandro Dalcin PetscInt dtcount=(PetscInt)info->dtcount,nnz_max; 3252599ef60dSHong Zhang PetscTruth missing; 3253599ef60dSHong Zhang 3254599ef60dSHong Zhang PetscFunctionBegin; 3255f61a948aSLisandro Dalcin 3256f61a948aSLisandro Dalcin if (dt == PETSC_DEFAULT) dt = 0.005; 3257f61a948aSLisandro Dalcin if (dtcol == PETSC_DEFAULT) dtcol = 0.01; /* XXX unused! */ 3258f61a948aSLisandro Dalcin if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax); 3259f61a948aSLisandro Dalcin 32601d098868SHong Zhang /* ------- symbolic factorization, can be reused ---------*/ 32611d098868SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 3262e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 32631d098868SHong Zhang adiag=a->diag; 3264599ef60dSHong Zhang 3265599ef60dSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 3266599ef60dSHong Zhang 3267599ef60dSHong Zhang /* bdiag is location of diagonal in factor */ 32681bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); /* becomes b->diag */ 32691bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag_rev);CHKERRQ(ierr); /* temporary */ 3270599ef60dSHong Zhang 32711d098868SHong Zhang /* allocate row pointers bi */ 32728fc3a347SHong Zhang ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 32731d098868SHong Zhang 32741d098868SHong Zhang /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */ 3275393d3a68SHong Zhang if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */ 32761d098868SHong Zhang nnz_max = ai[n]+2*n*dtcount+2; 32778fc3a347SHong Zhang 32788fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 32798fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(MatScalar),&ba);CHKERRQ(ierr); 3280599ef60dSHong Zhang 3281599ef60dSHong Zhang /* put together the new matrix */ 3282599ef60dSHong Zhang ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 3283599ef60dSHong Zhang ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 3284f284f12aSHong Zhang b = (Mat_SeqAIJ*)B->data; 3285599ef60dSHong Zhang b->free_a = PETSC_TRUE; 3286599ef60dSHong Zhang b->free_ij = PETSC_TRUE; 3287599ef60dSHong Zhang b->singlemalloc = PETSC_FALSE; 3288599ef60dSHong Zhang b->a = ba; 3289599ef60dSHong Zhang b->j = bj; 3290599ef60dSHong Zhang b->i = bi; 3291599ef60dSHong Zhang b->diag = bdiag; 3292599ef60dSHong Zhang b->ilen = 0; 3293599ef60dSHong Zhang b->imax = 0; 3294599ef60dSHong Zhang b->row = isrow; 3295599ef60dSHong Zhang b->col = iscol; 3296599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 3297599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 3298599ef60dSHong Zhang b->icol = isicol; 3299599ef60dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 3300599ef60dSHong Zhang 33011d098868SHong Zhang ierr = PetscLogObjectMemory(B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 33021d098868SHong Zhang b->maxnz = nnz_max; 3303599ef60dSHong Zhang 3304d5f3da31SBarry Smith B->factortype = MAT_FACTOR_ILUDT; 3305f284f12aSHong Zhang B->info.factor_mallocs = 0; 3306f284f12aSHong Zhang B->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]); 33071d098868SHong Zhang CHKMEMQ; 33081d098868SHong Zhang /* ------- end of symbolic factorization ---------*/ 3309599ef60dSHong Zhang 3310599ef60dSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 3311599ef60dSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 3312599ef60dSHong Zhang ics = ic; 3313599ef60dSHong Zhang 3314599ef60dSHong Zhang /* linked list for storing column indices of the active row */ 3315599ef60dSHong Zhang nlnk = n + 1; 3316599ef60dSHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 33171d098868SHong Zhang 33181d098868SHong Zhang /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */ 33190e83c824SBarry Smith ierr = PetscMalloc2(n,PetscInt,&im,n,PetscInt,&jtmp);CHKERRQ(ierr); 33201d098868SHong Zhang /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */ 33210e83c824SBarry Smith ierr = PetscMalloc2(n,MatScalar,&rtmp,n,MatScalar,&vtmp);CHKERRQ(ierr); 33220e83c824SBarry Smith ierr = PetscMemzero(rtmp,n*sizeof(MatScalar));CHKERRQ(ierr); 3323599ef60dSHong Zhang 3324599ef60dSHong Zhang bi[0] = 0; 33258fc3a347SHong Zhang bdiag[0] = nnz_max-1; /* location of diag[0] in factor B */ 3326dc3a2fd3SHong Zhang bdiag_rev[n] = bdiag[0]; 33278fc3a347SHong Zhang bi[2*n+1] = bdiag[0]+1; /* endof bj and ba array */ 3328599ef60dSHong Zhang for (i=0; i<n; i++) { 3329599ef60dSHong Zhang /* copy initial fill into linked list */ 3330599ef60dSHong Zhang nzi = 0; /* nonzeros for active row i */ 33318369b28dSHong Zhang nzi = ai[r[i]+1] - ai[r[i]]; 3332e32f2f54SBarry Smith if (!nzi) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 33338369b28dSHong Zhang nzi_al = adiag[r[i]] - ai[r[i]]; 33348369b28dSHong Zhang nzi_au = ai[r[i]+1] - adiag[r[i]] -1; 3335599ef60dSHong Zhang ajtmp = aj + ai[r[i]]; 33368369b28dSHong Zhang ierr = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3337599ef60dSHong Zhang 3338599ef60dSHong Zhang /* load in initial (unfactored row) */ 33391d098868SHong Zhang aatmp = a->a + ai[r[i]]; 33408369b28dSHong Zhang for (j=0; j<nzi; j++) { 33411d098868SHong Zhang rtmp[ics[*ajtmp++]] = *aatmp++; 3342599ef60dSHong Zhang } 3343599ef60dSHong Zhang 3344599ef60dSHong Zhang /* add pivot rows into linked list */ 3345599ef60dSHong Zhang row = lnk[n]; 3346599ef60dSHong Zhang while (row < i ) { 33471d098868SHong Zhang nzi_bl = bi[row+1] - bi[row] + 1; 33481d098868SHong Zhang bjtmp = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */ 33491d098868SHong Zhang ierr = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr); 3350599ef60dSHong Zhang nzi += nlnk; 3351599ef60dSHong Zhang row = lnk[row]; 3352599ef60dSHong Zhang } 3353599ef60dSHong Zhang 33548369b28dSHong Zhang /* copy data from lnk into jtmp, then initialize lnk */ 33558369b28dSHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr); 3356599ef60dSHong Zhang 3357599ef60dSHong Zhang /* numerical factorization */ 33588369b28dSHong Zhang bjtmp = jtmp; 3359599ef60dSHong Zhang row = *bjtmp++; /* 1st pivot row */ 3360599ef60dSHong Zhang while ( row < i ) { 3361599ef60dSHong Zhang pc = rtmp + row; 33623c2a7987SHong Zhang pv = ba + bdiag[row]; /* 1./(diag of the pivot row) */ 33633c2a7987SHong Zhang multiplier = (*pc) * (*pv); 3364599ef60dSHong Zhang *pc = multiplier; 33653c2a7987SHong Zhang if (PetscAbsScalar(*pc) > dt){ /* apply tolerance dropping rule */ 33661d098868SHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 33671d098868SHong Zhang pv = ba + bdiag[row+1] + 1; 33681d098868SHong Zhang /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */ 33691d098868SHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 33701d098868SHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 3371dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 3372599ef60dSHong Zhang } 3373599ef60dSHong Zhang row = *bjtmp++; 3374599ef60dSHong Zhang } 3375599ef60dSHong Zhang 33768369b28dSHong Zhang /* copy sparse rtmp into contiguous vtmp; separate L and U part */ 33778aee2decSHong Zhang diag_tmp = rtmp[i]; /* save diagonal value - may not needed?? */ 33788369b28dSHong Zhang nzi_bl = 0; j = 0; 33798369b28dSHong Zhang while (jtmp[j] < i){ /* Note: jtmp is sorted */ 33808aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 33818369b28dSHong Zhang nzi_bl++; j++; 33828369b28dSHong Zhang } 33838369b28dSHong Zhang nzi_bu = nzi - nzi_bl -1; 33848369b28dSHong Zhang while (j < nzi){ 33858aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 33868369b28dSHong Zhang j++; 33878369b28dSHong Zhang } 33888369b28dSHong Zhang 3389599ef60dSHong Zhang bjtmp = bj + bi[i]; 3390599ef60dSHong Zhang batmp = ba + bi[i]; 33918369b28dSHong Zhang /* apply level dropping rule to L part */ 33928369b28dSHong Zhang ncut = nzi_al + dtcount; 33938369b28dSHong Zhang if (ncut < nzi_bl){ 33948369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr); 33958369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr); 3396599ef60dSHong Zhang } else { 33978369b28dSHong Zhang ncut = nzi_bl; 33988369b28dSHong Zhang } 33998369b28dSHong Zhang for (j=0; j<ncut; j++){ 34008369b28dSHong Zhang bjtmp[j] = jtmp[j]; 34018369b28dSHong Zhang batmp[j] = vtmp[j]; 34026da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */ 34038369b28dSHong Zhang } 34041d098868SHong Zhang bi[i+1] = bi[i] + ncut; 34058369b28dSHong Zhang nzi = ncut + 1; 34068369b28dSHong Zhang 34078369b28dSHong Zhang /* apply level dropping rule to U part */ 34088369b28dSHong Zhang ncut = nzi_au + dtcount; 34098369b28dSHong Zhang if (ncut < nzi_bu){ 34108369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr); 34118369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr); 34128369b28dSHong Zhang } else { 34138369b28dSHong Zhang ncut = nzi_bu; 34148369b28dSHong Zhang } 34158369b28dSHong Zhang nzi += ncut; 34161d098868SHong Zhang 34171d098868SHong Zhang /* mark bdiagonal */ 34181d098868SHong Zhang bdiag[i+1] = bdiag[i] - (ncut + 1); 3419dc3a2fd3SHong Zhang bdiag_rev[n-i-1] = bdiag[i+1]; 34208fc3a347SHong Zhang bi[2*n - i] = bi[2*n - i +1] - (ncut + 1); 34211d098868SHong Zhang bjtmp = bj + bdiag[i]; 34221d098868SHong Zhang batmp = ba + bdiag[i]; 34231d098868SHong Zhang *bjtmp = i; 34248aee2decSHong Zhang *batmp = diag_tmp; /* rtmp[i]; */ 3425c9c72f8fSHong Zhang if (*batmp == 0.0) { 3426c9c72f8fSHong Zhang *batmp = dt+shift; 3427bd1bc851SBarry Smith /* printf(" row %d add shift %g\n",i,shift); */ 3428c9c72f8fSHong Zhang } 3429b78a477dSHong Zhang *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */ 34306da515a1SHong Zhang /* printf(" (%d,%g),",*bjtmp,*batmp); */ 34311d098868SHong Zhang 34321d098868SHong Zhang bjtmp = bj + bdiag[i+1]+1; 34331d098868SHong Zhang batmp = ba + bdiag[i+1]+1; 34348369b28dSHong Zhang for (k=0; k<ncut; k++){ 34351d098868SHong Zhang bjtmp[k] = jtmp[nzi_bl+1+k]; 34361d098868SHong Zhang batmp[k] = vtmp[nzi_bl+1+k]; 34376da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */ 3438599ef60dSHong Zhang } 34396da515a1SHong Zhang /* printf("\n"); */ 3440599ef60dSHong Zhang 3441599ef60dSHong Zhang im[i] = nzi; /* used by PetscLLAddSortedLU() */ 3442599ef60dSHong Zhang /* 34431d098868SHong Zhang printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]); 34448369b28dSHong Zhang printf(" ----------------------------\n"); 3445599ef60dSHong Zhang */ 34468369b28dSHong Zhang } /* for (i=0; i<n; i++) */ 34471d098868SHong Zhang /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */ 3448e32f2f54SBarry Smith if (bi[n] >= bdiag[n]) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"end of L array %d cannot >= the beginning of U array %d",bi[n],bdiag[n]); 3449599ef60dSHong Zhang 3450599ef60dSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3451599ef60dSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3452599ef60dSHong Zhang 3453599ef60dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 34540e83c824SBarry Smith ierr = PetscFree2(im,jtmp);CHKERRQ(ierr); 34550e83c824SBarry Smith ierr = PetscFree2(rtmp,vtmp);CHKERRQ(ierr); 34561bfa06eaSJed Brown ierr = PetscFree(bdiag_rev);CHKERRQ(ierr); 3457599ef60dSHong Zhang 3458599ef60dSHong Zhang ierr = PetscLogFlops(B->cmap->n);CHKERRQ(ierr); 34591d098868SHong Zhang b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n]; 3460599ef60dSHong Zhang 3461599ef60dSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 3462599ef60dSHong Zhang ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr); 3463599ef60dSHong Zhang if (row_identity && icol_identity) { 346435233d99SShri Abhyankar B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 3465599ef60dSHong Zhang } else { 346635233d99SShri Abhyankar B->ops->solve = MatSolve_SeqAIJ; 3467599ef60dSHong Zhang } 3468599ef60dSHong Zhang 34691d098868SHong Zhang B->ops->solveadd = 0; 34701d098868SHong Zhang B->ops->solvetranspose = 0; 34711d098868SHong Zhang B->ops->solvetransposeadd = 0; 34721d098868SHong Zhang B->ops->matsolve = 0; 3473599ef60dSHong Zhang B->assembled = PETSC_TRUE; 3474599ef60dSHong Zhang B->preallocated = PETSC_TRUE; 3475599ef60dSHong Zhang PetscFunctionReturn(0); 3476599ef60dSHong Zhang } 3477599ef60dSHong Zhang 34783c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */ 34793c2a7987SHong Zhang #undef __FUNCT__ 34803c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorSymbolic_SeqAIJ" 3481fe97e370SBarry Smith /* 3482fe97e370SBarry Smith This will get a new name and become a varient of MatILUFactor_SeqAIJ() there is no longer seperate functions in the matrix function table for dt factors 3483fe97e370SBarry Smith */ 3484fe97e370SBarry Smith 34853c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info) 34863c2a7987SHong Zhang { 34873c2a7987SHong Zhang PetscErrorCode ierr; 34883c2a7987SHong Zhang 34893c2a7987SHong Zhang PetscFunctionBegin; 34903c2a7987SHong Zhang ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr); 34913c2a7987SHong Zhang PetscFunctionReturn(0); 34923c2a7987SHong Zhang } 34933c2a7987SHong Zhang 34943c2a7987SHong Zhang /* 34953c2a7987SHong Zhang same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors 34963c2a7987SHong Zhang - intend to replace existing MatLUFactorNumeric_SeqAIJ() 34973c2a7987SHong Zhang */ 34983c2a7987SHong Zhang #undef __FUNCT__ 34993c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorNumeric_SeqAIJ" 3500fe97e370SBarry Smith /* 3501fe97e370SBarry Smith This will get a new name and become a varient of MatILUFactor_SeqAIJ() there is no longer seperate functions in the matrix function table for dt factors 3502fe97e370SBarry Smith */ 3503fe97e370SBarry Smith 35043c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info) 35053c2a7987SHong Zhang { 35063c2a7987SHong Zhang Mat C=fact; 35073c2a7987SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 35083c2a7987SHong Zhang IS isrow = b->row,isicol = b->icol; 35093c2a7987SHong Zhang PetscErrorCode ierr; 35103c2a7987SHong Zhang const PetscInt *r,*ic,*ics; 3511b78a477dSHong Zhang PetscInt i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 3512b78a477dSHong Zhang PetscInt *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj; 35133c2a7987SHong Zhang MatScalar *rtmp,*pc,multiplier,*v,*pv,*aa=a->a; 3514182b8fbaSHong Zhang PetscReal dt=info->dt,shift=info->shiftamount; 3515b78a477dSHong Zhang PetscTruth row_identity, col_identity; 35163c2a7987SHong Zhang 35173c2a7987SHong Zhang PetscFunctionBegin; 35183c2a7987SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 35193c2a7987SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 3520b78a477dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 35213c2a7987SHong Zhang ics = ic; 35223c2a7987SHong Zhang 35233c2a7987SHong Zhang for (i=0; i<n; i++){ 3524b78a477dSHong Zhang /* initialize rtmp array */ 3525b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of nozeros in L(i,:) */ 35263c2a7987SHong Zhang bjtmp = bj + bi[i]; 3527b78a477dSHong Zhang for (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0; 3528b78a477dSHong Zhang rtmp[i] = 0.0; 3529b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */ 3530b78a477dSHong Zhang bjtmp = bj + bdiag[i+1] + 1; 3531b78a477dSHong Zhang for (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0; 35323c2a7987SHong Zhang 3533b78a477dSHong Zhang /* load in initial unfactored row of A */ 35346da515a1SHong Zhang /* printf("row %d\n",i); */ 35353c2a7987SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 35363c2a7987SHong Zhang ajtmp = aj + ai[r[i]]; 35373c2a7987SHong Zhang v = aa + ai[r[i]]; 35383c2a7987SHong Zhang for (j=0; j<nz; j++) { 3539b78a477dSHong Zhang rtmp[ics[*ajtmp++]] = v[j]; 35406da515a1SHong Zhang /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */ 35413c2a7987SHong Zhang } 35426da515a1SHong Zhang /* printf("\n"); */ 35433c2a7987SHong Zhang 3544b78a477dSHong Zhang /* numerical factorization */ 3545b78a477dSHong Zhang bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */ 3546b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of entries in L(i,:) */ 3547b78a477dSHong Zhang k = 0; 3548b78a477dSHong Zhang while (k < nzl){ 35493c2a7987SHong Zhang row = *bjtmp++; 35506da515a1SHong Zhang /* printf(" prow %d\n",row); */ 35513c2a7987SHong Zhang pc = rtmp + row; 3552b78a477dSHong Zhang pv = b->a + bdiag[row]; /* 1./(diag of the pivot row) */ 3553b78a477dSHong Zhang multiplier = (*pc) * (*pv); 35543c2a7987SHong Zhang *pc = multiplier; 3555b78a477dSHong Zhang if (PetscAbsScalar(multiplier) > dt){ 3556b78a477dSHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 3557b78a477dSHong Zhang pv = b->a + bdiag[row+1] + 1; 3558b78a477dSHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 3559b78a477dSHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 35606da515a1SHong Zhang /* ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); */ 35613c2a7987SHong Zhang } 3562b78a477dSHong Zhang k++; 35633c2a7987SHong Zhang } 35643c2a7987SHong Zhang 3565b78a477dSHong Zhang /* finished row so stick it into b->a */ 3566b78a477dSHong Zhang /* L-part */ 3567b78a477dSHong Zhang pv = b->a + bi[i] ; 3568b78a477dSHong Zhang pj = bj + bi[i] ; 3569b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; 3570b78a477dSHong Zhang for (j=0; j<nzl; j++) { 3571b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 35726da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 35733c2a7987SHong Zhang } 3574b78a477dSHong Zhang 3575b78a477dSHong Zhang /* diagonal: invert diagonal entries for simplier triangular solves */ 3576b78a477dSHong Zhang if (rtmp[i] == 0.0) rtmp[i] = dt+shift; 3577b78a477dSHong Zhang b->a[bdiag[i]] = 1.0/rtmp[i]; 35786da515a1SHong Zhang /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */ 3579b78a477dSHong Zhang 3580b78a477dSHong Zhang /* U-part */ 3581b78a477dSHong Zhang pv = b->a + bdiag[i+1] + 1; 3582b78a477dSHong Zhang pj = bj + bdiag[i+1] + 1; 3583b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1] - 1; 3584b78a477dSHong Zhang for (j=0; j<nzu; j++) { 3585b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 35866da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 3587b78a477dSHong Zhang } 35886da515a1SHong Zhang /* printf("\n"); */ 3589b78a477dSHong Zhang } 3590b78a477dSHong Zhang 35913c2a7987SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 35923c2a7987SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 35933c2a7987SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3594b78a477dSHong Zhang 35953c2a7987SHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 35963c2a7987SHong Zhang ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 35973c2a7987SHong Zhang if (row_identity && col_identity) { 359835233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 35993c2a7987SHong Zhang } else { 360035233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ; 36013c2a7987SHong Zhang } 3602b78a477dSHong Zhang C->ops->solveadd = 0; 3603b78a477dSHong Zhang C->ops->solvetranspose = 0; 3604b78a477dSHong Zhang C->ops->solvetransposeadd = 0; 3605b78a477dSHong Zhang C->ops->matsolve = 0; 36063c2a7987SHong Zhang C->assembled = PETSC_TRUE; 36073c2a7987SHong Zhang C->preallocated = PETSC_TRUE; 36083c2a7987SHong Zhang ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 36093c2a7987SHong Zhang PetscFunctionReturn(0); 36103c2a7987SHong Zhang } 3611