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__ 114a2ae208SSatish Balay #define __FUNCT__ "MatOrdering_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 148fa24674SBarry Smith */ 15dfbe8321SBarry Smith PetscErrorCode MatOrdering_Flow_SeqAIJ(Mat mat,const MatOrderingType type,IS *irow,IS *icol) 163b2fbd54SBarry Smith { 178fa24674SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)mat->data; 188fa24674SBarry Smith PetscErrorCode ierr; 19889a8dedSBarry Smith PetscInt i,j,jj,k, kk,n = mat->rmap->n, current = 0, newcurrent = 0,*order; 208fa24674SBarry Smith const PetscInt *ai = a->i, *aj = a->j; 218fa24674SBarry Smith const PetscScalar *aa = a->a; 228fa24674SBarry Smith PetscTruth *done; 23889a8dedSBarry Smith PetscReal best,past = 0,future; 243a40ed3dSBarry Smith 258fa24674SBarry Smith PetscFunctionBegin; 268fa24674SBarry Smith /* pick initial row */ 278fa24674SBarry Smith best = -1; 288fa24674SBarry Smith for (i=0; i<n; i++) { 298fa24674SBarry Smith future = 0; 308fa24674SBarry Smith for (j=ai[i]; j<ai[i+1]; j++) { 318fa24674SBarry Smith if (aj[j] != i) future += PetscAbsScalar(aa[j]); else past = PetscAbsScalar(aa[j]); 328fa24674SBarry Smith } 338fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 348fa24674SBarry Smith if (past/future > best) { 358fa24674SBarry Smith best = past/future; 368fa24674SBarry Smith current = i; 378fa24674SBarry Smith } 388fa24674SBarry Smith } 398fa24674SBarry Smith 408fa24674SBarry Smith ierr = PetscMalloc(n*sizeof(PetscTruth),&done);CHKERRQ(ierr); 418fa24674SBarry Smith ierr = PetscMalloc(n*sizeof(PetscInt),&order);CHKERRQ(ierr); 428fa24674SBarry Smith ierr = PetscMemzero(done,n*sizeof(PetscTruth));CHKERRQ(ierr); 438fa24674SBarry Smith order[0] = current; 448fa24674SBarry Smith for (i=0; i<n-1; i++) { 458fa24674SBarry Smith done[current] = PETSC_TRUE; 468fa24674SBarry Smith best = -1; 478fa24674SBarry Smith /* loop over all neighbors of current pivot */ 488fa24674SBarry Smith for (j=ai[current]; j<ai[current+1]; j++) { 498fa24674SBarry Smith jj = aj[j]; 508fa24674SBarry Smith if (done[jj]) continue; 518fa24674SBarry Smith /* loop over columns of potential next row computing weights for below and above diagonal */ 528fa24674SBarry Smith past = future = 0.0; 538fa24674SBarry Smith for (k=ai[jj]; k<ai[jj+1]; k++) { 548fa24674SBarry Smith kk = aj[k]; 558fa24674SBarry Smith if (done[kk]) past += PetscAbsScalar(aa[k]); 568fa24674SBarry Smith else if (kk != jj) future += PetscAbsScalar(aa[k]); 578fa24674SBarry Smith } 588fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 598fa24674SBarry Smith if (past/future > best) { 608fa24674SBarry Smith best = past/future; 618fa24674SBarry Smith newcurrent = jj; 628fa24674SBarry Smith } 638fa24674SBarry Smith } 648fa24674SBarry Smith if (best == -1) { /* no neighbors to select from so select best of all that remain */ 658fa24674SBarry Smith best = -1; 668fa24674SBarry Smith for (k=0; k<n; k++) { 678fa24674SBarry Smith if (done[k]) continue; 688fa24674SBarry Smith future = 0; 698fa24674SBarry Smith past = 0; 708fa24674SBarry Smith for (j=ai[k]; j<ai[k+1]; j++) { 718fa24674SBarry Smith kk = aj[j]; 728fa24674SBarry Smith if (done[kk]) past += PetscAbsScalar(aa[j]); 738fa24674SBarry Smith else if (kk != k) future += PetscAbsScalar(aa[j]); 748fa24674SBarry Smith } 758fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 768fa24674SBarry Smith if (past/future > best) { 778fa24674SBarry Smith best = past/future; 788fa24674SBarry Smith newcurrent = k; 798fa24674SBarry Smith } 808fa24674SBarry Smith } 818fa24674SBarry Smith } 828fa24674SBarry Smith if (current == newcurrent) SETERRQ(PETSC_ERR_PLIB,"newcurrent cannot be current"); 838fa24674SBarry Smith current = newcurrent; 848fa24674SBarry Smith order[i+1] = current; 858fa24674SBarry Smith } 868fa24674SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,n,order,irow);CHKERRQ(ierr); 878fa24674SBarry Smith *icol = *irow; 888fa24674SBarry Smith ierr = PetscObjectReference((PetscObject)*irow);CHKERRQ(ierr); 898fa24674SBarry Smith ierr = PetscFree(done);CHKERRQ(ierr); 908fa24674SBarry Smith ierr = PetscFree(order);CHKERRQ(ierr); 91d64ed03dSBarry Smith PetscFunctionReturn(0); 923b2fbd54SBarry Smith } 93f6a26d55SMatthew Knepley EXTERN_C_END 943b2fbd54SBarry Smith 95e631078cSBarry Smith EXTERN_C_BEGIN 964a2ae208SSatish Balay #undef __FUNCT__ 97db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqaij_petsc" 98db4efbfdSBarry Smith PetscErrorCode MatGetFactorAvailable_seqaij_petsc(Mat A,MatFactorType ftype,PetscTruth *flg) 99db4efbfdSBarry Smith { 100db4efbfdSBarry Smith PetscFunctionBegin; 101db4efbfdSBarry Smith *flg = PETSC_TRUE; 102db4efbfdSBarry Smith PetscFunctionReturn(0); 103db4efbfdSBarry Smith } 104db4efbfdSBarry Smith EXTERN_C_END 105db4efbfdSBarry Smith 106db4efbfdSBarry Smith EXTERN_C_BEGIN 107db4efbfdSBarry Smith #undef __FUNCT__ 108b24902e0SBarry Smith #define __FUNCT__ "MatGetFactor_seqaij_petsc" 109b24902e0SBarry Smith PetscErrorCode MatGetFactor_seqaij_petsc(Mat A,MatFactorType ftype,Mat *B) 110b24902e0SBarry Smith { 111d0f46423SBarry Smith PetscInt n = A->rmap->n; 112b24902e0SBarry Smith PetscErrorCode ierr; 113b24902e0SBarry Smith 114b24902e0SBarry Smith PetscFunctionBegin; 115b24902e0SBarry Smith ierr = MatCreate(((PetscObject)A)->comm,B);CHKERRQ(ierr); 116b24902e0SBarry Smith ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr); 117599ef60dSHong Zhang if (ftype == MAT_FACTOR_LU || ftype == MAT_FACTOR_ILU || ftype == MAT_FACTOR_ILUDT){ 118b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQAIJ);CHKERRQ(ierr); 119db4efbfdSBarry Smith (*B)->ops->ilufactorsymbolic = MatILUFactorSymbolic_SeqAIJ; 12035bd34faSBarry Smith (*B)->ops->lufactorsymbolic = MatLUFactorSymbolic_SeqAIJ; 1211da05e5aSHong Zhang (*B)->ops->iludtfactor = MatILUDTFactor_SeqAIJ; 122b24902e0SBarry Smith } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 123b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr); 124b24902e0SBarry Smith ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 1255c9eb25fSBarry Smith (*B)->ops->iccfactorsymbolic = MatICCFactorSymbolic_SeqAIJ; 1265c9eb25fSBarry Smith (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ; 127b24902e0SBarry Smith } else SETERRQ(PETSC_ERR_SUP,"Factor type not supported"); 128db4efbfdSBarry Smith (*B)->factor = ftype; 129b24902e0SBarry Smith PetscFunctionReturn(0); 130b24902e0SBarry Smith } 131e631078cSBarry Smith EXTERN_C_END 132b24902e0SBarry Smith 133b24902e0SBarry Smith #undef __FUNCT__ 134b9617806SBarry Smith #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ" 1350481f469SBarry Smith PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 136289bc588SBarry Smith { 137416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 138289bc588SBarry Smith IS isicol; 1396849ba73SBarry Smith PetscErrorCode ierr; 1405d0c19d7SBarry Smith const PetscInt *r,*ic; 1415d0c19d7SBarry Smith PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 1421393bc97SHong Zhang PetscInt *bi,*bj,*ajtmp; 1431393bc97SHong Zhang PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 1449e046878SBarry Smith PetscReal f; 1454875ba38SHong Zhang PetscInt nlnk,*lnk,k,**bi_ptr; 146a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1471393bc97SHong Zhang PetscBT lnkbt; 148ce3d78c0SShri Abhyankar PetscTruth newdatastruct=PETSC_FALSE; 1499b48462bSShri Abhyankar PetscTruth newdatastruct_v2 = PETSC_FALSE; 150289bc588SBarry Smith 1513a40ed3dSBarry Smith PetscFunctionBegin; 152ce3d78c0SShri Abhyankar ierr = PetscOptionsGetTruth(PETSC_NULL,"-lu_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr); 153ce3d78c0SShri Abhyankar if(newdatastruct){ 154ce3d78c0SShri Abhyankar ierr = MatLUFactorSymbolic_SeqAIJ_newdatastruct(B,A,isrow,iscol,info);CHKERRQ(ierr); 155ce3d78c0SShri Abhyankar PetscFunctionReturn(0); 156ce3d78c0SShri Abhyankar } 1579b48462bSShri Abhyankar ierr = PetscOptionsGetTruth(PETSC_NULL,"-lu_new_v2",&newdatastruct_v2,PETSC_NULL);CHKERRQ(ierr); 1589b48462bSShri Abhyankar if(newdatastruct_v2){ 1599b48462bSShri Abhyankar ierr = MatLUFactorSymbolic_SeqAIJ_newdatastruct_v2(B,A,isrow,iscol,info);CHKERRQ(ierr); 1609b48462bSShri Abhyankar PetscFunctionReturn(0); 1619b48462bSShri Abhyankar } 1629b48462bSShri Abhyankar 163ce3d78c0SShri Abhyankar 164d0f46423SBarry Smith if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square"); 1654c49b128SBarry Smith ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 166ac121b3dSBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 167ac121b3dSBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 168289bc588SBarry Smith 169289bc588SBarry Smith /* get new row pointers */ 1701393bc97SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 1711393bc97SHong Zhang bi[0] = 0; 1721393bc97SHong Zhang 1731393bc97SHong Zhang /* bdiag is location of diagonal in factor */ 1741393bc97SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 1751393bc97SHong Zhang bdiag[0] = 0; 1761393bc97SHong Zhang 1771393bc97SHong Zhang /* linked list for storing column indices of the active row */ 1781393bc97SHong Zhang nlnk = n + 1; 1791393bc97SHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1801393bc97SHong Zhang 18135baf27eSSatish Balay ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 1821393bc97SHong Zhang 1831393bc97SHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 184d3d32019SBarry Smith f = info->fill; 185a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 1861393bc97SHong Zhang current_space = free_space; 187289bc588SBarry Smith 188289bc588SBarry Smith for (i=0; i<n; i++) { 1891393bc97SHong Zhang /* copy previous fill into linked list */ 190289bc588SBarry Smith nzi = 0; 1911393bc97SHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 1923b4a8b6dSBarry Smith if (!nnz) SETERRQ2(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 1931393bc97SHong Zhang ajtmp = aj + ai[r[i]]; 194afcc9904SHong Zhang ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1951393bc97SHong Zhang nzi += nlnk; 1961393bc97SHong Zhang 1971393bc97SHong Zhang /* add pivot rows into linked list */ 1981393bc97SHong Zhang row = lnk[n]; 1991393bc97SHong Zhang while (row < i) { 200d1170560SHong Zhang nzbd = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */ 201d1170560SHong Zhang ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 202d1170560SHong Zhang ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 2031393bc97SHong Zhang nzi += nlnk; 2041393bc97SHong Zhang row = lnk[row]; 205289bc588SBarry Smith } 2061393bc97SHong Zhang bi[i+1] = bi[i] + nzi; 2071393bc97SHong Zhang im[i] = nzi; 2081393bc97SHong Zhang 2091393bc97SHong Zhang /* mark bdiag */ 2101393bc97SHong Zhang nzbd = 0; 2111393bc97SHong Zhang nnz = nzi; 2121393bc97SHong Zhang k = lnk[n]; 2131393bc97SHong Zhang while (nnz-- && k < i){ 2141393bc97SHong Zhang nzbd++; 2151393bc97SHong Zhang k = lnk[k]; 2161393bc97SHong Zhang } 2171393bc97SHong Zhang bdiag[i] = bi[i] + nzbd; 2181393bc97SHong Zhang 2191393bc97SHong Zhang /* if free space is not available, make more free space */ 2201393bc97SHong Zhang if (current_space->local_remaining<nzi) { 2211393bc97SHong Zhang nnz = (n - i)*nzi; /* estimated and max additional space needed */ 222a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 2231393bc97SHong Zhang reallocs++; 2241393bc97SHong Zhang } 2251393bc97SHong Zhang 2261393bc97SHong Zhang /* copy data into free space, then initialize lnk */ 2271393bc97SHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 2281393bc97SHong Zhang bi_ptr[i] = current_space->array; 2291393bc97SHong Zhang current_space->array += nzi; 2301393bc97SHong Zhang current_space->local_used += nzi; 2311393bc97SHong Zhang current_space->local_remaining -= nzi; 232289bc588SBarry Smith } 2336cf91177SBarry Smith #if defined(PETSC_USE_INFO) 234464e8d44SSatish Balay if (ai[n] != 0) { 2351393bc97SHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 236ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 237ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 238ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 239ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 24005bf559cSSatish Balay } else { 241ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 24205bf559cSSatish Balay } 24363ba0a88SBarry Smith #endif 2442fb3ed76SBarry Smith 245898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 246898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 2471987afe7SBarry Smith 2481393bc97SHong Zhang /* destroy list of free space and other temporary array(s) */ 2491393bc97SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 250a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); 2511393bc97SHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 25235baf27eSSatish Balay ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 253289bc588SBarry Smith 254289bc588SBarry Smith /* put together the new matrix */ 255719d5645SBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 256719d5645SBarry Smith ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 257719d5645SBarry Smith b = (Mat_SeqAIJ*)(B)->data; 258e6b907acSBarry Smith b->free_a = PETSC_TRUE; 259e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 2607c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 2611393bc97SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 2621393bc97SHong Zhang b->j = bj; 2631393bc97SHong Zhang b->i = bi; 2641393bc97SHong Zhang b->diag = bdiag; 265416022c9SBarry Smith b->ilen = 0; 266416022c9SBarry Smith b->imax = 0; 267416022c9SBarry Smith b->row = isrow; 268416022c9SBarry Smith b->col = iscol; 269c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 270c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 27182bf6240SBarry Smith b->icol = isicol; 27287828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2731393bc97SHong Zhang 2741393bc97SHong Zhang /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 275719d5645SBarry Smith ierr = PetscLogObjectMemory(B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 2761393bc97SHong Zhang b->maxnz = b->nz = bi[n] ; 2777fd98bd6SLois Curfman McInnes 278719d5645SBarry Smith (B)->factor = MAT_FACTOR_LU; 279719d5645SBarry Smith (B)->info.factor_mallocs = reallocs; 280719d5645SBarry Smith (B)->info.fill_ratio_given = f; 281703deb49SSatish Balay 2829f7d0b68SHong Zhang if (ai[n]) { 283719d5645SBarry Smith (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 284eea2913aSSatish Balay } else { 285719d5645SBarry Smith (B)->info.fill_ratio_needed = 0.0; 286eea2913aSSatish Balay } 287719d5645SBarry Smith (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 288719d5645SBarry Smith (B)->ops->solve = MatSolve_SeqAIJ; 289719d5645SBarry Smith (B)->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 290db4efbfdSBarry Smith /* switch to inodes if appropriate */ 2914108e4d5SBarry Smith ierr = MatLUFactorSymbolic_SeqAIJ_Inode(B,A,isrow,iscol,info);CHKERRQ(ierr); 2923a40ed3dSBarry Smith PetscFunctionReturn(0); 293289bc588SBarry Smith } 2941393bc97SHong Zhang 295ce3d78c0SShri Abhyankar #undef __FUNCT__ 296ce3d78c0SShri Abhyankar #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ_newdatastruct" 297ce3d78c0SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ_newdatastruct(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 298ce3d78c0SShri Abhyankar { 299ce3d78c0SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 300ce3d78c0SShri Abhyankar IS isicol; 301ce3d78c0SShri Abhyankar PetscErrorCode ierr; 302ce3d78c0SShri Abhyankar const PetscInt *r,*ic; 303ce3d78c0SShri Abhyankar PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 304ce3d78c0SShri Abhyankar PetscInt *bi,*bj,*ajtmp; 305ce3d78c0SShri Abhyankar PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 306ce3d78c0SShri Abhyankar PetscReal f; 307ce3d78c0SShri Abhyankar PetscInt nlnk,*lnk,k,**bi_ptr; 308ce3d78c0SShri Abhyankar PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 309ce3d78c0SShri Abhyankar PetscBT lnkbt; 310ce3d78c0SShri Abhyankar 311ce3d78c0SShri Abhyankar PetscFunctionBegin; 312ce3d78c0SShri Abhyankar if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square"); 313ce3d78c0SShri Abhyankar ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 314ce3d78c0SShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 315ce3d78c0SShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 316ce3d78c0SShri Abhyankar 317ce3d78c0SShri Abhyankar /* get new row pointers */ 318ce3d78c0SShri Abhyankar ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 319ce3d78c0SShri Abhyankar bi[0] = 0; 320ce3d78c0SShri Abhyankar 321ce3d78c0SShri Abhyankar /* bdiag is location of diagonal in factor */ 322ce3d78c0SShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 323ce3d78c0SShri Abhyankar bdiag[0] = 0; 324ce3d78c0SShri Abhyankar 325ce3d78c0SShri Abhyankar /* linked list for storing column indices of the active row */ 326ce3d78c0SShri Abhyankar nlnk = n + 1; 327ce3d78c0SShri Abhyankar ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 328ce3d78c0SShri Abhyankar 329ce3d78c0SShri Abhyankar ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 330ce3d78c0SShri Abhyankar 331ce3d78c0SShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 332ce3d78c0SShri Abhyankar f = info->fill; 333ce3d78c0SShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 334ce3d78c0SShri Abhyankar current_space = free_space; 335ce3d78c0SShri Abhyankar 336ce3d78c0SShri Abhyankar for (i=0; i<n; i++) { 337ce3d78c0SShri Abhyankar /* copy previous fill into linked list */ 338ce3d78c0SShri Abhyankar nzi = 0; 339ce3d78c0SShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 340ce3d78c0SShri Abhyankar if (!nnz) SETERRQ2(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 341ce3d78c0SShri Abhyankar ajtmp = aj + ai[r[i]]; 342ce3d78c0SShri Abhyankar ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 343ce3d78c0SShri Abhyankar nzi += nlnk; 344ce3d78c0SShri Abhyankar 345ce3d78c0SShri Abhyankar /* add pivot rows into linked list */ 346ce3d78c0SShri Abhyankar row = lnk[n]; 347ce3d78c0SShri Abhyankar while (row < i){ 34870319c84SHong Zhang nzbd = bdiag[row] + 1; /* num of entries in the row with column index <= row */ 349ce3d78c0SShri Abhyankar ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 350ce3d78c0SShri Abhyankar ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 351ce3d78c0SShri Abhyankar nzi += nlnk; 352ce3d78c0SShri Abhyankar row = lnk[row]; 353ce3d78c0SShri Abhyankar } 354ce3d78c0SShri Abhyankar bi[i+1] = bi[i] + nzi; 355ce3d78c0SShri Abhyankar im[i] = nzi; 356ce3d78c0SShri Abhyankar 357ce3d78c0SShri Abhyankar /* mark bdiag */ 358ce3d78c0SShri Abhyankar nzbd = 0; 359ce3d78c0SShri Abhyankar nnz = nzi; 360ce3d78c0SShri Abhyankar k = lnk[n]; 361ce3d78c0SShri Abhyankar while (nnz-- && k < i){ 362ce3d78c0SShri Abhyankar nzbd++; 363ce3d78c0SShri Abhyankar k = lnk[k]; 364ce3d78c0SShri Abhyankar } 365783ef271SHong Zhang bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */ 366ce3d78c0SShri Abhyankar 367ce3d78c0SShri Abhyankar /* if free space is not available, make more free space */ 368ce3d78c0SShri Abhyankar if (current_space->local_remaining<nzi) { 369ce3d78c0SShri Abhyankar nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */ 370ce3d78c0SShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 371ce3d78c0SShri Abhyankar reallocs++; 372ce3d78c0SShri Abhyankar } 373ce3d78c0SShri Abhyankar 374ce3d78c0SShri Abhyankar /* copy data into free space, then initialize lnk */ 375ce3d78c0SShri Abhyankar ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 376ce3d78c0SShri Abhyankar bi_ptr[i] = current_space->array; 377ce3d78c0SShri Abhyankar current_space->array += nzi; 378ce3d78c0SShri Abhyankar current_space->local_used += nzi; 379ce3d78c0SShri Abhyankar current_space->local_remaining -= nzi; 380ce3d78c0SShri Abhyankar } 381ce3d78c0SShri Abhyankar #if defined(PETSC_USE_INFO) 382ce3d78c0SShri Abhyankar if (ai[n] != 0) { 383ce3d78c0SShri Abhyankar PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 384ce3d78c0SShri Abhyankar ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 385ce3d78c0SShri Abhyankar ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 386ce3d78c0SShri Abhyankar ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 387ce3d78c0SShri Abhyankar ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 388ce3d78c0SShri Abhyankar } else { 389ce3d78c0SShri Abhyankar ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 390ce3d78c0SShri Abhyankar } 391ce3d78c0SShri Abhyankar #endif 392ce3d78c0SShri Abhyankar 393ce3d78c0SShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 394ce3d78c0SShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 395ce3d78c0SShri Abhyankar 396ce3d78c0SShri Abhyankar /* destroy list of free space and other temporary array(s) */ 397ce3d78c0SShri Abhyankar ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 398783ef271SHong Zhang ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 399ce3d78c0SShri Abhyankar ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 400ce3d78c0SShri Abhyankar ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 401ce3d78c0SShri Abhyankar 402ce3d78c0SShri Abhyankar /* put together the new matrix */ 403ce3d78c0SShri Abhyankar ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 404ce3d78c0SShri Abhyankar ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 405ce3d78c0SShri Abhyankar b = (Mat_SeqAIJ*)(B)->data; 406ce3d78c0SShri Abhyankar b->free_a = PETSC_TRUE; 407ce3d78c0SShri Abhyankar b->free_ij = PETSC_TRUE; 408ce3d78c0SShri Abhyankar b->singlemalloc = PETSC_FALSE; 409ce3d78c0SShri Abhyankar ierr = PetscMalloc((bi[2*n+1])*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 410ce3d78c0SShri Abhyankar b->j = bj; 411ce3d78c0SShri Abhyankar b->i = bi; 412ce3d78c0SShri Abhyankar b->diag = bdiag; 413ce3d78c0SShri Abhyankar b->ilen = 0; 414ce3d78c0SShri Abhyankar b->imax = 0; 415ce3d78c0SShri Abhyankar b->row = isrow; 416ce3d78c0SShri Abhyankar b->col = iscol; 417ce3d78c0SShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 418ce3d78c0SShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 419ce3d78c0SShri Abhyankar b->icol = isicol; 420ce3d78c0SShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 421ce3d78c0SShri Abhyankar 422ce3d78c0SShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 423ce3d78c0SShri Abhyankar ierr = PetscLogObjectMemory(B,bi[2*n+1]*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 424ce3d78c0SShri Abhyankar b->maxnz = b->nz = bi[2*n+1] ; 425ce3d78c0SShri Abhyankar 426ce3d78c0SShri Abhyankar (B)->factor = MAT_FACTOR_LU; 427ce3d78c0SShri Abhyankar (B)->info.factor_mallocs = reallocs; 428ce3d78c0SShri Abhyankar (B)->info.fill_ratio_given = f; 429ce3d78c0SShri Abhyankar 430ce3d78c0SShri Abhyankar if (ai[n]) { 431894a2336SHong Zhang (B)->info.fill_ratio_needed = ((PetscReal)bi[2*n+1])/((PetscReal)ai[n]); 432ce3d78c0SShri Abhyankar } else { 433ce3d78c0SShri Abhyankar (B)->info.fill_ratio_needed = 0.0; 434ce3d78c0SShri Abhyankar } 435ce3d78c0SShri Abhyankar (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct; 436ce3d78c0SShri Abhyankar PetscFunctionReturn(0); 437ce3d78c0SShri Abhyankar } 438ce3d78c0SShri Abhyankar 4399b48462bSShri Abhyankar #undef __FUNCT__ 4409b48462bSShri Abhyankar #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ_newdatastruct_v2" 4419b48462bSShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ_newdatastruct_v2(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 4429b48462bSShri Abhyankar { 4439b48462bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 4449b48462bSShri Abhyankar IS isicol; 4459b48462bSShri Abhyankar PetscErrorCode ierr; 4469b48462bSShri Abhyankar const PetscInt *r,*ic; 4479b48462bSShri Abhyankar PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 4489b48462bSShri Abhyankar PetscInt *bi,*bj,*ajtmp; 4499b48462bSShri Abhyankar PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 4509b48462bSShri Abhyankar PetscReal f; 4519b48462bSShri Abhyankar PetscInt nlnk,*lnk,k,**bi_ptr; 4529b48462bSShri Abhyankar PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 4539b48462bSShri Abhyankar PetscBT lnkbt; 4549b48462bSShri Abhyankar 4559b48462bSShri Abhyankar PetscFunctionBegin; 4569b48462bSShri Abhyankar if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square"); 4579b48462bSShri Abhyankar ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 4589b48462bSShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 4599b48462bSShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 4609b48462bSShri Abhyankar 4619b48462bSShri Abhyankar /* get new row pointers */ 4629b48462bSShri Abhyankar /* ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); */ 4639b48462bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 4649b48462bSShri Abhyankar bi[0] = 0; 4659b48462bSShri Abhyankar 4669b48462bSShri Abhyankar /* bdiag is location of diagonal in factor */ 4679b48462bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 4689b48462bSShri Abhyankar bdiag[0] = 0; 4699b48462bSShri Abhyankar 4709b48462bSShri Abhyankar /* linked list for storing column indices of the active row */ 4719b48462bSShri Abhyankar nlnk = n + 1; 4729b48462bSShri Abhyankar ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 4739b48462bSShri Abhyankar 4749b48462bSShri Abhyankar ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 4759b48462bSShri Abhyankar 4769b48462bSShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 4779b48462bSShri Abhyankar f = info->fill; 4789b48462bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 4799b48462bSShri Abhyankar current_space = free_space; 4809b48462bSShri Abhyankar 4819b48462bSShri Abhyankar for (i=0; i<n; i++) { 4829b48462bSShri Abhyankar /* copy previous fill into linked list */ 4839b48462bSShri Abhyankar nzi = 0; 4849b48462bSShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 4859b48462bSShri Abhyankar if (!nnz) SETERRQ2(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 4869b48462bSShri Abhyankar ajtmp = aj + ai[r[i]]; 4879b48462bSShri Abhyankar ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 4889b48462bSShri Abhyankar nzi += nlnk; 4899b48462bSShri Abhyankar 4909b48462bSShri Abhyankar /* add pivot rows into linked list */ 4919b48462bSShri Abhyankar row = lnk[n]; 4929b48462bSShri Abhyankar while (row < i){ 4939b48462bSShri Abhyankar nzbd = bdiag[row] + 1; /* num of entries in the row with column index <= row */ 4949b48462bSShri Abhyankar ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 4959b48462bSShri Abhyankar ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 4969b48462bSShri Abhyankar nzi += nlnk; 4979b48462bSShri Abhyankar row = lnk[row]; 4989b48462bSShri Abhyankar } 4999b48462bSShri Abhyankar bi[i+1] = bi[i] + nzi; 5009b48462bSShri Abhyankar im[i] = nzi; 5019b48462bSShri Abhyankar 5029b48462bSShri Abhyankar /* mark bdiag */ 5039b48462bSShri Abhyankar nzbd = 0; 5049b48462bSShri Abhyankar nnz = nzi; 5059b48462bSShri Abhyankar k = lnk[n]; 5069b48462bSShri Abhyankar while (nnz-- && k < i){ 5079b48462bSShri Abhyankar nzbd++; 5089b48462bSShri Abhyankar k = lnk[k]; 5099b48462bSShri Abhyankar } 5109b48462bSShri Abhyankar bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */ 5119b48462bSShri Abhyankar 5129b48462bSShri Abhyankar /* if free space is not available, make more free space */ 5139b48462bSShri Abhyankar if (current_space->local_remaining<nzi) { 5149b48462bSShri Abhyankar nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */ 5159b48462bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 5169b48462bSShri Abhyankar reallocs++; 5179b48462bSShri Abhyankar } 5189b48462bSShri Abhyankar 5199b48462bSShri Abhyankar /* copy data into free space, then initialize lnk */ 5209b48462bSShri Abhyankar ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 5219b48462bSShri Abhyankar bi_ptr[i] = current_space->array; 5229b48462bSShri Abhyankar current_space->array += nzi; 5239b48462bSShri Abhyankar current_space->local_used += nzi; 5249b48462bSShri Abhyankar current_space->local_remaining -= nzi; 5259b48462bSShri Abhyankar } 5269b48462bSShri Abhyankar #if defined(PETSC_USE_INFO) 5279b48462bSShri Abhyankar if (ai[n] != 0) { 5289b48462bSShri Abhyankar PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 5299b48462bSShri Abhyankar ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 5309b48462bSShri Abhyankar ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 5319b48462bSShri Abhyankar ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 5329b48462bSShri Abhyankar ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 5339b48462bSShri Abhyankar } else { 5349b48462bSShri Abhyankar ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 5359b48462bSShri Abhyankar } 5369b48462bSShri Abhyankar #endif 5379b48462bSShri Abhyankar 5389b48462bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 5399b48462bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 5409b48462bSShri Abhyankar 5419b48462bSShri Abhyankar /* destroy list of free space and other temporary array(s) */ 5429b48462bSShri Abhyankar ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 5439b48462bSShri Abhyankar /* ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); */ 5449b48462bSShri Abhyankar ierr = PetscFreeSpaceContiguous_LU_v2(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 5459b48462bSShri Abhyankar ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 5469b48462bSShri Abhyankar ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 5479b48462bSShri Abhyankar 5489b48462bSShri Abhyankar /* put together the new matrix */ 5499b48462bSShri Abhyankar ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 5509b48462bSShri Abhyankar ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 5519b48462bSShri Abhyankar b = (Mat_SeqAIJ*)(B)->data; 5529b48462bSShri Abhyankar b->free_a = PETSC_TRUE; 5539b48462bSShri Abhyankar b->free_ij = PETSC_TRUE; 5549b48462bSShri Abhyankar b->singlemalloc = PETSC_FALSE; 5559b48462bSShri Abhyankar /*ierr = PetscMalloc((bi[2*n+1])*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);*/ 5569b48462bSShri Abhyankar ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 5579b48462bSShri Abhyankar b->j = bj; 5589b48462bSShri Abhyankar b->i = bi; 5599b48462bSShri Abhyankar b->diag = bdiag; 5609b48462bSShri Abhyankar b->ilen = 0; 5619b48462bSShri Abhyankar b->imax = 0; 5629b48462bSShri Abhyankar b->row = isrow; 5639b48462bSShri Abhyankar b->col = iscol; 5649b48462bSShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 5659b48462bSShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 5669b48462bSShri Abhyankar b->icol = isicol; 5679b48462bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 5689b48462bSShri Abhyankar 5699b48462bSShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 5709b48462bSShri Abhyankar /* ierr = PetscLogObjectMemory(B,bi[2*n+1]*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 5719b48462bSShri Abhyankar b->maxnz = b->nz = bi[2*n+1] ; */ 5729b48462bSShri Abhyankar ierr = PetscLogObjectMemory(B,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 5739b48462bSShri Abhyankar b->maxnz = b->nz = bdiag[0]+1; 5749b48462bSShri Abhyankar (B)->factor = MAT_FACTOR_LU; 5759b48462bSShri Abhyankar (B)->info.factor_mallocs = reallocs; 5769b48462bSShri Abhyankar (B)->info.fill_ratio_given = f; 5779b48462bSShri Abhyankar 5789b48462bSShri Abhyankar if (ai[n]) { 5799b48462bSShri Abhyankar /* (B)->info.fill_ratio_needed = ((PetscReal)bi[2*n+1])/((PetscReal)ai[n]); */ 5809b48462bSShri Abhyankar (B)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 5819b48462bSShri Abhyankar } else { 5829b48462bSShri Abhyankar (B)->info.fill_ratio_needed = 0.0; 5839b48462bSShri Abhyankar } 5849b48462bSShri Abhyankar /* (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct;*/ 5859b48462bSShri Abhyankar (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct_v2; 5869b48462bSShri Abhyankar PetscFunctionReturn(0); 5879b48462bSShri Abhyankar } 5889b48462bSShri Abhyankar 5895b5bc046SBarry Smith /* 5905b5bc046SBarry Smith Trouble in factorization, should we dump the original matrix? 5915b5bc046SBarry Smith */ 5925b5bc046SBarry Smith #undef __FUNCT__ 5935b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix" 5945b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A) 5955b5bc046SBarry Smith { 5965b5bc046SBarry Smith PetscErrorCode ierr; 59790d69ab7SBarry Smith PetscTruth flg = PETSC_FALSE; 5985b5bc046SBarry Smith 5995b5bc046SBarry Smith PetscFunctionBegin; 60090d69ab7SBarry Smith ierr = PetscOptionsGetTruth(PETSC_NULL,"-mat_factor_dump_on_error",&flg,PETSC_NULL);CHKERRQ(ierr); 6015b5bc046SBarry Smith if (flg) { 6025b5bc046SBarry Smith PetscViewer viewer; 6035b5bc046SBarry Smith char filename[PETSC_MAX_PATH_LEN]; 6045b5bc046SBarry Smith 6055b5bc046SBarry Smith ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr); 6067adad957SLisandro Dalcin ierr = PetscViewerBinaryOpen(((PetscObject)A)->comm,filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr); 6075b5bc046SBarry Smith ierr = MatView(A,viewer);CHKERRQ(ierr); 6085b5bc046SBarry Smith ierr = PetscViewerDestroy(viewer);CHKERRQ(ierr); 6095b5bc046SBarry Smith } 6105b5bc046SBarry Smith PetscFunctionReturn(0); 6115b5bc046SBarry Smith } 6125b5bc046SBarry Smith 6134108e4d5SBarry Smith extern PetscErrorCode MatSolve_SeqAIJ_Inode(Mat,Vec,Vec); 614db4efbfdSBarry Smith 6150a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 616894a2336SHong Zhang extern PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_newdatastruct(Mat,Vec,Vec); 617894a2336SHong Zhang extern PetscErrorCode MatSolve_SeqAIJ_newdatastruct(Mat,Vec,Vec); 618c9c72f8fSHong Zhang 619c9c72f8fSHong Zhang #undef __FUNCT__ 620c9c72f8fSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_newdatastruct" 621c9c72f8fSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_newdatastruct(Mat B,Mat A,const MatFactorInfo *info) 622c9c72f8fSHong Zhang { 623c9c72f8fSHong Zhang Mat C=B; 624c9c72f8fSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 625c9c72f8fSHong Zhang IS isrow = b->row,isicol = b->icol; 626c9c72f8fSHong Zhang PetscErrorCode ierr; 627c9c72f8fSHong Zhang const PetscInt *r,*ic,*ics; 628c9c72f8fSHong Zhang PetscInt i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 629c9c72f8fSHong Zhang PetscInt *ajtmp,*bjtmp,nz,nzL,row,*bdiag=b->diag,*pj; 630c9c72f8fSHong Zhang MatScalar *rtmp,*pc,multiplier,*v,*pv,*aa=a->a; 631c9c72f8fSHong Zhang PetscTruth row_identity,col_identity; 632c9c72f8fSHong Zhang 633d4ad06f7SHong Zhang LUShift_Ctx sctx; 634d4ad06f7SHong Zhang PetscInt *ddiag,newshift; 635d4ad06f7SHong Zhang PetscReal rs; 636d4ad06f7SHong Zhang MatScalar d; 637d4ad06f7SHong Zhang 638c9c72f8fSHong Zhang PetscFunctionBegin; 639d6e5152cSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 640d6e5152cSHong Zhang ierr = PetscMemzero(&sctx,sizeof(LUShift_Ctx));CHKERRQ(ierr); 641d4ad06f7SHong Zhang 642d4ad06f7SHong Zhang /* if both shift schemes are chosen by user, only use info->shiftpd */ 643d4ad06f7SHong Zhang if (info->shiftpd) { /* set sctx.shift_top=max{rs} */ 644d4ad06f7SHong Zhang ddiag = a->diag; 645d4ad06f7SHong Zhang sctx.shift_top = info->zeropivot; 646d4ad06f7SHong Zhang for (i=0; i<n; i++) { 647d4ad06f7SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 648d4ad06f7SHong Zhang d = (aa)[ddiag[i]]; 649d4ad06f7SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 650d4ad06f7SHong Zhang v = aa+ai[i]; 651d4ad06f7SHong Zhang nz = ai[i+1] - ai[i]; 652d4ad06f7SHong Zhang for (j=0; j<nz; j++) 653d4ad06f7SHong Zhang rs += PetscAbsScalar(v[j]); 654d4ad06f7SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 655d4ad06f7SHong Zhang } 656d4ad06f7SHong Zhang sctx.shift_top *= 1.1; 657d4ad06f7SHong Zhang sctx.nshift_max = 5; 658d4ad06f7SHong Zhang sctx.shift_lo = 0.; 659d4ad06f7SHong Zhang sctx.shift_hi = 1.; 660d4ad06f7SHong Zhang } 661d4ad06f7SHong Zhang 662c9c72f8fSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 663c9c72f8fSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 664c9c72f8fSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 665c9c72f8fSHong Zhang ics = ic; 666c9c72f8fSHong Zhang 667d4ad06f7SHong Zhang do { 668d4ad06f7SHong Zhang sctx.lushift = PETSC_FALSE; 669c9c72f8fSHong Zhang for (i=0; i<n; i++){ 670c9c72f8fSHong Zhang /* zero rtmp */ 671c9c72f8fSHong Zhang /* L part */ 672c9c72f8fSHong Zhang nz = bi[i+1] - bi[i]; 673c9c72f8fSHong Zhang bjtmp = bj + bi[i]; 674c9c72f8fSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 675c9c72f8fSHong Zhang 676c9c72f8fSHong Zhang /* U part */ 6778fc3a347SHong Zhang nz = bi[2*n-i+1] - bi[2*n-i]; 6788fc3a347SHong Zhang bjtmp = bj + bi[2*n-i]; 679c9c72f8fSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 680c9c72f8fSHong Zhang 681c9c72f8fSHong Zhang /* load in initial (unfactored row) */ 682c9c72f8fSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 683c9c72f8fSHong Zhang ajtmp = aj + ai[r[i]]; 684c9c72f8fSHong Zhang v = aa + ai[r[i]]; 685c9c72f8fSHong Zhang for (j=0; j<nz; j++) { 686c9c72f8fSHong Zhang rtmp[ics[ajtmp[j]]] = v[j]; 687c9c72f8fSHong Zhang } 688d4ad06f7SHong Zhang /* ZeropivotApply() */ 689d4ad06f7SHong Zhang rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 690c9c72f8fSHong Zhang 691c9c72f8fSHong Zhang /* elimination */ 692c9c72f8fSHong Zhang bjtmp = bj + bi[i]; 693c9c72f8fSHong Zhang row = *bjtmp++; 694c9c72f8fSHong Zhang nzL = bi[i+1] - bi[i]; 695c9c72f8fSHong Zhang k = 0; 696c9c72f8fSHong Zhang while (k < nzL) { 697c9c72f8fSHong Zhang pc = rtmp + row; 698c9c72f8fSHong Zhang if (*pc != 0.0) { 699c9c72f8fSHong Zhang pv = b->a + bdiag[row]; 700c9c72f8fSHong Zhang multiplier = *pc * (*pv); 701c9c72f8fSHong Zhang *pc = multiplier; 7028fc3a347SHong Zhang pj = b->j + bi[2*n-row]; /* begining of U(row,:) */ 7038fc3a347SHong Zhang pv = b->a + bi[2*n-row]; 7048fc3a347SHong Zhang nz = bi[2*n-row+1] - bi[2*n-row] - 1; /* num of entries in U(row,:), excluding diag */ 705c9c72f8fSHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 706c9c72f8fSHong Zhang ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 707c9c72f8fSHong Zhang } 708c9c72f8fSHong Zhang row = *bjtmp++; k++; 709c9c72f8fSHong Zhang } 710c9c72f8fSHong Zhang 711c9c72f8fSHong Zhang /* finished row so stick it into b->a */ 712d4ad06f7SHong Zhang rs = 0.0; 713c9c72f8fSHong Zhang /* L part */ 714c9c72f8fSHong Zhang pv = b->a + bi[i] ; 715c9c72f8fSHong Zhang pj = b->j + bi[i] ; 716c9c72f8fSHong Zhang nz = bi[i+1] - bi[i]; 717c9c72f8fSHong Zhang for (j=0; j<nz; j++) { 718d4ad06f7SHong Zhang pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 719c9c72f8fSHong Zhang } 720c9c72f8fSHong Zhang 721c9c72f8fSHong Zhang /* U part */ 7228fc3a347SHong Zhang pv = b->a + bi[2*n-i]; 7238fc3a347SHong Zhang pj = b->j + bi[2*n-i]; 7248fc3a347SHong Zhang nz = bi[2*n-i+1] - bi[2*n-i] - 1; 725d4ad06f7SHong Zhang for (j=0; j<nz; j++) { 726d4ad06f7SHong Zhang pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 727c9c72f8fSHong Zhang } 728d4ad06f7SHong Zhang 729d6e5152cSHong Zhang /* MatPivotCheck() */ 730d4ad06f7SHong Zhang sctx.rs = rs; 731d4ad06f7SHong Zhang sctx.pv = rtmp[i]; 732d6e5152cSHong Zhang /* ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); */ 733d6e5152cSHong Zhang if (info->shiftnz){ 734d6e5152cSHong Zhang ierr = MatPivotCheck_nz(info,sctx,i,newshift);CHKERRQ(ierr); 735d6e5152cSHong Zhang } else if (info->shiftpd){ 736d6e5152cSHong Zhang ierr = MatPivotCheck_pd(info,sctx,i,newshift);CHKERRQ(ierr); 737d6e5152cSHong Zhang } else if (info->shiftinblocks){ 738d6e5152cSHong Zhang ierr = MatPivotCheck_inblocks(info,sctx,i,newshift);CHKERRQ(ierr); 739d6e5152cSHong Zhang } else { 740d6e5152cSHong Zhang ierr = MatPivotCheck_none(info,sctx,i,newshift);CHKERRQ(ierr); 741d6e5152cSHong Zhang } 742d6e5152cSHong Zhang rtmp[i] = sctx.pv; 743d4ad06f7SHong Zhang if (newshift == 1) break; 744d4ad06f7SHong Zhang 745d4ad06f7SHong Zhang /* Mark diagonal and invert diagonal for simplier triangular solves */ 746d4ad06f7SHong Zhang pv = b->a + bdiag[i]; 747d4ad06f7SHong Zhang *pv = 1.0/rtmp[i]; 748d4ad06f7SHong Zhang 749d4ad06f7SHong Zhang } /* endof for (i=0; i<n; i++){ */ 750d4ad06f7SHong Zhang 751d6e5152cSHong Zhang /* MatPivotRefine() */ 752d4ad06f7SHong Zhang if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max){ 753d4ad06f7SHong Zhang /* 754d4ad06f7SHong Zhang * if no shift in this attempt & shifting & started shifting & can refine, 755d4ad06f7SHong Zhang * then try lower shift 756d4ad06f7SHong Zhang */ 757d4ad06f7SHong Zhang sctx.shift_hi = sctx.shift_fraction; 758d4ad06f7SHong Zhang sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 759d4ad06f7SHong Zhang sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 760d4ad06f7SHong Zhang sctx.lushift = PETSC_TRUE; 761d4ad06f7SHong Zhang sctx.nshift++; 762d4ad06f7SHong Zhang } 763d4ad06f7SHong Zhang } while (sctx.lushift); 764d4ad06f7SHong Zhang 765c9c72f8fSHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 766c9c72f8fSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 767c9c72f8fSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 768c9c72f8fSHong Zhang 769c9c72f8fSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 770c9c72f8fSHong Zhang ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 771c9c72f8fSHong Zhang if (row_identity && col_identity) { 772894a2336SHong Zhang C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct; 773c9c72f8fSHong Zhang } else { 774894a2336SHong Zhang C->ops->solve = MatSolve_SeqAIJ_newdatastruct; 775c9c72f8fSHong Zhang } 776c9c72f8fSHong Zhang 777c9c72f8fSHong Zhang C->ops->solveadd = 0; 778c9c72f8fSHong Zhang C->ops->solvetranspose = 0; 779c9c72f8fSHong Zhang C->ops->solvetransposeadd = 0; 780c9c72f8fSHong Zhang C->ops->matsolve = 0; 781c9c72f8fSHong Zhang C->assembled = PETSC_TRUE; 782c9c72f8fSHong Zhang C->preallocated = PETSC_TRUE; 783c9c72f8fSHong Zhang ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 784d4ad06f7SHong Zhang 785d6e5152cSHong Zhang /* MatPivotView() */ 786d4ad06f7SHong Zhang if (sctx.nshift){ 787d4ad06f7SHong Zhang if (info->shiftpd) { 788d4ad06f7SHong 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); 789d4ad06f7SHong Zhang } else if (info->shiftnz) { 790d4ad06f7SHong Zhang ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 791d4ad06f7SHong Zhang } 792d4ad06f7SHong Zhang } 793c9c72f8fSHong Zhang PetscFunctionReturn(0); 794c9c72f8fSHong Zhang } 795c9c72f8fSHong Zhang 796813a1f2bSShri Abhyankar /* ----------------------------------------------------------- */ 797813a1f2bSShri Abhyankar extern PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_newdatastruct_v2(Mat,Vec,Vec); 798f268cba8SShri Abhyankar extern PetscErrorCode MatSolve_SeqAIJ_newdatastruct_v2(Mat,Vec,Vec); 799813a1f2bSShri Abhyankar 800813a1f2bSShri Abhyankar #undef __FUNCT__ 801813a1f2bSShri Abhyankar #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_newdatastruct_v2" 802813a1f2bSShri Abhyankar PetscErrorCode MatLUFactorNumeric_SeqAIJ_newdatastruct_v2(Mat B,Mat A,const MatFactorInfo *info) 803813a1f2bSShri Abhyankar { 804813a1f2bSShri Abhyankar Mat C=B; 805813a1f2bSShri Abhyankar Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 806813a1f2bSShri Abhyankar IS isrow = b->row,isicol = b->icol; 807813a1f2bSShri Abhyankar PetscErrorCode ierr; 808813a1f2bSShri Abhyankar const PetscInt *r,*ic,*ics; 809813a1f2bSShri Abhyankar PetscInt i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 810813a1f2bSShri Abhyankar PetscInt *ajtmp,*bjtmp,nz,nzL,row,*bdiag=b->diag,*pj; 811813a1f2bSShri Abhyankar MatScalar *rtmp,*pc,multiplier,*v,*pv,*aa=a->a; 812813a1f2bSShri Abhyankar PetscTruth row_identity,col_identity; 813813a1f2bSShri Abhyankar 814813a1f2bSShri Abhyankar LUShift_Ctx sctx; 815813a1f2bSShri Abhyankar PetscInt *ddiag,newshift; 816813a1f2bSShri Abhyankar PetscReal rs; 817813a1f2bSShri Abhyankar MatScalar d; 818813a1f2bSShri Abhyankar 819813a1f2bSShri Abhyankar PetscFunctionBegin; 8208ff23777SHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 8218ff23777SHong Zhang ierr = PetscMemzero(&sctx,sizeof(LUShift_Ctx));CHKERRQ(ierr); 822813a1f2bSShri Abhyankar 823813a1f2bSShri Abhyankar /* if both shift schemes are chosen by user, only use info->shiftpd */ 824813a1f2bSShri Abhyankar if (info->shiftpd) { /* set sctx.shift_top=max{rs} */ 825813a1f2bSShri Abhyankar ddiag = a->diag; 826813a1f2bSShri Abhyankar sctx.shift_top = info->zeropivot; 827813a1f2bSShri Abhyankar for (i=0; i<n; i++) { 828813a1f2bSShri Abhyankar /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 829813a1f2bSShri Abhyankar d = (aa)[ddiag[i]]; 830813a1f2bSShri Abhyankar rs = -PetscAbsScalar(d) - PetscRealPart(d); 831813a1f2bSShri Abhyankar v = aa+ai[i]; 832813a1f2bSShri Abhyankar nz = ai[i+1] - ai[i]; 833813a1f2bSShri Abhyankar for (j=0; j<nz; j++) 834813a1f2bSShri Abhyankar rs += PetscAbsScalar(v[j]); 835813a1f2bSShri Abhyankar if (rs>sctx.shift_top) sctx.shift_top = rs; 836813a1f2bSShri Abhyankar } 837813a1f2bSShri Abhyankar sctx.shift_top *= 1.1; 838813a1f2bSShri Abhyankar sctx.nshift_max = 5; 839813a1f2bSShri Abhyankar sctx.shift_lo = 0.; 840813a1f2bSShri Abhyankar sctx.shift_hi = 1.; 841813a1f2bSShri Abhyankar } 842813a1f2bSShri Abhyankar 843813a1f2bSShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 844813a1f2bSShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 845813a1f2bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 846813a1f2bSShri Abhyankar ics = ic; 847813a1f2bSShri Abhyankar 848813a1f2bSShri Abhyankar do { 849813a1f2bSShri Abhyankar sctx.lushift = PETSC_FALSE; 850813a1f2bSShri Abhyankar for (i=0; i<n; i++){ 851813a1f2bSShri Abhyankar /* zero rtmp */ 852813a1f2bSShri Abhyankar /* L part */ 853813a1f2bSShri Abhyankar nz = bi[i+1] - bi[i]; 854813a1f2bSShri Abhyankar bjtmp = bj + bi[i]; 855813a1f2bSShri Abhyankar for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 856813a1f2bSShri Abhyankar 857813a1f2bSShri Abhyankar /* U part */ 85862fbd6afSShri Abhyankar /* nz = bi[2*n-i+1] - bi[2*n-i]; 85962fbd6afSShri Abhyankar bjtmp = bj + bi[2*n-i]; */ 86062fbd6afSShri Abhyankar nz = bdiag[i]-bdiag[i+1]; 86162fbd6afSShri Abhyankar bjtmp = bj + bdiag[i+1]+1; 862813a1f2bSShri Abhyankar for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 863813a1f2bSShri Abhyankar 864813a1f2bSShri Abhyankar /* load in initial (unfactored row) */ 865813a1f2bSShri Abhyankar nz = ai[r[i]+1] - ai[r[i]]; 866813a1f2bSShri Abhyankar ajtmp = aj + ai[r[i]]; 867813a1f2bSShri Abhyankar v = aa + ai[r[i]]; 868813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 869813a1f2bSShri Abhyankar rtmp[ics[ajtmp[j]]] = v[j]; 870813a1f2bSShri Abhyankar } 871813a1f2bSShri Abhyankar /* ZeropivotApply() */ 872813a1f2bSShri Abhyankar rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 873813a1f2bSShri Abhyankar 874813a1f2bSShri Abhyankar /* elimination */ 875813a1f2bSShri Abhyankar bjtmp = bj + bi[i]; 876813a1f2bSShri Abhyankar row = *bjtmp++; 877813a1f2bSShri Abhyankar nzL = bi[i+1] - bi[i]; 878f268cba8SShri Abhyankar for(k=0; k < nzL;k++) { 879813a1f2bSShri Abhyankar pc = rtmp + row; 880813a1f2bSShri Abhyankar if (*pc != 0.0) { 881813a1f2bSShri Abhyankar pv = b->a + bdiag[row]; 882813a1f2bSShri Abhyankar multiplier = *pc * (*pv); 883813a1f2bSShri Abhyankar *pc = multiplier; 88462fbd6afSShri Abhyankar /* pj = b->j + bi[2*n-row]; */ /* begining of U(row,:) */ 88562fbd6afSShri Abhyankar /* pv = b->a + bi[2*n-row]; 88662fbd6afSShri Abhyankar nz = bi[2*n-row+1] - bi[2*n-row] - 1; */ /* num of entries in U(row,:), excluding diag */ 88762fbd6afSShri Abhyankar pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */ 88862fbd6afSShri Abhyankar pv = b->a + bdiag[row+1]+1; 88962fbd6afSShri Abhyankar nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */ 890813a1f2bSShri Abhyankar for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 891813a1f2bSShri Abhyankar ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 892813a1f2bSShri Abhyankar } 893f268cba8SShri Abhyankar row = *bjtmp++; 894813a1f2bSShri Abhyankar } 895813a1f2bSShri Abhyankar 896813a1f2bSShri Abhyankar /* finished row so stick it into b->a */ 897813a1f2bSShri Abhyankar rs = 0.0; 898813a1f2bSShri Abhyankar /* L part */ 899813a1f2bSShri Abhyankar pv = b->a + bi[i] ; 900813a1f2bSShri Abhyankar pj = b->j + bi[i] ; 901813a1f2bSShri Abhyankar nz = bi[i+1] - bi[i]; 902813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 903813a1f2bSShri Abhyankar pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 904813a1f2bSShri Abhyankar } 905813a1f2bSShri Abhyankar 906813a1f2bSShri Abhyankar /* U part */ 90762fbd6afSShri Abhyankar /* pv = b->a + bi[2*n-i]; 908813a1f2bSShri Abhyankar pj = b->j + bi[2*n-i]; 909813a1f2bSShri Abhyankar nz = bi[2*n-i+1] - bi[2*n-i] - 1; 91062fbd6afSShri Abhyankar */ 91162fbd6afSShri Abhyankar pv = b->a + bdiag[i+1]+1; 91262fbd6afSShri Abhyankar pj = b->j + bdiag[i+1]+1; 91362fbd6afSShri Abhyankar nz = bdiag[i] - bdiag[i+1]-1; 914813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 915813a1f2bSShri Abhyankar pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 916813a1f2bSShri Abhyankar } 917813a1f2bSShri Abhyankar 9188ff23777SHong Zhang /* MatPivotCheck() */ 919813a1f2bSShri Abhyankar sctx.rs = rs; 920813a1f2bSShri Abhyankar sctx.pv = rtmp[i]; 9218ff23777SHong Zhang /* ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); */ 9228ff23777SHong Zhang if (info->shiftnz){ 9238ff23777SHong Zhang ierr = MatPivotCheck_nz(info,sctx,i,newshift);CHKERRQ(ierr); 9248ff23777SHong Zhang } else if (info->shiftpd){ 9258ff23777SHong Zhang ierr = MatPivotCheck_pd(info,sctx,i,newshift);CHKERRQ(ierr); 9268ff23777SHong Zhang } else if (info->shiftinblocks){ 9278ff23777SHong Zhang ierr = MatPivotCheck_inblocks(info,sctx,i,newshift);CHKERRQ(ierr); 9288ff23777SHong Zhang } else { 9298ff23777SHong Zhang ierr = MatPivotCheck_none(info,sctx,i,newshift);CHKERRQ(ierr); 9308ff23777SHong Zhang } 9318ff23777SHong Zhang rtmp[i] = sctx.pv; 932813a1f2bSShri Abhyankar if (newshift == 1) break; 933813a1f2bSShri Abhyankar 934813a1f2bSShri Abhyankar /* Mark diagonal and invert diagonal for simplier triangular solves */ 935813a1f2bSShri Abhyankar pv = b->a + bdiag[i]; 936813a1f2bSShri Abhyankar *pv = 1.0/rtmp[i]; 937813a1f2bSShri Abhyankar 938813a1f2bSShri Abhyankar } /* endof for (i=0; i<n; i++){ */ 939813a1f2bSShri Abhyankar 9408ff23777SHong Zhang /* MatPivotRefine() */ 941813a1f2bSShri Abhyankar if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max){ 942813a1f2bSShri Abhyankar /* 943813a1f2bSShri Abhyankar * if no shift in this attempt & shifting & started shifting & can refine, 944813a1f2bSShri Abhyankar * then try lower shift 945813a1f2bSShri Abhyankar */ 946813a1f2bSShri Abhyankar sctx.shift_hi = sctx.shift_fraction; 947813a1f2bSShri Abhyankar sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 948813a1f2bSShri Abhyankar sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 949813a1f2bSShri Abhyankar sctx.lushift = PETSC_TRUE; 950813a1f2bSShri Abhyankar sctx.nshift++; 951813a1f2bSShri Abhyankar } 952813a1f2bSShri Abhyankar } while (sctx.lushift); 953813a1f2bSShri Abhyankar 954813a1f2bSShri Abhyankar ierr = PetscFree(rtmp);CHKERRQ(ierr); 955813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 956813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 957813a1f2bSShri Abhyankar 958813a1f2bSShri Abhyankar ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 959813a1f2bSShri Abhyankar ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 960813a1f2bSShri Abhyankar if (row_identity && col_identity) { 96162fbd6afSShri Abhyankar C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct_v2; 962813a1f2bSShri Abhyankar } else { 963f268cba8SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ_newdatastruct_v2; 964813a1f2bSShri Abhyankar } 965813a1f2bSShri Abhyankar 966813a1f2bSShri Abhyankar C->ops->solveadd = 0; 967813a1f2bSShri Abhyankar C->ops->solvetranspose = 0; 968813a1f2bSShri Abhyankar C->ops->solvetransposeadd = 0; 969813a1f2bSShri Abhyankar C->ops->matsolve = 0; 970813a1f2bSShri Abhyankar C->assembled = PETSC_TRUE; 971813a1f2bSShri Abhyankar C->preallocated = PETSC_TRUE; 972813a1f2bSShri Abhyankar ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 973813a1f2bSShri Abhyankar 9748ff23777SHong Zhang /* MatPivotView() */ 975813a1f2bSShri Abhyankar if (sctx.nshift){ 976813a1f2bSShri Abhyankar if (info->shiftpd) { 977813a1f2bSShri 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); 978813a1f2bSShri Abhyankar } else if (info->shiftnz) { 979813a1f2bSShri Abhyankar ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 9808ff23777SHong Zhang } else if (info->shiftinblocks){ 9818ff23777SHong Zhang ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftinblocks);CHKERRQ(ierr); 982813a1f2bSShri Abhyankar } 983813a1f2bSShri Abhyankar } 984813a1f2bSShri Abhyankar PetscFunctionReturn(0); 985813a1f2bSShri Abhyankar } 986813a1f2bSShri Abhyankar 9874a2ae208SSatish Balay #undef __FUNCT__ 9884a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ" 9890481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 990289bc588SBarry Smith { 991719d5645SBarry Smith Mat C=B; 992416022c9SBarry Smith Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 99382bf6240SBarry Smith IS isrow = b->row,isicol = b->icol; 9946849ba73SBarry Smith PetscErrorCode ierr; 9955d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 996d278edc7SBarry Smith PetscInt nz,row,i,j,n=A->rmap->n,diag; 997d278edc7SBarry Smith const PetscInt *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 998d278edc7SBarry Smith const PetscInt *ajtmp,*bjtmp,*diag_offset = b->diag,*pj; 999d278edc7SBarry Smith MatScalar *pv,*rtmp,*pc,multiplier,d; 1000d278edc7SBarry Smith const MatScalar *v,*aa=a->a; 100133f9893dSHong Zhang PetscReal rs=0.0; 1002b3bf805bSHong Zhang LUShift_Ctx sctx; 100342898933SHong Zhang PetscInt newshift,*ddiag; 1004289bc588SBarry Smith 10053a40ed3dSBarry Smith PetscFunctionBegin; 100678b31e54SBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 100778b31e54SBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 1008b78a477dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 10099f7d0b68SHong Zhang ics = ic; 1010289bc588SBarry Smith 1011d4ad06f7SHong Zhang /* initialize shift context sctx */ 1012d4ad06f7SHong Zhang sctx.nshift = 0; 1013ada7143aSSatish Balay sctx.nshift_max = 0; 1014d4ad06f7SHong Zhang sctx.shift_top = 0.0; 1015d4ad06f7SHong Zhang sctx.shift_lo = 0.0; 1016d4ad06f7SHong Zhang sctx.shift_hi = 0.0; 1017d4ad06f7SHong Zhang sctx.shift_fraction = 0.0; 1018d4ad06f7SHong Zhang sctx.shift_amount = 0.0; 1019ada7143aSSatish Balay 1020118f5789SBarry Smith /* if both shift schemes are chosen by user, only use info->shiftpd */ 10211a890ab1SHong Zhang if (info->shiftpd) { /* set sctx.shift_top=max{rs} */ 102242898933SHong Zhang ddiag = a->diag; 10239f7d0b68SHong Zhang sctx.shift_top = info->zeropivot; 10246cc28720Svictorle for (i=0; i<n; i++) { 10259f95998fSHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 10269f7d0b68SHong Zhang d = (aa)[ddiag[i]]; 10271a890ab1SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 10289f7d0b68SHong Zhang v = aa+ai[i]; 10299f7d0b68SHong Zhang nz = ai[i+1] - ai[i]; 1030e24cfe64SHong Zhang for (j=0; j<nz; j++) 10311a890ab1SHong Zhang rs += PetscAbsScalar(v[j]); 10321a890ab1SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 10336cc28720Svictorle } 1034118f5789SBarry Smith sctx.shift_top *= 1.1; 1035d2276718SHong Zhang sctx.nshift_max = 5; 1036d2276718SHong Zhang sctx.shift_lo = 0.; 1037d2276718SHong Zhang sctx.shift_hi = 1.; 1038a0a356efSHong Zhang } 1039a0a356efSHong Zhang 1040085256b3SBarry Smith do { 1041d2276718SHong Zhang sctx.lushift = PETSC_FALSE; 1042289bc588SBarry Smith for (i=0; i<n; i++){ 1043b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 1044b3bf805bSHong Zhang bjtmp = bj + bi[i]; 10459f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 1046289bc588SBarry Smith 1047289bc588SBarry Smith /* load in initial (unfactored row) */ 10489f7d0b68SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 10499f7d0b68SHong Zhang ajtmp = aj + ai[r[i]]; 10509f7d0b68SHong Zhang v = aa + ai[r[i]]; 1051085256b3SBarry Smith for (j=0; j<nz; j++) { 1052b3bf805bSHong Zhang rtmp[ics[ajtmp[j]]] = v[j]; 1053085256b3SBarry Smith } 1054d2276718SHong Zhang rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 1055bd1bc851SBarry Smith /* if (sctx.shift_amount > 0.0) printf("row %d, shift %g\n",i,sctx.shift_amount); */ 1056289bc588SBarry Smith 1057b3bf805bSHong Zhang row = *bjtmp++; 1058f2582111SSatish Balay while (row < i) { 10598c37ef55SBarry Smith pc = rtmp + row; 10608c37ef55SBarry Smith if (*pc != 0.0) { 1061010a20caSHong Zhang pv = b->a + diag_offset[row]; 1062010a20caSHong Zhang pj = b->j + diag_offset[row] + 1; 106335aab85fSBarry Smith multiplier = *pc / *pv++; 10648c37ef55SBarry Smith *pc = multiplier; 1065b3bf805bSHong Zhang nz = bi[row+1] - diag_offset[row] - 1; 10669f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 1067dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 1068289bc588SBarry Smith } 1069b3bf805bSHong Zhang row = *bjtmp++; 1070289bc588SBarry Smith } 1071416022c9SBarry Smith /* finished row so stick it into b->a */ 1072b3bf805bSHong Zhang pv = b->a + bi[i] ; 1073b3bf805bSHong Zhang pj = b->j + bi[i] ; 1074b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 1075b3bf805bSHong Zhang diag = diag_offset[i] - bi[i]; 1076e57ff13aSBarry Smith rs = 0.0; 1077d3d32019SBarry Smith for (j=0; j<nz; j++) { 10789f7d0b68SHong Zhang pv[j] = rtmp[pj[j]]; 10799f7d0b68SHong Zhang rs += PetscAbsScalar(pv[j]); 1080d3d32019SBarry Smith } 1081e57ff13aSBarry Smith rs -= PetscAbsScalar(pv[diag]); 1082d2276718SHong Zhang 1083b3bf805bSHong Zhang /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */ 1084d2276718SHong Zhang sctx.rs = rs; 1085d2276718SHong Zhang sctx.pv = pv[diag]; 1086c6b1f410SHong Zhang ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); 10875b5bc046SBarry Smith if (newshift == 1) break; 108835aab85fSBarry Smith } 1089d2276718SHong Zhang 10900481f469SBarry Smith if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 10916cc28720Svictorle /* 10926c037d1bSvictorle * if no shift in this attempt & shifting & started shifting & can refine, 10936cc28720Svictorle * then try lower shift 10946cc28720Svictorle */ 10950481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 10960481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 10970481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 1098d2276718SHong Zhang sctx.lushift = PETSC_TRUE; 1099d2276718SHong Zhang sctx.nshift++; 11006cc28720Svictorle } 1101d2276718SHong Zhang } while (sctx.lushift); 11020f11f9aeSSatish Balay 110335aab85fSBarry Smith /* invert diagonal entries for simplier triangular solves */ 110435aab85fSBarry Smith for (i=0; i<n; i++) { 1105010a20caSHong Zhang b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]]; 110635aab85fSBarry Smith } 1107606d414cSSatish Balay ierr = PetscFree(rtmp);CHKERRQ(ierr); 110878b31e54SBarry Smith ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 110978b31e54SBarry Smith ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 1110db4efbfdSBarry Smith if (b->inode.use) { 11114108e4d5SBarry Smith C->ops->solve = MatSolve_SeqAIJ_Inode; 1112db4efbfdSBarry Smith } else { 11138d8c7c43SBarry Smith PetscTruth row_identity, col_identity; 11148d8c7c43SBarry Smith ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 11158d8c7c43SBarry Smith ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 11168d8c7c43SBarry Smith if (row_identity && col_identity) { 11178d8c7c43SBarry Smith C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 11188d8c7c43SBarry Smith } else { 1119db4efbfdSBarry Smith C->ops->solve = MatSolve_SeqAIJ; 1120db4efbfdSBarry Smith } 11218d8c7c43SBarry Smith } 1122db4efbfdSBarry Smith C->ops->solveadd = MatSolveAdd_SeqAIJ; 1123db4efbfdSBarry Smith C->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 1124db4efbfdSBarry Smith C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 1125de26497bSBarry Smith C->ops->matsolve = MatMatSolve_SeqAIJ; 1126c456f294SBarry Smith C->assembled = PETSC_TRUE; 11275c9eb25fSBarry Smith C->preallocated = PETSC_TRUE; 1128d0f46423SBarry Smith ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 1129d2276718SHong Zhang if (sctx.nshift){ 1130e738e422SBarry Smith if (info->shiftpd) { 11310481f469SBarry 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); 1132e738e422SBarry Smith } else if (info->shiftnz) { 1133e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 11346cc28720Svictorle } 11350697b6caSHong Zhang } 11363a40ed3dSBarry Smith PetscFunctionReturn(0); 1137289bc588SBarry Smith } 11380889b5dcSvictorle 1139137fb511SHong Zhang /* 1140137fb511SHong Zhang This routine implements inplace ILU(0) with row or/and column permutations. 1141137fb511SHong Zhang Input: 1142137fb511SHong Zhang A - original matrix 1143137fb511SHong Zhang Output; 1144137fb511SHong Zhang A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i] 1145137fb511SHong Zhang a->j (col index) is permuted by the inverse of colperm, then sorted 1146137fb511SHong Zhang a->a reordered accordingly with a->j 1147137fb511SHong Zhang a->diag (ptr to diagonal elements) is updated. 1148137fb511SHong Zhang */ 1149137fb511SHong Zhang #undef __FUNCT__ 1150137fb511SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_InplaceWithPerm" 11510481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info) 1152137fb511SHong Zhang { 1153b051339dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 1154b051339dSHong Zhang IS isrow = a->row,isicol = a->icol; 1155137fb511SHong Zhang PetscErrorCode ierr; 11565d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 11575d0c19d7SBarry Smith PetscInt i,j,n=A->rmap->n,*ai=a->i,*aj=a->j; 11585d0c19d7SBarry Smith PetscInt *ajtmp,nz,row; 1159b051339dSHong Zhang PetscInt *diag = a->diag,nbdiag,*pj; 1160a77337e4SBarry Smith PetscScalar *rtmp,*pc,multiplier,d; 1161a77337e4SBarry Smith MatScalar *v,*pv; 1162137fb511SHong Zhang PetscReal rs; 1163137fb511SHong Zhang LUShift_Ctx sctx; 1164b051339dSHong Zhang PetscInt newshift; 1165137fb511SHong Zhang 1166137fb511SHong Zhang PetscFunctionBegin; 1167719d5645SBarry Smith if (A != B) SETERRQ(PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address"); 1168137fb511SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1169137fb511SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 1170137fb511SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr); 1171137fb511SHong Zhang ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 1172b051339dSHong Zhang ics = ic; 1173137fb511SHong Zhang 1174137fb511SHong Zhang sctx.shift_top = 0; 1175137fb511SHong Zhang sctx.nshift_max = 0; 1176137fb511SHong Zhang sctx.shift_lo = 0; 1177137fb511SHong Zhang sctx.shift_hi = 0; 11780481f469SBarry Smith sctx.shift_fraction = 0; 1179137fb511SHong Zhang 1180137fb511SHong Zhang /* if both shift schemes are chosen by user, only use info->shiftpd */ 1181137fb511SHong Zhang if (info->shiftpd) { /* set sctx.shift_top=max{rs} */ 1182137fb511SHong Zhang sctx.shift_top = 0; 1183137fb511SHong Zhang for (i=0; i<n; i++) { 1184137fb511SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 1185b051339dSHong Zhang d = (a->a)[diag[i]]; 1186137fb511SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 1187b051339dSHong Zhang v = a->a+ai[i]; 1188b051339dSHong Zhang nz = ai[i+1] - ai[i]; 1189137fb511SHong Zhang for (j=0; j<nz; j++) 1190137fb511SHong Zhang rs += PetscAbsScalar(v[j]); 1191137fb511SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 1192137fb511SHong Zhang } 1193137fb511SHong Zhang if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot; 1194137fb511SHong Zhang sctx.shift_top *= 1.1; 1195137fb511SHong Zhang sctx.nshift_max = 5; 1196137fb511SHong Zhang sctx.shift_lo = 0.; 1197137fb511SHong Zhang sctx.shift_hi = 1.; 1198137fb511SHong Zhang } 1199137fb511SHong Zhang 1200137fb511SHong Zhang sctx.shift_amount = 0; 1201137fb511SHong Zhang sctx.nshift = 0; 1202137fb511SHong Zhang do { 1203137fb511SHong Zhang sctx.lushift = PETSC_FALSE; 1204137fb511SHong Zhang for (i=0; i<n; i++){ 1205b051339dSHong Zhang /* load in initial unfactored row */ 1206b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 1207b051339dSHong Zhang ajtmp = aj + ai[r[i]]; 1208b051339dSHong Zhang v = a->a + ai[r[i]]; 1209137fb511SHong Zhang /* sort permuted ajtmp and values v accordingly */ 1210b051339dSHong Zhang for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]]; 1211137fb511SHong Zhang ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr); 1212137fb511SHong Zhang 1213b051339dSHong Zhang diag[r[i]] = ai[r[i]]; 1214137fb511SHong Zhang for (j=0; j<nz; j++) { 1215137fb511SHong Zhang rtmp[ajtmp[j]] = v[j]; 1216b051339dSHong Zhang if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */ 1217137fb511SHong Zhang } 1218137fb511SHong Zhang rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 1219137fb511SHong Zhang 1220b051339dSHong Zhang row = *ajtmp++; 1221137fb511SHong Zhang while (row < i) { 1222137fb511SHong Zhang pc = rtmp + row; 1223137fb511SHong Zhang if (*pc != 0.0) { 1224b051339dSHong Zhang pv = a->a + diag[r[row]]; 1225b051339dSHong Zhang pj = aj + diag[r[row]] + 1; 1226137fb511SHong Zhang 1227137fb511SHong Zhang multiplier = *pc / *pv++; 1228137fb511SHong Zhang *pc = multiplier; 1229b051339dSHong Zhang nz = ai[r[row]+1] - diag[r[row]] - 1; 1230b051339dSHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 1231dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 1232137fb511SHong Zhang } 1233b051339dSHong Zhang row = *ajtmp++; 1234137fb511SHong Zhang } 1235b051339dSHong Zhang /* finished row so overwrite it onto a->a */ 1236b051339dSHong Zhang pv = a->a + ai[r[i]] ; 1237b051339dSHong Zhang pj = aj + ai[r[i]] ; 1238b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 1239b051339dSHong Zhang nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */ 1240137fb511SHong Zhang 1241137fb511SHong Zhang rs = 0.0; 1242137fb511SHong Zhang for (j=0; j<nz; j++) { 1243b051339dSHong Zhang pv[j] = rtmp[pj[j]]; 1244b051339dSHong Zhang if (j != nbdiag) rs += PetscAbsScalar(pv[j]); 1245137fb511SHong Zhang } 1246137fb511SHong Zhang 1247137fb511SHong Zhang /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */ 1248137fb511SHong Zhang sctx.rs = rs; 1249b051339dSHong Zhang sctx.pv = pv[nbdiag]; 1250c6b1f410SHong Zhang ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); 1251137fb511SHong Zhang if (newshift == 1) break; 1252137fb511SHong Zhang } 1253137fb511SHong Zhang 12540481f469SBarry Smith if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 1255137fb511SHong Zhang /* 1256137fb511SHong Zhang * if no shift in this attempt & shifting & started shifting & can refine, 1257137fb511SHong Zhang * then try lower shift 1258137fb511SHong Zhang */ 12590481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 12600481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 12610481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 1262137fb511SHong Zhang sctx.lushift = PETSC_TRUE; 1263137fb511SHong Zhang sctx.nshift++; 1264137fb511SHong Zhang } 1265137fb511SHong Zhang } while (sctx.lushift); 1266137fb511SHong Zhang 1267137fb511SHong Zhang /* invert diagonal entries for simplier triangular solves */ 1268137fb511SHong Zhang for (i=0; i<n; i++) { 1269b051339dSHong Zhang a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]]; 1270137fb511SHong Zhang } 1271137fb511SHong Zhang 1272137fb511SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 1273137fb511SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 1274137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 1275b051339dSHong Zhang A->ops->solve = MatSolve_SeqAIJ_InplaceWithPerm; 1276db4efbfdSBarry Smith A->ops->solveadd = MatSolveAdd_SeqAIJ; 1277db4efbfdSBarry Smith A->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 1278db4efbfdSBarry Smith A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 1279b051339dSHong Zhang A->assembled = PETSC_TRUE; 12805c9eb25fSBarry Smith A->preallocated = PETSC_TRUE; 1281d0f46423SBarry Smith ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr); 1282137fb511SHong Zhang if (sctx.nshift){ 1283e738e422SBarry Smith if (info->shiftpd) { 12840481f469SBarry 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); 1285e738e422SBarry Smith } else if (info->shiftnz) { 1286e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 1287137fb511SHong Zhang } 1288137fb511SHong Zhang } 1289137fb511SHong Zhang PetscFunctionReturn(0); 1290137fb511SHong Zhang } 1291137fb511SHong Zhang 12920a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 12934a2ae208SSatish Balay #undef __FUNCT__ 12944a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ" 12950481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info) 1296da3a660dSBarry Smith { 1297dfbe8321SBarry Smith PetscErrorCode ierr; 1298416022c9SBarry Smith Mat C; 1299416022c9SBarry Smith 13003a40ed3dSBarry Smith PetscFunctionBegin; 130143244d56SBarry Smith ierr = MatGetFactor(A,MAT_SOLVER_PETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr); 1302719d5645SBarry Smith ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr); 1303719d5645SBarry Smith ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr); 1304db4efbfdSBarry Smith A->ops->solve = C->ops->solve; 1305db4efbfdSBarry Smith A->ops->solvetranspose = C->ops->solvetranspose; 1306273d9f13SBarry Smith ierr = MatHeaderCopy(A,C);CHKERRQ(ierr); 130752e6d16bSBarry Smith ierr = PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr); 13083a40ed3dSBarry Smith PetscFunctionReturn(0); 1309da3a660dSBarry Smith } 13100a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 13111d098868SHong Zhang 1312889a8dedSBarry Smith 13134a2ae208SSatish Balay #undef __FUNCT__ 13144a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ" 1315dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx) 13168c37ef55SBarry Smith { 1317416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1318416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 13196849ba73SBarry Smith PetscErrorCode ierr; 13205d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 13215d0c19d7SBarry Smith PetscInt nz; 13225d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 1323dd6ea824SBarry Smith PetscScalar *x,*tmp,*tmps,sum; 1324d9fead3dSBarry Smith const PetscScalar *b; 1325dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 13268c37ef55SBarry Smith 13273a40ed3dSBarry Smith PetscFunctionBegin; 13283a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 132991bf9adeSBarry Smith 1330d9fead3dSBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 13311ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1332416022c9SBarry Smith tmp = a->solve_work; 13338c37ef55SBarry Smith 13343b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 13353b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 13368c37ef55SBarry Smith 13378c37ef55SBarry Smith /* forward solve the lower triangular */ 13388c37ef55SBarry Smith tmp[0] = b[*r++]; 1339010a20caSHong Zhang tmps = tmp; 13408c37ef55SBarry Smith for (i=1; i<n; i++) { 1341010a20caSHong Zhang v = aa + ai[i] ; 1342010a20caSHong Zhang vi = aj + ai[i] ; 134353e7425aSBarry Smith nz = a->diag[i] - ai[i]; 134453e7425aSBarry Smith sum = b[*r++]; 1345003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 13468c37ef55SBarry Smith tmp[i] = sum; 13478c37ef55SBarry Smith } 13488c37ef55SBarry Smith 13498c37ef55SBarry Smith /* backward solve the upper triangular */ 13508c37ef55SBarry Smith for (i=n-1; i>=0; i--){ 1351010a20caSHong Zhang v = aa + a->diag[i] + 1; 1352010a20caSHong Zhang vi = aj + a->diag[i] + 1; 1353416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 13548c37ef55SBarry Smith sum = tmp[i]; 1355003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1356010a20caSHong Zhang x[*c--] = tmp[i] = sum*aa[a->diag[i]]; 13578c37ef55SBarry Smith } 13588c37ef55SBarry Smith 13593b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 13603b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1361d9fead3dSBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 13621ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1363dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 13643a40ed3dSBarry Smith PetscFunctionReturn(0); 13658c37ef55SBarry Smith } 1366026e39d0SSatish Balay 13672bebee5dSHong Zhang #undef __FUNCT__ 13682bebee5dSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ" 13692bebee5dSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X) 13702bebee5dSHong Zhang { 13712bebee5dSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 13722bebee5dSHong Zhang IS iscol = a->col,isrow = a->row; 13732bebee5dSHong Zhang PetscErrorCode ierr; 13745d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 13755d0c19d7SBarry Smith PetscInt nz,neq; 13765d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 1377dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1378dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1379db4efbfdSBarry Smith PetscTruth bisdense,xisdense; 13802bebee5dSHong Zhang 13812bebee5dSHong Zhang PetscFunctionBegin; 13822bebee5dSHong Zhang if (!n) PetscFunctionReturn(0); 13832bebee5dSHong Zhang 1384db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr); 1385db4efbfdSBarry Smith if (!bisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix"); 1386db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr); 1387db4efbfdSBarry Smith if (!xisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix"); 1388db4efbfdSBarry Smith 13892bebee5dSHong Zhang ierr = MatGetArray(B,&b);CHKERRQ(ierr); 13902bebee5dSHong Zhang ierr = MatGetArray(X,&x);CHKERRQ(ierr); 13912bebee5dSHong Zhang 13922bebee5dSHong Zhang tmp = a->solve_work; 13932bebee5dSHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 13942bebee5dSHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 13952bebee5dSHong Zhang 1396a36b22ccSSatish Balay for (neq=0; neq<B->cmap->n; neq++){ 13972bebee5dSHong Zhang /* forward solve the lower triangular */ 13982bebee5dSHong Zhang tmp[0] = b[r[0]]; 13992bebee5dSHong Zhang tmps = tmp; 14002bebee5dSHong Zhang for (i=1; i<n; i++) { 14012bebee5dSHong Zhang v = aa + ai[i] ; 14022bebee5dSHong Zhang vi = aj + ai[i] ; 14032bebee5dSHong Zhang nz = a->diag[i] - ai[i]; 14042bebee5dSHong Zhang sum = b[r[i]]; 1405003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 14062bebee5dSHong Zhang tmp[i] = sum; 14072bebee5dSHong Zhang } 14082bebee5dSHong Zhang /* backward solve the upper triangular */ 14092bebee5dSHong Zhang for (i=n-1; i>=0; i--){ 14102bebee5dSHong Zhang v = aa + a->diag[i] + 1; 14112bebee5dSHong Zhang vi = aj + a->diag[i] + 1; 14122bebee5dSHong Zhang nz = ai[i+1] - a->diag[i] - 1; 14132bebee5dSHong Zhang sum = tmp[i]; 1414003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 14152bebee5dSHong Zhang x[c[i]] = tmp[i] = sum*aa[a->diag[i]]; 14162bebee5dSHong Zhang } 14172bebee5dSHong Zhang 14182bebee5dSHong Zhang b += n; 14192bebee5dSHong Zhang x += n; 14202bebee5dSHong Zhang } 14212bebee5dSHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 14222bebee5dSHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 14232bebee5dSHong Zhang ierr = MatRestoreArray(B,&b);CHKERRQ(ierr); 14242bebee5dSHong Zhang ierr = MatRestoreArray(X,&x);CHKERRQ(ierr); 1425dc0b31edSSatish Balay ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr); 14262bebee5dSHong Zhang PetscFunctionReturn(0); 14272bebee5dSHong Zhang } 14282bebee5dSHong Zhang 1429137fb511SHong Zhang #undef __FUNCT__ 1430137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm" 1431137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx) 1432137fb511SHong Zhang { 1433137fb511SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1434137fb511SHong Zhang IS iscol = a->col,isrow = a->row; 1435137fb511SHong Zhang PetscErrorCode ierr; 14365d0c19d7SBarry Smith const PetscInt *r,*c,*rout,*cout; 14375d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 14385d0c19d7SBarry Smith PetscInt nz,row; 1439dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1440dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1441137fb511SHong Zhang 1442137fb511SHong Zhang PetscFunctionBegin; 1443137fb511SHong Zhang if (!n) PetscFunctionReturn(0); 1444137fb511SHong Zhang 1445137fb511SHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 1446137fb511SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1447137fb511SHong Zhang tmp = a->solve_work; 1448137fb511SHong Zhang 1449137fb511SHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1450137fb511SHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1451137fb511SHong Zhang 1452137fb511SHong Zhang /* forward solve the lower triangular */ 1453137fb511SHong Zhang tmp[0] = b[*r++]; 1454137fb511SHong Zhang tmps = tmp; 1455137fb511SHong Zhang for (row=1; row<n; row++) { 1456137fb511SHong Zhang i = rout[row]; /* permuted row */ 1457137fb511SHong Zhang v = aa + ai[i] ; 1458137fb511SHong Zhang vi = aj + ai[i] ; 1459137fb511SHong Zhang nz = a->diag[i] - ai[i]; 1460137fb511SHong Zhang sum = b[*r++]; 1461003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1462137fb511SHong Zhang tmp[row] = sum; 1463137fb511SHong Zhang } 1464137fb511SHong Zhang 1465137fb511SHong Zhang /* backward solve the upper triangular */ 1466137fb511SHong Zhang for (row=n-1; row>=0; row--){ 1467137fb511SHong Zhang i = rout[row]; /* permuted row */ 1468137fb511SHong Zhang v = aa + a->diag[i] + 1; 1469137fb511SHong Zhang vi = aj + a->diag[i] + 1; 1470137fb511SHong Zhang nz = ai[i+1] - a->diag[i] - 1; 1471137fb511SHong Zhang sum = tmp[row]; 1472003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1473137fb511SHong Zhang x[*c--] = tmp[row] = sum*aa[a->diag[i]]; 1474137fb511SHong Zhang } 1475137fb511SHong Zhang 1476137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1477137fb511SHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1478137fb511SHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 1479137fb511SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1480dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 1481137fb511SHong Zhang PetscFunctionReturn(0); 1482137fb511SHong Zhang } 1483137fb511SHong Zhang 1484930ae53cSBarry Smith /* ----------------------------------------------------------- */ 1485a4005a5dSBarry Smith #include "../src/mat/impls/aij/seq/ftn-kernels/fsolve.h" 14864a2ae208SSatish Balay #undef __FUNCT__ 14874a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering" 1488dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx) 1489930ae53cSBarry Smith { 1490930ae53cSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 14916849ba73SBarry Smith PetscErrorCode ierr; 1492d0f46423SBarry Smith PetscInt n = A->rmap->n; 1493b377110cSBarry Smith const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag; 1494356650c2SBarry Smith PetscScalar *x; 1495356650c2SBarry Smith const PetscScalar *b; 1496dd6ea824SBarry Smith const MatScalar *aa = a->a; 1497aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 1498356650c2SBarry Smith PetscInt adiag_i,i,nz,ai_i; 1499b377110cSBarry Smith const PetscInt *vi; 1500dd6ea824SBarry Smith const MatScalar *v; 1501dd6ea824SBarry Smith PetscScalar sum; 1502d85afc42SSatish Balay #endif 1503930ae53cSBarry Smith 15043a40ed3dSBarry Smith PetscFunctionBegin; 15053a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 1506930ae53cSBarry Smith 1507356650c2SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 15081ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1509930ae53cSBarry Smith 1510aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 15111c4feecaSSatish Balay fortransolveaij_(&n,x,ai,aj,adiag,aa,b); 15121c4feecaSSatish Balay #else 1513930ae53cSBarry Smith /* forward solve the lower triangular */ 1514930ae53cSBarry Smith x[0] = b[0]; 1515930ae53cSBarry Smith for (i=1; i<n; i++) { 1516930ae53cSBarry Smith ai_i = ai[i]; 1517930ae53cSBarry Smith v = aa + ai_i; 1518930ae53cSBarry Smith vi = aj + ai_i; 1519930ae53cSBarry Smith nz = adiag[i] - ai_i; 1520930ae53cSBarry Smith sum = b[i]; 1521003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1522930ae53cSBarry Smith x[i] = sum; 1523930ae53cSBarry Smith } 1524930ae53cSBarry Smith 1525930ae53cSBarry Smith /* backward solve the upper triangular */ 1526930ae53cSBarry Smith for (i=n-1; i>=0; i--){ 1527930ae53cSBarry Smith adiag_i = adiag[i]; 1528d708749eSBarry Smith v = aa + adiag_i + 1; 1529d708749eSBarry Smith vi = aj + adiag_i + 1; 1530930ae53cSBarry Smith nz = ai[i+1] - adiag_i - 1; 1531930ae53cSBarry Smith sum = x[i]; 1532003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1533930ae53cSBarry Smith x[i] = sum*aa[adiag_i]; 1534930ae53cSBarry Smith } 15351c4feecaSSatish Balay #endif 1536dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 1537356650c2SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 15381ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 15393a40ed3dSBarry Smith PetscFunctionReturn(0); 1540930ae53cSBarry Smith } 1541930ae53cSBarry Smith 15424a2ae208SSatish Balay #undef __FUNCT__ 15434a2ae208SSatish Balay #define __FUNCT__ "MatSolveAdd_SeqAIJ" 1544dfbe8321SBarry Smith PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx) 1545da3a660dSBarry Smith { 1546416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1547416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 15486849ba73SBarry Smith PetscErrorCode ierr; 154904fbf559SBarry Smith PetscInt i, n = A->rmap->n,j; 15505d0c19d7SBarry Smith PetscInt nz; 155104fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j; 155204fbf559SBarry Smith PetscScalar *x,*tmp,sum; 155304fbf559SBarry Smith const PetscScalar *b; 1554dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1555da3a660dSBarry Smith 15563a40ed3dSBarry Smith PetscFunctionBegin; 155778b31e54SBarry Smith if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 1558da3a660dSBarry Smith 155904fbf559SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 15601ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1561416022c9SBarry Smith tmp = a->solve_work; 1562da3a660dSBarry Smith 15633b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 15643b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1565da3a660dSBarry Smith 1566da3a660dSBarry Smith /* forward solve the lower triangular */ 1567da3a660dSBarry Smith tmp[0] = b[*r++]; 1568da3a660dSBarry Smith for (i=1; i<n; i++) { 1569010a20caSHong Zhang v = aa + ai[i] ; 1570010a20caSHong Zhang vi = aj + ai[i] ; 1571416022c9SBarry Smith nz = a->diag[i] - ai[i]; 1572da3a660dSBarry Smith sum = b[*r++]; 157304fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1574da3a660dSBarry Smith tmp[i] = sum; 1575da3a660dSBarry Smith } 1576da3a660dSBarry Smith 1577da3a660dSBarry Smith /* backward solve the upper triangular */ 1578da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1579010a20caSHong Zhang v = aa + a->diag[i] + 1; 1580010a20caSHong Zhang vi = aj + a->diag[i] + 1; 1581416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 1582da3a660dSBarry Smith sum = tmp[i]; 158304fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1584010a20caSHong Zhang tmp[i] = sum*aa[a->diag[i]]; 1585da3a660dSBarry Smith x[*c--] += tmp[i]; 1586da3a660dSBarry Smith } 1587da3a660dSBarry Smith 15883b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 15893b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 159004fbf559SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 15911ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1592dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 1593898183ecSLois Curfman McInnes 15943a40ed3dSBarry Smith PetscFunctionReturn(0); 1595da3a660dSBarry Smith } 159604fbf559SBarry Smith 15974a2ae208SSatish Balay #undef __FUNCT__ 15984a2ae208SSatish Balay #define __FUNCT__ "MatSolveTranspose_SeqAIJ" 1599dfbe8321SBarry Smith PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx) 1600da3a660dSBarry Smith { 1601416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 16022235e667SBarry Smith IS iscol = a->col,isrow = a->row; 16036849ba73SBarry Smith PetscErrorCode ierr; 160404fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 160504fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 160604fbf559SBarry Smith PetscInt nz; 160704fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1608dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 160904fbf559SBarry Smith const PetscScalar *b; 1610da3a660dSBarry Smith 16113a40ed3dSBarry Smith PetscFunctionBegin; 161204fbf559SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 16131ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1614416022c9SBarry Smith tmp = a->solve_work; 1615da3a660dSBarry Smith 16162235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 16172235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1618da3a660dSBarry Smith 1619da3a660dSBarry Smith /* copy the b into temp work space according to permutation */ 16202235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1621da3a660dSBarry Smith 1622da3a660dSBarry Smith /* forward solve the U^T */ 1623da3a660dSBarry Smith for (i=0; i<n; i++) { 1624010a20caSHong Zhang v = aa + diag[i] ; 1625010a20caSHong Zhang vi = aj + diag[i] + 1; 1626f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 1627c38d4ed2SBarry Smith s1 = tmp[i]; 1628ef66eb69SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 162904fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1630c38d4ed2SBarry Smith tmp[i] = s1; 1631da3a660dSBarry Smith } 1632da3a660dSBarry Smith 1633da3a660dSBarry Smith /* backward solve the L^T */ 1634da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1635010a20caSHong Zhang v = aa + diag[i] - 1 ; 1636010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1637f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 1638f1af5d2fSBarry Smith s1 = tmp[i]; 163904fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 1640da3a660dSBarry Smith } 1641da3a660dSBarry Smith 1642da3a660dSBarry Smith /* copy tmp into x according to permutation */ 1643591294e4SBarry Smith for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1644da3a660dSBarry Smith 16452235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 16462235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 164704fbf559SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 16481ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1649da3a660dSBarry Smith 1650dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 16513a40ed3dSBarry Smith PetscFunctionReturn(0); 1652da3a660dSBarry Smith } 1653da3a660dSBarry Smith 16544a2ae208SSatish Balay #undef __FUNCT__ 16554a2ae208SSatish Balay #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ" 1656dfbe8321SBarry Smith PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx) 1657da3a660dSBarry Smith { 1658416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 16592235e667SBarry Smith IS iscol = a->col,isrow = a->row; 16606849ba73SBarry Smith PetscErrorCode ierr; 166104fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 166204fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 166304fbf559SBarry Smith PetscInt nz; 166404fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1665dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 166604fbf559SBarry Smith const PetscScalar *b; 16676abc6512SBarry Smith 16683a40ed3dSBarry Smith PetscFunctionBegin; 166923bc6035SMatthew Knepley if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 167004fbf559SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 16711ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1672416022c9SBarry Smith tmp = a->solve_work; 16736abc6512SBarry Smith 16742235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 16752235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 16766abc6512SBarry Smith 16776abc6512SBarry Smith /* copy the b into temp work space according to permutation */ 16782235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 16796abc6512SBarry Smith 16806abc6512SBarry Smith /* forward solve the U^T */ 16816abc6512SBarry Smith for (i=0; i<n; i++) { 1682010a20caSHong Zhang v = aa + diag[i] ; 1683010a20caSHong Zhang vi = aj + diag[i] + 1; 1684f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 168504fbf559SBarry Smith s1 = tmp[i]; 168604fbf559SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 168704fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 168804fbf559SBarry Smith tmp[i] = s1; 16896abc6512SBarry Smith } 16906abc6512SBarry Smith 16916abc6512SBarry Smith /* backward solve the L^T */ 16926abc6512SBarry Smith for (i=n-1; i>=0; i--){ 1693010a20caSHong Zhang v = aa + diag[i] - 1 ; 1694010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1695f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 169604fbf559SBarry Smith s1 = tmp[i]; 169704fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 16986abc6512SBarry Smith } 16996abc6512SBarry Smith 17006abc6512SBarry Smith /* copy tmp into x according to permutation */ 17016abc6512SBarry Smith for (i=0; i<n; i++) x[r[i]] += tmp[i]; 17026abc6512SBarry Smith 17032235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 17042235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 170504fbf559SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 17061ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 17076abc6512SBarry Smith 170804fbf559SBarry Smith ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 17093a40ed3dSBarry Smith PetscFunctionReturn(0); 1710da3a660dSBarry Smith } 171104fbf559SBarry Smith 17129e25ed09SBarry Smith /* ----------------------------------------------------------------*/ 17133c5fc038SBarry Smith EXTERN PetscErrorCode Mat_CheckInode(Mat,PetscTruth); 1714f77e22a1SHong Zhang EXTERN PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,Mat,MatDuplicateOption,PetscTruth); 171508480c60SBarry Smith 17168fc3a347SHong Zhang /* 17178fc3a347SHong Zhang ilu(0) with natural ordering under new data structure. 17188fc3a347SHong Zhang Factored arrays bj and ba are stored as 17198fc3a347SHong Zhang L(0,:), L(1,:), ...,L(n-1,:), U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:) 17208fc3a347SHong Zhang 17218fc3a347SHong Zhang bi=fact->i is an array of size 2n+2, in which 17228fc3a347SHong Zhang bi+ 17238fc3a347SHong Zhang bi[i] -> 1st entry of L(i,:),i=0,...,i-1 1724f77e22a1SHong Zhang bi[n] -> points to L(n-1,:)+1 17258fc3a347SHong Zhang bi[n+1] -> 1st entry of U(n-1,:) 17268fc3a347SHong Zhang bi[2n-i] -> 1st entry of U(i,:) 17278fc3a347SHong Zhang bi[2n-i+1] -> end of U(i,:)+1, the 1st entry of U(i-1,:) 1728f77e22a1SHong Zhang bi[2n] -> 1st entry of U(0,:) 1729f77e22a1SHong Zhang bi[2n+1] -> points to U(0,:)+1 17308fc3a347SHong Zhang 17318fc3a347SHong Zhang U(i,:) contains diag[i] as its last entry, i.e., 17328fc3a347SHong Zhang U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 17338fc3a347SHong Zhang */ 1734c9c72f8fSHong Zhang #undef __FUNCT__ 17358fc3a347SHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct" 17368fc3a347SHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1737c9c72f8fSHong Zhang { 1738c9c72f8fSHong Zhang 1739c9c72f8fSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1740c9c72f8fSHong Zhang PetscErrorCode ierr; 1741c9c72f8fSHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag; 17428fc3a347SHong Zhang PetscInt i,j,nz,*bi,*bj,*bdiag; 1743c9c72f8fSHong Zhang 1744c9c72f8fSHong Zhang PetscFunctionBegin; 1745f77e22a1SHong Zhang ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr); 1746c9c72f8fSHong Zhang b = (Mat_SeqAIJ*)(fact)->data; 1747c9c72f8fSHong Zhang 1748f77e22a1SHong Zhang /* allocate matrix arrays for new data structure */ 1749f77e22a1SHong Zhang ierr = PetscMalloc3(ai[n]+1,PetscScalar,&b->a,ai[n]+1,PetscInt,&b->j,2*n+2,PetscInt,&b->i);CHKERRQ(ierr); 1750f77e22a1SHong Zhang ierr = PetscLogObjectMemory(fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(2*n+2)*sizeof(PetscInt));CHKERRQ(ierr); 1751f77e22a1SHong Zhang b->singlemalloc = PETSC_TRUE; 1752f77e22a1SHong Zhang if (!b->diag){ 17538fc3a347SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&b->diag);CHKERRQ(ierr); 1754f77e22a1SHong Zhang } 1755c9c72f8fSHong Zhang bdiag = b->diag; 1756c9c72f8fSHong Zhang 1757c9c72f8fSHong Zhang if (n > 0) { 175816a2bf60SHong Zhang ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr); 1759c9c72f8fSHong Zhang } 1760c9c72f8fSHong Zhang 1761c9c72f8fSHong Zhang /* set bi and bj with new data structure */ 1762c9c72f8fSHong Zhang bi = b->i; 1763c9c72f8fSHong Zhang bj = b->j; 1764c9c72f8fSHong Zhang 1765c9c72f8fSHong Zhang /* L part */ 1766c9c72f8fSHong Zhang bi[0] = 0; 1767c9c72f8fSHong Zhang for (i=0; i<n; i++){ 1768c9c72f8fSHong Zhang nz = adiag[i] - ai[i]; 1769c9c72f8fSHong Zhang bi[i+1] = bi[i] + nz; 1770c9c72f8fSHong Zhang aj = a->j + ai[i]; 1771c9c72f8fSHong Zhang for (j=0; j<nz; j++){ 1772c9c72f8fSHong Zhang *bj = aj[j]; bj++; 1773c9c72f8fSHong Zhang } 1774c9c72f8fSHong Zhang } 1775c9c72f8fSHong Zhang 1776c9c72f8fSHong Zhang /* U part */ 17778fc3a347SHong Zhang bi[n+1] = bi[n]; 1778c9c72f8fSHong Zhang for (i=n-1; i>=0; i--){ 1779c9c72f8fSHong Zhang nz = ai[i+1] - adiag[i] - 1; 17808fc3a347SHong Zhang bi[2*n-i+1] = bi[2*n-i] + nz + 1; 1781c9c72f8fSHong Zhang aj = a->j + adiag[i] + 1; 1782c9c72f8fSHong Zhang for (j=0; j<nz; j++){ 1783c9c72f8fSHong Zhang *bj = aj[j]; bj++; 1784c9c72f8fSHong Zhang } 1785c9c72f8fSHong Zhang /* diag[i] */ 1786c9c72f8fSHong Zhang *bj = i; bj++; 17878fc3a347SHong Zhang bdiag[i] = bi[2*n-i+1]-1; 1788c9c72f8fSHong Zhang } 1789c9c72f8fSHong Zhang PetscFunctionReturn(0); 1790c9c72f8fSHong Zhang } 1791c9c72f8fSHong Zhang 17924a2ae208SSatish Balay #undef __FUNCT__ 17936516239fSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_newdatastruct" 17946516239fSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_newdatastruct(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 17959e25ed09SBarry Smith { 1796416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 17979e25ed09SBarry Smith IS isicol; 17986849ba73SBarry Smith PetscErrorCode ierr; 17995d0c19d7SBarry Smith const PetscInt *r,*ic; 18005d0c19d7SBarry Smith PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,d; 18015a8e39fbSHong Zhang PetscInt *bi,*cols,nnz,*cols_lvl; 1802052f0c41SBarry Smith PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 18035a8e39fbSHong Zhang PetscInt i,levels,diagonal_fill; 180477c4ece6SBarry Smith PetscTruth col_identity,row_identity; 1805329f5518SBarry Smith PetscReal f; 18065a8e39fbSHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 18075a8e39fbSHong Zhang PetscBT lnkbt; 18085a8e39fbSHong Zhang PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 1809a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1810a1a86e44SBarry Smith PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 181109f38230SBarry Smith PetscTruth missing; 18129e25ed09SBarry Smith 18133a40ed3dSBarry Smith PetscFunctionBegin; 1814d0f46423SBarry Smith if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 18156cf997b0SBarry Smith f = info->fill; 181638baddfdSBarry Smith levels = (PetscInt)info->levels; 181738baddfdSBarry Smith diagonal_fill = (PetscInt)info->diagonal_fill; 18184c49b128SBarry Smith ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 181982bf6240SBarry Smith 1820be0abb6dSBarry Smith ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 1821be0abb6dSBarry Smith ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 1822c9c72f8fSHong Zhang 18236516239fSHong Zhang if (!levels && row_identity && col_identity) { 18246516239fSHong Zhang /* special case: ilu(0) with natural ordering */ 18258fc3a347SHong Zhang ierr = MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct(fact,A,isrow,iscol,info);CHKERRQ(ierr); 1826c9c72f8fSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct; 18276516239fSHong Zhang 18286516239fSHong Zhang fact->factor = MAT_FACTOR_ILU; 18296516239fSHong Zhang (fact)->info.factor_mallocs = 0; 18306516239fSHong Zhang (fact)->info.fill_ratio_given = info->fill; 18316516239fSHong Zhang (fact)->info.fill_ratio_needed = 1.0; 18326516239fSHong Zhang b = (Mat_SeqAIJ*)(fact)->data; 18336516239fSHong Zhang ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 18346516239fSHong Zhang if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 18356516239fSHong Zhang b->row = isrow; 18366516239fSHong Zhang b->col = iscol; 18376516239fSHong Zhang b->icol = isicol; 18386516239fSHong Zhang ierr = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 18396516239fSHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 18406516239fSHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 18414108e4d5SBarry Smith /* ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); */ 18426516239fSHong Zhang PetscFunctionReturn(0); 18436516239fSHong Zhang } 18446516239fSHong Zhang 18456516239fSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 18466516239fSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 18476516239fSHong Zhang 18486516239fSHong Zhang /* get new row pointers */ 18496516239fSHong Zhang ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 18506516239fSHong Zhang bi[0] = 0; 18516516239fSHong Zhang /* bdiag is location of diagonal in factor */ 18526516239fSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 18536516239fSHong Zhang bdiag[0] = 0; 18546516239fSHong Zhang 18556516239fSHong Zhang ierr = PetscMalloc((2*n+1)*sizeof(PetscInt**),&bj_ptr);CHKERRQ(ierr); 18566516239fSHong Zhang bjlvl_ptr = (PetscInt**)(bj_ptr + n); 18576516239fSHong Zhang 18586516239fSHong Zhang /* create a linked list for storing column indices of the active row */ 18596516239fSHong Zhang nlnk = n + 1; 18606516239fSHong Zhang ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 18616516239fSHong Zhang 18626516239fSHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 18636516239fSHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 18646516239fSHong Zhang current_space = free_space; 18656516239fSHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 18666516239fSHong Zhang current_space_lvl = free_space_lvl; 18676516239fSHong Zhang 18686516239fSHong Zhang for (i=0; i<n; i++) { 18696516239fSHong Zhang nzi = 0; 18706516239fSHong Zhang /* copy current row into linked list */ 18716516239fSHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 18726516239fSHong Zhang if (!nnz) SETERRQ2(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 18736516239fSHong Zhang cols = aj + ai[r[i]]; 18746516239fSHong Zhang lnk[i] = -1; /* marker to indicate if diagonal exists */ 18756516239fSHong Zhang ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 18766516239fSHong Zhang nzi += nlnk; 18776516239fSHong Zhang 18786516239fSHong Zhang /* make sure diagonal entry is included */ 18796516239fSHong Zhang if (diagonal_fill && lnk[i] == -1) { 18806516239fSHong Zhang fm = n; 18816516239fSHong Zhang while (lnk[fm] < i) fm = lnk[fm]; 18826516239fSHong Zhang lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 18836516239fSHong Zhang lnk[fm] = i; 18846516239fSHong Zhang lnk_lvl[i] = 0; 18856516239fSHong Zhang nzi++; dcount++; 18866516239fSHong Zhang } 18876516239fSHong Zhang 18886516239fSHong Zhang /* add pivot rows into the active row */ 18896516239fSHong Zhang nzbd = 0; 18906516239fSHong Zhang prow = lnk[n]; 18916516239fSHong Zhang while (prow < i) { 18926516239fSHong Zhang nnz = bdiag[prow]; 18936516239fSHong Zhang cols = bj_ptr[prow] + nnz + 1; 18946516239fSHong Zhang cols_lvl = bjlvl_ptr[prow] + nnz + 1; 18956516239fSHong Zhang nnz = bi[prow+1] - bi[prow] - nnz - 1; 18966516239fSHong Zhang ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 18976516239fSHong Zhang nzi += nlnk; 18986516239fSHong Zhang prow = lnk[prow]; 18996516239fSHong Zhang nzbd++; 19006516239fSHong Zhang } 19016516239fSHong Zhang bdiag[i] = nzbd; 19026516239fSHong Zhang bi[i+1] = bi[i] + nzi; 19036516239fSHong Zhang 19046516239fSHong Zhang /* if free space is not available, make more free space */ 19056516239fSHong Zhang if (current_space->local_remaining<nzi) { 19066516239fSHong Zhang nnz = 2*nzi*(n - i); /* estimated and max additional space needed */ 19076516239fSHong Zhang ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 19086516239fSHong Zhang ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 19096516239fSHong Zhang reallocs++; 19106516239fSHong Zhang } 19116516239fSHong Zhang 19126516239fSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 19136516239fSHong Zhang ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 19146516239fSHong Zhang bj_ptr[i] = current_space->array; 19156516239fSHong Zhang bjlvl_ptr[i] = current_space_lvl->array; 19166516239fSHong Zhang 19176516239fSHong Zhang /* make sure the active row i has diagonal entry */ 19186516239fSHong Zhang if (*(bj_ptr[i]+bdiag[i]) != i) { 19196516239fSHong Zhang SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\ 19206516239fSHong Zhang try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i); 19216516239fSHong Zhang } 19226516239fSHong Zhang 19236516239fSHong Zhang current_space->array += nzi; 19246516239fSHong Zhang current_space->local_used += nzi; 19256516239fSHong Zhang current_space->local_remaining -= nzi; 19266516239fSHong Zhang current_space_lvl->array += nzi; 19276516239fSHong Zhang current_space_lvl->local_used += nzi; 19286516239fSHong Zhang current_space_lvl->local_remaining -= nzi; 19296516239fSHong Zhang } 19306516239fSHong Zhang 19316516239fSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 19326516239fSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 19336516239fSHong Zhang 19346516239fSHong Zhang /* destroy list of free space and other temporary arrays */ 19356516239fSHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 19366516239fSHong Zhang 19376516239fSHong Zhang /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */ 1938783ef271SHong Zhang ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 19396516239fSHong Zhang 19406516239fSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 19416516239fSHong Zhang ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 19426516239fSHong Zhang ierr = PetscFree(bj_ptr);CHKERRQ(ierr); 19436516239fSHong Zhang 19446516239fSHong Zhang #if defined(PETSC_USE_INFO) 19456516239fSHong Zhang { 19466516239fSHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 19476516239fSHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 19486516239fSHong Zhang ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 19496516239fSHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 19506516239fSHong Zhang ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 19516516239fSHong Zhang if (diagonal_fill) { 19526516239fSHong Zhang ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 19536516239fSHong Zhang } 19546516239fSHong Zhang } 19556516239fSHong Zhang #endif 19566516239fSHong Zhang 19576516239fSHong Zhang /* put together the new matrix */ 19586516239fSHong Zhang ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 19596516239fSHong Zhang ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 19606516239fSHong Zhang b = (Mat_SeqAIJ*)(fact)->data; 19616516239fSHong Zhang b->free_a = PETSC_TRUE; 19626516239fSHong Zhang b->free_ij = PETSC_TRUE; 19636516239fSHong Zhang b->singlemalloc = PETSC_FALSE; 19646516239fSHong Zhang ierr = PetscMalloc( (bi[2*n+1] )*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 19656516239fSHong Zhang b->j = bj; 19666516239fSHong Zhang b->i = bi; 19676516239fSHong Zhang b->diag = bdiag; 19686516239fSHong Zhang b->ilen = 0; 19696516239fSHong Zhang b->imax = 0; 19706516239fSHong Zhang b->row = isrow; 19716516239fSHong Zhang b->col = iscol; 19726516239fSHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 19736516239fSHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 19746516239fSHong Zhang b->icol = isicol; 19756516239fSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 19766516239fSHong Zhang /* In b structure: Free imax, ilen, old a, old j. 19776516239fSHong Zhang Allocate bdiag, solve_work, new a, new j */ 19786516239fSHong Zhang ierr = PetscLogObjectMemory(fact,bi[2*n+1] * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 19796516239fSHong Zhang b->maxnz = b->nz = bi[2*n+1] ; 19806516239fSHong Zhang (fact)->info.factor_mallocs = reallocs; 19816516239fSHong Zhang (fact)->info.fill_ratio_given = f; 19826516239fSHong Zhang (fact)->info.fill_ratio_needed = ((PetscReal)bi[2*n+1])/((PetscReal)ai[n]); 19836516239fSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct; 19844108e4d5SBarry Smith /* ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); */ 19856516239fSHong Zhang PetscFunctionReturn(0); 19866516239fSHong Zhang } 19876516239fSHong Zhang 1988813a1f2bSShri Abhyankar 1989813a1f2bSShri Abhyankar /* 1990*8a76ed4fSHong Zhang ilu() under revised new data structure. 1991813a1f2bSShri Abhyankar Factored arrays bj and ba are stored as 1992813a1f2bSShri Abhyankar L(0,:), L(1,:), ...,L(n-1,:), U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:) 1993813a1f2bSShri Abhyankar 1994813a1f2bSShri Abhyankar bi=fact->i is an array of size n+1, in which 1995813a1f2bSShri Abhyankar bi+ 1996baabb860SHong Zhang bi[i]: points to 1st entry of L(i,:),i=0,...,n-1 1997baabb860SHong Zhang bi[n]: points to L(n-1,n-1)+1 1998baabb860SHong Zhang 1999813a1f2bSShri Abhyankar bdiag=fact->diag is an array of size n+1,in which 2000baabb860SHong Zhang bdiag[i]: points to diagonal of U(i,:), i=0,...,n-1 2001baabb860SHong Zhang bdiag[n]: points to diagonal of U(n-1,0)-1 2002813a1f2bSShri Abhyankar 2003813a1f2bSShri Abhyankar U(i,:) contains bdiag[i] as its last entry, i.e., 2004813a1f2bSShri Abhyankar U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 2005813a1f2bSShri Abhyankar */ 2006813a1f2bSShri Abhyankar #undef __FUNCT__ 2007813a1f2bSShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct_v2" 2008813a1f2bSShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct_v2(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 2009813a1f2bSShri Abhyankar { 2010813a1f2bSShri Abhyankar 2011813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 2012813a1f2bSShri Abhyankar PetscErrorCode ierr; 2013813a1f2bSShri Abhyankar PetscInt n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag; 2014813a1f2bSShri Abhyankar PetscInt i,j,nz,*bi,*bj,*bdiag,bi_temp; 2015813a1f2bSShri Abhyankar 2016813a1f2bSShri Abhyankar PetscFunctionBegin; 2017813a1f2bSShri Abhyankar ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr); 2018813a1f2bSShri Abhyankar b = (Mat_SeqAIJ*)(fact)->data; 2019813a1f2bSShri Abhyankar 2020813a1f2bSShri Abhyankar /* allocate matrix arrays for new data structure */ 2021813a1f2bSShri Abhyankar ierr = PetscMalloc3(ai[n]+1,PetscScalar,&b->a,ai[n]+1,PetscInt,&b->j,n+1,PetscInt,&b->i);CHKERRQ(ierr); 2022813a1f2bSShri Abhyankar ierr = PetscLogObjectMemory(fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(n+1)*sizeof(PetscInt));CHKERRQ(ierr); 2023813a1f2bSShri Abhyankar b->singlemalloc = PETSC_TRUE; 2024813a1f2bSShri Abhyankar if (!b->diag){ 2025813a1f2bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscInt),&b->diag);CHKERRQ(ierr); 2026813a1f2bSShri Abhyankar ierr = PetscLogObjectMemory(fact,(n+1)*sizeof(PetscInt));CHKERRQ(ierr); 2027813a1f2bSShri Abhyankar } 2028813a1f2bSShri Abhyankar bdiag = b->diag; 2029813a1f2bSShri Abhyankar 2030813a1f2bSShri Abhyankar if (n > 0) { 2031813a1f2bSShri Abhyankar ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr); 2032813a1f2bSShri Abhyankar } 2033813a1f2bSShri Abhyankar 2034813a1f2bSShri Abhyankar /* set bi and bj with new data structure */ 2035813a1f2bSShri Abhyankar bi = b->i; 2036813a1f2bSShri Abhyankar bj = b->j; 2037813a1f2bSShri Abhyankar 2038813a1f2bSShri Abhyankar /* L part */ 2039813a1f2bSShri Abhyankar bi[0] = 0; 2040813a1f2bSShri Abhyankar for (i=0; i<n; i++){ 2041813a1f2bSShri Abhyankar nz = adiag[i] - ai[i]; 2042813a1f2bSShri Abhyankar bi[i+1] = bi[i] + nz; 2043813a1f2bSShri Abhyankar aj = a->j + ai[i]; 2044813a1f2bSShri Abhyankar for (j=0; j<nz; j++){ 2045813a1f2bSShri Abhyankar *bj = aj[j]; bj++; 2046813a1f2bSShri Abhyankar } 2047813a1f2bSShri Abhyankar } 2048813a1f2bSShri Abhyankar 2049813a1f2bSShri Abhyankar /* U part */ 2050813a1f2bSShri Abhyankar /* bi[n+1] = bi[n]; */ 2051813a1f2bSShri Abhyankar bi_temp = bi[n]; 205262fbd6afSShri Abhyankar bdiag[n] = bi[n]-1; 2053813a1f2bSShri Abhyankar for (i=n-1; i>=0; i--){ 2054813a1f2bSShri Abhyankar nz = ai[i+1] - adiag[i] - 1; 2055813a1f2bSShri Abhyankar /* bi[2*n-i+1] = bi[2*n-i] + nz + 1; */ 2056813a1f2bSShri Abhyankar bi_temp = bi_temp + nz + 1; 2057813a1f2bSShri Abhyankar aj = a->j + adiag[i] + 1; 2058813a1f2bSShri Abhyankar for (j=0; j<nz; j++){ 2059813a1f2bSShri Abhyankar *bj = aj[j]; bj++; 2060813a1f2bSShri Abhyankar } 2061813a1f2bSShri Abhyankar /* diag[i] */ 2062813a1f2bSShri Abhyankar *bj = i; bj++; 2063813a1f2bSShri Abhyankar /* bdiag[i] = bi[2*n-i+1]-1; */ 2064813a1f2bSShri Abhyankar bdiag[i] = bi_temp - 1; 2065813a1f2bSShri Abhyankar } 2066813a1f2bSShri Abhyankar PetscFunctionReturn(0); 2067813a1f2bSShri Abhyankar } 2068813a1f2bSShri Abhyankar 2069813a1f2bSShri Abhyankar #undef __FUNCT__ 2070813a1f2bSShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_newdatastruct_v2" 2071813a1f2bSShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_newdatastruct_v2(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 2072813a1f2bSShri Abhyankar { 2073813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 2074813a1f2bSShri Abhyankar IS isicol; 2075813a1f2bSShri Abhyankar PetscErrorCode ierr; 2076813a1f2bSShri Abhyankar const PetscInt *r,*ic; 2077813a1f2bSShri Abhyankar PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,d; 2078813a1f2bSShri Abhyankar PetscInt *bi,*cols,nnz,*cols_lvl; 2079813a1f2bSShri Abhyankar PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 2080813a1f2bSShri Abhyankar PetscInt i,levels,diagonal_fill; 2081813a1f2bSShri Abhyankar PetscTruth col_identity,row_identity; 2082813a1f2bSShri Abhyankar PetscReal f; 2083813a1f2bSShri Abhyankar PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 2084813a1f2bSShri Abhyankar PetscBT lnkbt; 2085813a1f2bSShri Abhyankar PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 2086813a1f2bSShri Abhyankar PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 2087813a1f2bSShri Abhyankar PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 2088813a1f2bSShri Abhyankar PetscTruth missing; 2089813a1f2bSShri Abhyankar 2090813a1f2bSShri Abhyankar PetscFunctionBegin; 2091813a1f2bSShri Abhyankar if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 2092813a1f2bSShri Abhyankar f = info->fill; 2093813a1f2bSShri Abhyankar levels = (PetscInt)info->levels; 2094813a1f2bSShri Abhyankar diagonal_fill = (PetscInt)info->diagonal_fill; 2095813a1f2bSShri Abhyankar ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 2096813a1f2bSShri Abhyankar 2097813a1f2bSShri Abhyankar ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 2098813a1f2bSShri Abhyankar ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 2099813a1f2bSShri Abhyankar 2100813a1f2bSShri Abhyankar if (!levels && row_identity && col_identity) { 2101813a1f2bSShri Abhyankar /* special case: ilu(0) with natural ordering */ 2102813a1f2bSShri Abhyankar ierr = MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct_v2(fact,A,isrow,iscol,info);CHKERRQ(ierr); 2103813a1f2bSShri Abhyankar (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct_v2; 2104813a1f2bSShri Abhyankar 2105813a1f2bSShri Abhyankar fact->factor = MAT_FACTOR_ILU; 2106813a1f2bSShri Abhyankar (fact)->info.factor_mallocs = 0; 2107813a1f2bSShri Abhyankar (fact)->info.fill_ratio_given = info->fill; 2108813a1f2bSShri Abhyankar (fact)->info.fill_ratio_needed = 1.0; 2109813a1f2bSShri Abhyankar b = (Mat_SeqAIJ*)(fact)->data; 2110813a1f2bSShri Abhyankar ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 2111813a1f2bSShri Abhyankar if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 2112813a1f2bSShri Abhyankar b->row = isrow; 2113813a1f2bSShri Abhyankar b->col = iscol; 2114813a1f2bSShri Abhyankar b->icol = isicol; 2115813a1f2bSShri Abhyankar ierr = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2116813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 2117813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 21184108e4d5SBarry Smith /* ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); */ 2119813a1f2bSShri Abhyankar PetscFunctionReturn(0); 2120813a1f2bSShri Abhyankar } 2121813a1f2bSShri Abhyankar 2122813a1f2bSShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 2123813a1f2bSShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 2124813a1f2bSShri Abhyankar 2125813a1f2bSShri Abhyankar /* get new row pointers */ 2126f268cba8SShri Abhyankar /* ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); */ 2127f268cba8SShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 2128813a1f2bSShri Abhyankar bi[0] = 0; 2129813a1f2bSShri Abhyankar /* bdiag is location of diagonal in factor */ 2130813a1f2bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 2131813a1f2bSShri Abhyankar bdiag[0] = 0; 2132813a1f2bSShri Abhyankar 2133813a1f2bSShri Abhyankar ierr = PetscMalloc((2*n+1)*sizeof(PetscInt**),&bj_ptr);CHKERRQ(ierr); 2134813a1f2bSShri Abhyankar bjlvl_ptr = (PetscInt**)(bj_ptr + n); 2135813a1f2bSShri Abhyankar 2136813a1f2bSShri Abhyankar /* create a linked list for storing column indices of the active row */ 2137813a1f2bSShri Abhyankar nlnk = n + 1; 2138813a1f2bSShri Abhyankar ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 2139813a1f2bSShri Abhyankar 2140813a1f2bSShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 2141813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 2142813a1f2bSShri Abhyankar current_space = free_space; 2143813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 2144813a1f2bSShri Abhyankar current_space_lvl = free_space_lvl; 2145813a1f2bSShri Abhyankar 2146813a1f2bSShri Abhyankar for (i=0; i<n; i++) { 2147813a1f2bSShri Abhyankar nzi = 0; 2148813a1f2bSShri Abhyankar /* copy current row into linked list */ 2149813a1f2bSShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 2150813a1f2bSShri Abhyankar if (!nnz) SETERRQ2(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 2151813a1f2bSShri Abhyankar cols = aj + ai[r[i]]; 2152813a1f2bSShri Abhyankar lnk[i] = -1; /* marker to indicate if diagonal exists */ 2153813a1f2bSShri Abhyankar ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 2154813a1f2bSShri Abhyankar nzi += nlnk; 2155813a1f2bSShri Abhyankar 2156813a1f2bSShri Abhyankar /* make sure diagonal entry is included */ 2157813a1f2bSShri Abhyankar if (diagonal_fill && lnk[i] == -1) { 2158813a1f2bSShri Abhyankar fm = n; 2159813a1f2bSShri Abhyankar while (lnk[fm] < i) fm = lnk[fm]; 2160813a1f2bSShri Abhyankar lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 2161813a1f2bSShri Abhyankar lnk[fm] = i; 2162813a1f2bSShri Abhyankar lnk_lvl[i] = 0; 2163813a1f2bSShri Abhyankar nzi++; dcount++; 2164813a1f2bSShri Abhyankar } 2165813a1f2bSShri Abhyankar 2166813a1f2bSShri Abhyankar /* add pivot rows into the active row */ 2167813a1f2bSShri Abhyankar nzbd = 0; 2168813a1f2bSShri Abhyankar prow = lnk[n]; 2169813a1f2bSShri Abhyankar while (prow < i) { 2170813a1f2bSShri Abhyankar nnz = bdiag[prow]; 2171813a1f2bSShri Abhyankar cols = bj_ptr[prow] + nnz + 1; 2172813a1f2bSShri Abhyankar cols_lvl = bjlvl_ptr[prow] + nnz + 1; 2173813a1f2bSShri Abhyankar nnz = bi[prow+1] - bi[prow] - nnz - 1; 2174813a1f2bSShri Abhyankar ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 2175813a1f2bSShri Abhyankar nzi += nlnk; 2176813a1f2bSShri Abhyankar prow = lnk[prow]; 2177813a1f2bSShri Abhyankar nzbd++; 2178813a1f2bSShri Abhyankar } 2179813a1f2bSShri Abhyankar bdiag[i] = nzbd; 2180813a1f2bSShri Abhyankar bi[i+1] = bi[i] + nzi; 2181813a1f2bSShri Abhyankar 2182813a1f2bSShri Abhyankar /* if free space is not available, make more free space */ 2183813a1f2bSShri Abhyankar if (current_space->local_remaining<nzi) { 2184813a1f2bSShri Abhyankar nnz = 2*nzi*(n - i); /* estimated and max additional space needed */ 2185813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 2186813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 2187813a1f2bSShri Abhyankar reallocs++; 2188813a1f2bSShri Abhyankar } 2189813a1f2bSShri Abhyankar 2190813a1f2bSShri Abhyankar /* copy data into free_space and free_space_lvl, then initialize lnk */ 2191813a1f2bSShri Abhyankar ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 2192813a1f2bSShri Abhyankar bj_ptr[i] = current_space->array; 2193813a1f2bSShri Abhyankar bjlvl_ptr[i] = current_space_lvl->array; 2194813a1f2bSShri Abhyankar 2195813a1f2bSShri Abhyankar /* make sure the active row i has diagonal entry */ 2196813a1f2bSShri Abhyankar if (*(bj_ptr[i]+bdiag[i]) != i) { 2197813a1f2bSShri Abhyankar SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\ 2198813a1f2bSShri Abhyankar try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i); 2199813a1f2bSShri Abhyankar } 2200813a1f2bSShri Abhyankar 2201813a1f2bSShri Abhyankar current_space->array += nzi; 2202813a1f2bSShri Abhyankar current_space->local_used += nzi; 2203813a1f2bSShri Abhyankar current_space->local_remaining -= nzi; 2204813a1f2bSShri Abhyankar current_space_lvl->array += nzi; 2205813a1f2bSShri Abhyankar current_space_lvl->local_used += nzi; 2206813a1f2bSShri Abhyankar current_space_lvl->local_remaining -= nzi; 2207813a1f2bSShri Abhyankar } 2208813a1f2bSShri Abhyankar 2209813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 2210813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 2211813a1f2bSShri Abhyankar 2212813a1f2bSShri Abhyankar /* destroy list of free space and other temporary arrays */ 2213813a1f2bSShri Abhyankar ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 2214813a1f2bSShri Abhyankar 2215813a1f2bSShri Abhyankar /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */ 2216f268cba8SShri Abhyankar /* ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); */ 2217f268cba8SShri Abhyankar ierr = PetscFreeSpaceContiguous_LU_v2(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 2218813a1f2bSShri Abhyankar 2219813a1f2bSShri Abhyankar ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2220813a1f2bSShri Abhyankar ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 2221813a1f2bSShri Abhyankar ierr = PetscFree(bj_ptr);CHKERRQ(ierr); 2222813a1f2bSShri Abhyankar 2223813a1f2bSShri Abhyankar #if defined(PETSC_USE_INFO) 2224813a1f2bSShri Abhyankar { 2225813a1f2bSShri Abhyankar PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 2226813a1f2bSShri Abhyankar ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 2227813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2228813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 2229813a1f2bSShri Abhyankar ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 2230813a1f2bSShri Abhyankar if (diagonal_fill) { 2231813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 2232813a1f2bSShri Abhyankar } 2233813a1f2bSShri Abhyankar } 2234813a1f2bSShri Abhyankar #endif 2235813a1f2bSShri Abhyankar 2236813a1f2bSShri Abhyankar /* put together the new matrix */ 2237813a1f2bSShri Abhyankar ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 2238813a1f2bSShri Abhyankar ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 2239813a1f2bSShri Abhyankar b = (Mat_SeqAIJ*)(fact)->data; 2240813a1f2bSShri Abhyankar b->free_a = PETSC_TRUE; 2241813a1f2bSShri Abhyankar b->free_ij = PETSC_TRUE; 2242813a1f2bSShri Abhyankar b->singlemalloc = PETSC_FALSE; 2243f268cba8SShri Abhyankar /* ierr = PetscMalloc( (bi[2*n+1] )*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); */ 2244f268cba8SShri Abhyankar ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 2245813a1f2bSShri Abhyankar b->j = bj; 2246813a1f2bSShri Abhyankar b->i = bi; 2247813a1f2bSShri Abhyankar b->diag = bdiag; 2248813a1f2bSShri Abhyankar b->ilen = 0; 2249813a1f2bSShri Abhyankar b->imax = 0; 2250813a1f2bSShri Abhyankar b->row = isrow; 2251813a1f2bSShri Abhyankar b->col = iscol; 2252813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 2253813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 2254813a1f2bSShri Abhyankar b->icol = isicol; 2255813a1f2bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2256813a1f2bSShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. 2257813a1f2bSShri Abhyankar Allocate bdiag, solve_work, new a, new j */ 2258f268cba8SShri Abhyankar 2259f268cba8SShri Abhyankar /* ierr = PetscLogObjectMemory(fact,bi[2*n+1] * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 2260f268cba8SShri Abhyankar b->maxnz = b->nz = bi[2*n+1] ; */ 2261f268cba8SShri Abhyankar ierr = PetscLogObjectMemory(fact,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 2262f268cba8SShri Abhyankar b->maxnz = b->nz = bdiag[0]+1; 2263813a1f2bSShri Abhyankar (fact)->info.factor_mallocs = reallocs; 2264813a1f2bSShri Abhyankar (fact)->info.fill_ratio_given = f; 2265f268cba8SShri Abhyankar /* (fact)->info.fill_ratio_needed = ((PetscReal)bi[2*n+1])/((PetscReal)ai[n]); */ 2266f268cba8SShri Abhyankar (fact)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 2267f268cba8SShri Abhyankar /* (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct; */ 2268f268cba8SShri Abhyankar (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct_v2; 22694108e4d5SBarry Smith /* ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); */ 2270813a1f2bSShri Abhyankar PetscFunctionReturn(0); 2271813a1f2bSShri Abhyankar } 2272813a1f2bSShri Abhyankar 22736516239fSHong Zhang #undef __FUNCT__ 22746516239fSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ" 22756516239fSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 22766516239fSHong Zhang { 22776516239fSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 22786516239fSHong Zhang IS isicol; 22796516239fSHong Zhang PetscErrorCode ierr; 22806516239fSHong Zhang const PetscInt *r,*ic; 22816516239fSHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,d; 22826516239fSHong Zhang PetscInt *bi,*cols,nnz,*cols_lvl; 22836516239fSHong Zhang PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 22846516239fSHong Zhang PetscInt i,levels,diagonal_fill; 22856516239fSHong Zhang PetscTruth col_identity,row_identity; 22866516239fSHong Zhang PetscReal f; 22876516239fSHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 22886516239fSHong Zhang PetscBT lnkbt; 22896516239fSHong Zhang PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 22906516239fSHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 22916516239fSHong Zhang PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 22926516239fSHong Zhang PetscTruth missing; 2293813a1f2bSShri Abhyankar PetscTruth newdatastruct=PETSC_FALSE,newdatastruct_v2=PETSC_FALSE; 22946516239fSHong Zhang 22956516239fSHong Zhang PetscFunctionBegin; 22966516239fSHong Zhang ierr = PetscOptionsGetTruth(PETSC_NULL,"-ilu_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr); 22976516239fSHong Zhang if (newdatastruct){ 22986516239fSHong Zhang ierr = MatILUFactorSymbolic_SeqAIJ_newdatastruct(fact,A,isrow,iscol,info);CHKERRQ(ierr); 22996516239fSHong Zhang PetscFunctionReturn(0); 23006516239fSHong Zhang } 2301813a1f2bSShri Abhyankar ierr = PetscOptionsGetTruth(PETSC_NULL,"-ilu_new_v2",&newdatastruct_v2,PETSC_NULL);CHKERRQ(ierr); 2302813a1f2bSShri Abhyankar if(newdatastruct_v2){ 2303813a1f2bSShri Abhyankar ierr = MatILUFactorSymbolic_SeqAIJ_newdatastruct_v2(fact,A,isrow,iscol,info);CHKERRQ(ierr); 2304813a1f2bSShri Abhyankar PetscFunctionReturn(0); 2305813a1f2bSShri Abhyankar } 23066516239fSHong Zhang 23076516239fSHong Zhang if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 23086516239fSHong Zhang f = info->fill; 23096516239fSHong Zhang levels = (PetscInt)info->levels; 23106516239fSHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 23116516239fSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 23126516239fSHong Zhang 23136516239fSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 23146516239fSHong Zhang ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 23156516239fSHong Zhang if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */ 2316f77e22a1SHong Zhang ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr); 2317c9c72f8fSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 2318c9c72f8fSHong Zhang 2319719d5645SBarry Smith fact->factor = MAT_FACTOR_ILU; 2320719d5645SBarry Smith (fact)->info.factor_mallocs = 0; 2321719d5645SBarry Smith (fact)->info.fill_ratio_given = info->fill; 2322719d5645SBarry Smith (fact)->info.fill_ratio_needed = 1.0; 2323719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 23248ef3462fSBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 232509f38230SBarry Smith if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 232608480c60SBarry Smith b->row = isrow; 232708480c60SBarry Smith b->col = iscol; 232882bf6240SBarry Smith b->icol = isicol; 2329719d5645SBarry Smith ierr = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2330c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 2331c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 23324108e4d5SBarry Smith ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); 23333a40ed3dSBarry Smith PetscFunctionReturn(0); 233408480c60SBarry Smith } 233508480c60SBarry Smith 2336898183ecSLois Curfman McInnes ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 2337898183ecSLois Curfman McInnes ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 23389e25ed09SBarry Smith 23399e25ed09SBarry Smith /* get new row pointers */ 23405a8e39fbSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 23415a8e39fbSHong Zhang bi[0] = 0; 23425a8e39fbSHong Zhang /* bdiag is location of diagonal in factor */ 23435a8e39fbSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 23445a8e39fbSHong Zhang bdiag[0] = 0; 23456cf997b0SBarry Smith 23465a8e39fbSHong Zhang ierr = PetscMalloc((2*n+1)*sizeof(PetscInt**),&bj_ptr);CHKERRQ(ierr); 23475a8e39fbSHong Zhang bjlvl_ptr = (PetscInt**)(bj_ptr + n); 23485a8e39fbSHong Zhang 23495a8e39fbSHong Zhang /* create a linked list for storing column indices of the active row */ 23505a8e39fbSHong Zhang nlnk = n + 1; 23515a8e39fbSHong Zhang ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 23525a8e39fbSHong Zhang 23535a8e39fbSHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 2354a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 23555a8e39fbSHong Zhang current_space = free_space; 2356a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 23575a8e39fbSHong Zhang current_space_lvl = free_space_lvl; 23585a8e39fbSHong Zhang 23595a8e39fbSHong Zhang for (i=0; i<n; i++) { 23605a8e39fbSHong Zhang nzi = 0; 23616cf997b0SBarry Smith /* copy current row into linked list */ 23625a8e39fbSHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 23633b4a8b6dSBarry Smith if (!nnz) SETERRQ2(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 23645a8e39fbSHong Zhang cols = aj + ai[r[i]]; 23655a8e39fbSHong Zhang lnk[i] = -1; /* marker to indicate if diagonal exists */ 23665a8e39fbSHong Zhang ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 23675a8e39fbSHong Zhang nzi += nlnk; 23686cf997b0SBarry Smith 23696cf997b0SBarry Smith /* make sure diagonal entry is included */ 23705a8e39fbSHong Zhang if (diagonal_fill && lnk[i] == -1) { 23716cf997b0SBarry Smith fm = n; 23725a8e39fbSHong Zhang while (lnk[fm] < i) fm = lnk[fm]; 23735a8e39fbSHong Zhang lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 23745a8e39fbSHong Zhang lnk[fm] = i; 23755a8e39fbSHong Zhang lnk_lvl[i] = 0; 23765a8e39fbSHong Zhang nzi++; dcount++; 23776cf997b0SBarry Smith } 23786cf997b0SBarry Smith 23795a8e39fbSHong Zhang /* add pivot rows into the active row */ 23805a8e39fbSHong Zhang nzbd = 0; 23815a8e39fbSHong Zhang prow = lnk[n]; 23825a8e39fbSHong Zhang while (prow < i) { 23835a8e39fbSHong Zhang nnz = bdiag[prow]; 23845a8e39fbSHong Zhang cols = bj_ptr[prow] + nnz + 1; 23855a8e39fbSHong Zhang cols_lvl = bjlvl_ptr[prow] + nnz + 1; 23865a8e39fbSHong Zhang nnz = bi[prow+1] - bi[prow] - nnz - 1; 23875a8e39fbSHong Zhang ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 23885a8e39fbSHong Zhang nzi += nlnk; 23895a8e39fbSHong Zhang prow = lnk[prow]; 23905a8e39fbSHong Zhang nzbd++; 239156beaf04SBarry Smith } 23925a8e39fbSHong Zhang bdiag[i] = nzbd; 23935a8e39fbSHong Zhang bi[i+1] = bi[i] + nzi; 2394ecf371e4SBarry Smith 23955a8e39fbSHong Zhang /* if free space is not available, make more free space */ 23965a8e39fbSHong Zhang if (current_space->local_remaining<nzi) { 23975a8e39fbSHong Zhang nnz = nzi*(n - i); /* estimated and max additional space needed */ 2398a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 2399a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 24005a8e39fbSHong Zhang reallocs++; 24015a8e39fbSHong Zhang } 2402ecf371e4SBarry Smith 24035a8e39fbSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 24045a8e39fbSHong Zhang ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 24055a8e39fbSHong Zhang bj_ptr[i] = current_space->array; 24065a8e39fbSHong Zhang bjlvl_ptr[i] = current_space_lvl->array; 24075a8e39fbSHong Zhang 24085a8e39fbSHong Zhang /* make sure the active row i has diagonal entry */ 24095a8e39fbSHong Zhang if (*(bj_ptr[i]+bdiag[i]) != i) { 241077431f27SBarry Smith SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\ 2411d7ee0231SBarry Smith try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i); 24126cf997b0SBarry Smith } 24135a8e39fbSHong Zhang 24145a8e39fbSHong Zhang current_space->array += nzi; 24155a8e39fbSHong Zhang current_space->local_used += nzi; 24165a8e39fbSHong Zhang current_space->local_remaining -= nzi; 24175a8e39fbSHong Zhang current_space_lvl->array += nzi; 24185a8e39fbSHong Zhang current_space_lvl->local_used += nzi; 24195a8e39fbSHong Zhang current_space_lvl->local_remaining -= nzi; 24209e25ed09SBarry Smith } 24215a8e39fbSHong Zhang 2422898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 2423898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 24245a8e39fbSHong Zhang 24255a8e39fbSHong Zhang /* destroy list of free space and other temporary arrays */ 24265a8e39fbSHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 24276516239fSHong Zhang ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */ 24285a8e39fbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2429a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 24305a8e39fbSHong Zhang ierr = PetscFree(bj_ptr);CHKERRQ(ierr); 24319e25ed09SBarry Smith 24326cf91177SBarry Smith #if defined(PETSC_USE_INFO) 2433f64065bbSBarry Smith { 24345a8e39fbSHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 2435ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 2436ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2437ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 2438ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 2439335d9088SBarry Smith if (diagonal_fill) { 2440ae15b995SBarry Smith ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 2441335d9088SBarry Smith } 2442f64065bbSBarry Smith } 244363ba0a88SBarry Smith #endif 2444bbb6d6a8SBarry Smith 24459e25ed09SBarry Smith /* put together the new matrix */ 244607dbf95fSBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 2447719d5645SBarry Smith ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 2448719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 2449e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2450e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 24517c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 2452052f0c41SBarry Smith ierr = PetscMalloc( (bi[n] )*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 24535a8e39fbSHong Zhang b->j = bj; 24545a8e39fbSHong Zhang b->i = bi; 24555a8e39fbSHong Zhang for (i=0; i<n; i++) bdiag[i] += bi[i]; 24565a8e39fbSHong Zhang b->diag = bdiag; 2457416022c9SBarry Smith b->ilen = 0; 2458416022c9SBarry Smith b->imax = 0; 2459416022c9SBarry Smith b->row = isrow; 2460416022c9SBarry Smith b->col = iscol; 2461c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 2462c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 246382bf6240SBarry Smith b->icol = isicol; 246487828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2465416022c9SBarry Smith /* In b structure: Free imax, ilen, old a, old j. 24665a8e39fbSHong Zhang Allocate bdiag, solve_work, new a, new j */ 2467719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 24685a8e39fbSHong Zhang b->maxnz = b->nz = bi[n] ; 2469719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 2470719d5645SBarry Smith (fact)->info.fill_ratio_given = f; 2471719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 2472719d5645SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 24734108e4d5SBarry Smith ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); 24743a40ed3dSBarry Smith PetscFunctionReturn(0); 24759e25ed09SBarry Smith } 2476d5d45c9bSBarry Smith 2477a6175056SHong Zhang #undef __FUNCT__ 24785f44c854SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ_newdatastruct" 24795f44c854SHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_newdatastruct(Mat B,Mat A,const MatFactorInfo *info) 24805f44c854SHong Zhang { 24815f44c854SHong Zhang Mat C = B; 24825f44c854SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 24835f44c854SHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 24845f44c854SHong Zhang IS ip=b->row,iip = b->icol; 24855f44c854SHong Zhang PetscErrorCode ierr; 24865f44c854SHong Zhang const PetscInt *rip,*riip; 24875f44c854SHong Zhang PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp; 24885f44c854SHong Zhang PetscInt *ai=a->i,*aj=a->j; 24895f44c854SHong Zhang PetscInt k,jmin,jmax,*c2r,*il,col,nexti,ili,nz; 24905f44c854SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 24915f44c854SHong Zhang PetscReal zeropivot,shiftnz; 24925f44c854SHong Zhang PetscReal shiftpd; 24935f44c854SHong Zhang PetscTruth perm_identity; 24945f44c854SHong Zhang 24955f44c854SHong Zhang PetscFunctionBegin; 24965f44c854SHong Zhang 24975f44c854SHong Zhang shiftnz = info->shiftnz; 24985f44c854SHong Zhang shiftpd = info->shiftpd; 24995f44c854SHong Zhang zeropivot = info->zeropivot; 25005f44c854SHong Zhang 25015f44c854SHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 25025f44c854SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 25035f44c854SHong Zhang 25045f44c854SHong Zhang /* allocate working arrays 25055f44c854SHong Zhang c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col 25065f44c854SHong 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 25075f44c854SHong Zhang */ 25085f44c854SHong Zhang nz = (2*mbs+1)*sizeof(PetscInt)+mbs*sizeof(MatScalar); 25095f44c854SHong Zhang ierr = PetscMalloc(nz,&il);CHKERRQ(ierr); 25105f44c854SHong Zhang c2r = il + mbs; 25115f44c854SHong Zhang rtmp = (MatScalar*)(c2r + mbs); 25125f44c854SHong Zhang 25135f44c854SHong Zhang for (i=0; i<mbs; i++) { 25145f44c854SHong Zhang c2r[i] = mbs; 25155f44c854SHong Zhang } 25165f44c854SHong Zhang il[0] = 0; 25175f44c854SHong Zhang 25185f44c854SHong Zhang for (k = 0; k<mbs; k++){ 25195f44c854SHong Zhang /* zero rtmp */ 25205f44c854SHong Zhang nz = bi[k+1] - bi[k]; 25215f44c854SHong Zhang bjtmp = bj + bi[k]; 25225f44c854SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 25235f44c854SHong Zhang 25245f44c854SHong Zhang /* load in initial unfactored row */ 25255f44c854SHong Zhang bval = ba + bi[k]; 25265f44c854SHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 25275f44c854SHong Zhang for (j = jmin; j < jmax; j++){ 25285f44c854SHong Zhang col = riip[aj[j]]; 25295f44c854SHong Zhang if (col >= k){ /* only take upper triangular entry */ 25305f44c854SHong Zhang rtmp[col] = aa[j]; 25315f44c854SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 25325f44c854SHong Zhang } 25335f44c854SHong Zhang } 25345f44c854SHong Zhang /* shift the diagonal of the matrix */ 25355f44c854SHong Zhang 25365f44c854SHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 25375f44c854SHong Zhang dk = rtmp[k]; 25385f44c854SHong Zhang i = c2r[k]; /* first row to be added to k_th row */ 25395f44c854SHong Zhang 25405f44c854SHong Zhang while (i < k){ 25415f44c854SHong Zhang nexti = c2r[i]; /* next row to be added to k_th row */ 25425f44c854SHong Zhang 25435f44c854SHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 25445f44c854SHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 25455f44c854SHong Zhang uikdi = - ba[ili]*ba[bdiag[i]]; /* diagonal(k) */ 25465f44c854SHong Zhang dk += uikdi*ba[ili]; /* update diag[k] */ 25475f44c854SHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 25485f44c854SHong Zhang 25495f44c854SHong Zhang /* add multiple of row i to k-th row */ 25505f44c854SHong Zhang jmin = ili + 1; jmax = bi[i+1]; 25515f44c854SHong Zhang if (jmin < jmax){ 25525f44c854SHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 25535f44c854SHong Zhang /* update il and c2r for row i */ 25545f44c854SHong Zhang il[i] = jmin; 25555f44c854SHong Zhang j = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i; 25565f44c854SHong Zhang } 25575f44c854SHong Zhang i = nexti; 25585f44c854SHong Zhang } 25595f44c854SHong Zhang 25605f44c854SHong Zhang /* copy data into U(k,:) */ 25615f44c854SHong Zhang ba[bdiag[k]] = 1.0/dk; /* U(k,k) */ 25625f44c854SHong Zhang jmin = bi[k]; jmax = bi[k+1]-1; 25635f44c854SHong Zhang if (jmin < jmax) { 25645f44c854SHong Zhang for (j=jmin; j<jmax; j++){ 25655f44c854SHong Zhang col = bj[j]; ba[j] = rtmp[col]; 25665f44c854SHong Zhang } 25675f44c854SHong Zhang /* add the k-th row into il and c2r */ 25685f44c854SHong Zhang il[k] = jmin; 25695f44c854SHong Zhang i = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k; 25705f44c854SHong Zhang } 25715f44c854SHong Zhang } 25725f44c854SHong Zhang 25735f44c854SHong Zhang ierr = PetscFree(il);CHKERRQ(ierr); 25745f44c854SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 25755f44c854SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 25765f44c854SHong Zhang 25775f44c854SHong Zhang ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 25785f44c854SHong Zhang if (perm_identity){ 25795f44c854SHong Zhang (B)->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering_newdatastruct; 25805f44c854SHong Zhang (B)->ops->solvetranspose = 0; 25815f44c854SHong Zhang (B)->ops->forwardsolve = 0; 25825f44c854SHong Zhang (B)->ops->backwardsolve = 0; 25835f44c854SHong Zhang } else { 2584783ef271SHong Zhang (B)->ops->solve = MatSolve_SeqSBAIJ_1_newdatastruct; 25855f44c854SHong Zhang (B)->ops->solvetranspose = 0; 25865f44c854SHong Zhang (B)->ops->forwardsolve = 0; 25875f44c854SHong Zhang (B)->ops->backwardsolve = 0; 25885f44c854SHong Zhang } 25895f44c854SHong Zhang 25905f44c854SHong Zhang C->assembled = PETSC_TRUE; 25915f44c854SHong Zhang C->preallocated = PETSC_TRUE; 25925f44c854SHong Zhang ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 25935f44c854SHong Zhang PetscFunctionReturn(0); 25945f44c854SHong Zhang } 25955f44c854SHong Zhang 25965f44c854SHong Zhang #undef __FUNCT__ 2597a6175056SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ" 25980481f469SBarry Smith PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 2599a6175056SHong Zhang { 2600719d5645SBarry Smith Mat C = B; 26010968510aSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 26022ed133dbSHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 26039bfd6278SHong Zhang IS ip=b->row,iip = b->icol; 26042ed133dbSHong Zhang PetscErrorCode ierr; 26055d0c19d7SBarry Smith const PetscInt *rip,*riip; 26065d0c19d7SBarry Smith PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol; 26072ed133dbSHong Zhang PetscInt *ai=a->i,*aj=a->j; 26082ed133dbSHong Zhang PetscInt k,jmin,jmax,*jl,*il,col,nexti,ili,nz; 2609622e2028SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 2610540703acSHong Zhang PetscReal zeropivot,rs,shiftnz; 2611fbf22428SSatish Balay PetscReal shiftpd; 2612540703acSHong Zhang ChShift_Ctx sctx; 2613540703acSHong Zhang PetscInt newshift; 2614db4efbfdSBarry Smith PetscTruth perm_identity; 2615d5d45c9bSBarry Smith 2616a6175056SHong Zhang PetscFunctionBegin; 2617540703acSHong Zhang shiftnz = info->shiftnz; 2618540703acSHong Zhang shiftpd = info->shiftpd; 2619ee45ca4aSHong Zhang zeropivot = info->zeropivot; 2620ee45ca4aSHong Zhang 26212ed133dbSHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 26229bfd6278SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 26232ed133dbSHong Zhang 26242ed133dbSHong Zhang /* initialization */ 26252ed133dbSHong Zhang nz = (2*mbs+1)*sizeof(PetscInt)+mbs*sizeof(MatScalar); 26262ed133dbSHong Zhang ierr = PetscMalloc(nz,&il);CHKERRQ(ierr); 26272ed133dbSHong Zhang jl = il + mbs; 26282ed133dbSHong Zhang rtmp = (MatScalar*)(jl + mbs); 26292ed133dbSHong Zhang 2630540703acSHong Zhang sctx.shift_amount = 0; 2631540703acSHong Zhang sctx.nshift = 0; 26322ed133dbSHong Zhang do { 2633540703acSHong Zhang sctx.chshift = PETSC_FALSE; 26342ed133dbSHong Zhang for (i=0; i<mbs; i++) { 26352ed133dbSHong Zhang rtmp[i] = 0.0; jl[i] = mbs; il[0] = 0; 2636a6175056SHong Zhang } 26372ed133dbSHong Zhang 26382ed133dbSHong Zhang for (k = 0; k<mbs; k++){ 2639c05c3958SHong Zhang bval = ba + bi[k]; 26402ed133dbSHong Zhang /* initialize k-th row by the perm[k]-th row of A */ 26412ed133dbSHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 26422ed133dbSHong Zhang for (j = jmin; j < jmax; j++){ 26439bfd6278SHong Zhang col = riip[aj[j]]; 26442ed133dbSHong Zhang if (col >= k){ /* only take upper triangular entry */ 26452ed133dbSHong Zhang rtmp[col] = aa[j]; 2646c05c3958SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 26472ed133dbSHong Zhang } 26482ed133dbSHong Zhang } 264939e3d630SHong Zhang /* shift the diagonal of the matrix */ 2650540703acSHong Zhang if (sctx.nshift) rtmp[k] += sctx.shift_amount; 26512ed133dbSHong Zhang 26522ed133dbSHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 26532ed133dbSHong Zhang dk = rtmp[k]; 26542ed133dbSHong Zhang i = jl[k]; /* first row to be added to k_th row */ 26552ed133dbSHong Zhang 26562ed133dbSHong Zhang while (i < k){ 26572ed133dbSHong Zhang nexti = jl[i]; /* next row to be added to k_th row */ 26582ed133dbSHong Zhang 26592ed133dbSHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 26602ed133dbSHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 26612ed133dbSHong Zhang uikdi = - ba[ili]*ba[bi[i]]; /* diagonal(k) */ 26622ed133dbSHong Zhang dk += uikdi*ba[ili]; 26632ed133dbSHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 26642ed133dbSHong Zhang 26652ed133dbSHong Zhang /* add multiple of row i to k-th row */ 26662ed133dbSHong Zhang jmin = ili + 1; jmax = bi[i+1]; 26672ed133dbSHong Zhang if (jmin < jmax){ 26682ed133dbSHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 26692ed133dbSHong Zhang /* update il and jl for row i */ 26702ed133dbSHong Zhang il[i] = jmin; 26712ed133dbSHong Zhang j = bj[jmin]; jl[i] = jl[j]; jl[j] = i; 26722ed133dbSHong Zhang } 26732ed133dbSHong Zhang i = nexti; 26742ed133dbSHong Zhang } 26752ed133dbSHong Zhang 26760697b6caSHong Zhang /* shift the diagonals when zero pivot is detected */ 26770697b6caSHong Zhang /* compute rs=sum of abs(off-diagonal) */ 26780697b6caSHong Zhang rs = 0.0; 26793c9e8598SHong Zhang jmin = bi[k]+1; 26803c9e8598SHong Zhang nz = bi[k+1] - jmin; 26813c9e8598SHong Zhang bcol = bj + jmin; 268204fbf559SBarry Smith for (j=0; j<nz; j++) { 268304fbf559SBarry Smith rs += PetscAbsScalar(rtmp[bcol[j]]); 26843c9e8598SHong Zhang } 2685540703acSHong Zhang 2686540703acSHong Zhang sctx.rs = rs; 2687540703acSHong Zhang sctx.pv = dk; 26885b5bc046SBarry Smith ierr = MatCholeskyCheckShift_inline(info,sctx,k,newshift);CHKERRQ(ierr); 2689117f1891Stmunson 2690117f1891Stmunson if (newshift == 1) { 2691117f1891Stmunson if (!sctx.shift_amount) { 2692117f1891Stmunson sctx.shift_amount = 1e-5; 2693117f1891Stmunson } 2694117f1891Stmunson break; 2695117f1891Stmunson } 26963c9e8598SHong Zhang 26973c9e8598SHong Zhang /* copy data into U(k,:) */ 269839e3d630SHong Zhang ba[bi[k]] = 1.0/dk; /* U(k,k) */ 269939e3d630SHong Zhang jmin = bi[k]+1; jmax = bi[k+1]; 270039e3d630SHong Zhang if (jmin < jmax) { 270139e3d630SHong Zhang for (j=jmin; j<jmax; j++){ 270239e3d630SHong Zhang col = bj[j]; ba[j] = rtmp[col]; rtmp[col] = 0.0; 27033c9e8598SHong Zhang } 270439e3d630SHong Zhang /* add the k-th row into il and jl */ 27053c9e8598SHong Zhang il[k] = jmin; 27063c9e8598SHong Zhang i = bj[jmin]; jl[k] = jl[i]; jl[i] = k; 27073c9e8598SHong Zhang } 27083c9e8598SHong Zhang } 2709540703acSHong Zhang } while (sctx.chshift); 27103c9e8598SHong Zhang ierr = PetscFree(il);CHKERRQ(ierr); 27113c9e8598SHong Zhang 271239e3d630SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 27139bfd6278SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 2714db4efbfdSBarry Smith 2715db4efbfdSBarry Smith ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 2716db4efbfdSBarry Smith if (perm_identity){ 2717719d5645SBarry Smith (B)->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering; 2718719d5645SBarry Smith (B)->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering; 2719719d5645SBarry Smith (B)->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering; 2720719d5645SBarry Smith (B)->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering; 2721db4efbfdSBarry Smith } else { 2722719d5645SBarry Smith (B)->ops->solve = MatSolve_SeqSBAIJ_1; 2723719d5645SBarry Smith (B)->ops->solvetranspose = MatSolve_SeqSBAIJ_1; 2724719d5645SBarry Smith (B)->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1; 2725719d5645SBarry Smith (B)->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1; 2726db4efbfdSBarry Smith } 2727db4efbfdSBarry Smith 27283c9e8598SHong Zhang C->assembled = PETSC_TRUE; 27293c9e8598SHong Zhang C->preallocated = PETSC_TRUE; 2730d0f46423SBarry Smith ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 2731540703acSHong Zhang if (sctx.nshift){ 2732540703acSHong Zhang if (shiftnz) { 27331e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 2734540703acSHong Zhang } else if (shiftpd) { 27351e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 27360697b6caSHong Zhang } 27373c9e8598SHong Zhang } 27383c9e8598SHong Zhang PetscFunctionReturn(0); 27393c9e8598SHong Zhang } 2740a6175056SHong Zhang 2741*8a76ed4fSHong Zhang /* 2742*8a76ed4fSHong Zhang icc() under revised new data structure. 2743*8a76ed4fSHong Zhang Factored arrays bj and ba are stored as 2744*8a76ed4fSHong Zhang U(0,:),...,U(i,:),U(n-1,:) 2745*8a76ed4fSHong Zhang 2746*8a76ed4fSHong Zhang ui=fact->i is an array of size n+1, in which 2747*8a76ed4fSHong Zhang ui+ 2748*8a76ed4fSHong Zhang ui[i]: points to 1st entry of U(i,:),i=0,...,n-1 2749*8a76ed4fSHong Zhang ui[n]: points to U(n-1,n-1)+1 2750*8a76ed4fSHong Zhang 2751*8a76ed4fSHong Zhang udiag=fact->diag is an array of size n,in which 2752*8a76ed4fSHong Zhang udiag[i]: points to diagonal of U(i,:), i=0,...,n-1 2753*8a76ed4fSHong Zhang 2754*8a76ed4fSHong Zhang U(i,:) contains udiag[i] as its last entry, i.e., 2755*8a76ed4fSHong Zhang U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 2756*8a76ed4fSHong Zhang */ 2757*8a76ed4fSHong Zhang 2758a6175056SHong Zhang #undef __FUNCT__ 27595f44c854SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ_newdatastruct" 27605f44c854SHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_newdatastruct(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2761a6175056SHong Zhang { 27620968510aSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2763ed59401aSHong Zhang Mat_SeqSBAIJ *b; 2764dfbe8321SBarry Smith PetscErrorCode ierr; 276558ebbce7SBarry Smith PetscTruth perm_identity,missing; 27665f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 27675d0c19d7SBarry Smith const PetscInt *rip,*riip; 2768ed59401aSHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 276958ebbce7SBarry Smith PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 27705a8e39fbSHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 2771ed59401aSHong Zhang PetscReal fill=info->fill,levels=info->levels; 2772a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 2773a1a86e44SBarry Smith PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 27747a48dd6fSHong Zhang PetscBT lnkbt; 2775b635d36bSHong Zhang IS iperm; 2776a6175056SHong Zhang 2777a6175056SHong Zhang PetscFunctionBegin; 2778d0f46423SBarry Smith if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 277958ebbce7SBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 278058ebbce7SBarry Smith if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 2781653a6975SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 2782b635d36bSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 27837a48dd6fSHong Zhang 278439e3d630SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 27855f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 278639e3d630SHong Zhang ui[0] = 0; 278739e3d630SHong Zhang 2788*8a76ed4fSHong Zhang /* ICC(0) without matrix ordering: simply rearrange column indices */ 2789622e2028SHong Zhang if (!levels && perm_identity) { 279058ebbce7SBarry Smith 2791ed59401aSHong Zhang for (i=0; i<am; i++) { 279239e3d630SHong Zhang ui[i+1] = ui[i] + ai[i+1] - a->diag[i]; 27935f44c854SHong Zhang udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */ 27945f44c854SHong Zhang } 27955f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 27965f44c854SHong Zhang cols = uj; 27975f44c854SHong Zhang for (i=0; i<am; i++) { 27985f44c854SHong Zhang aj = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */ 27995f44c854SHong Zhang ncols = ui[i+1] - ui[i] - 1; 28005f44c854SHong Zhang for (j=0; j<ncols; j++) *cols++ = aj[j]; 28015f44c854SHong Zhang *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */ 28025f44c854SHong Zhang } 28035f44c854SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 2804*8a76ed4fSHong Zhang if (levels) SETERRQ(PETSC_ERR_ARG_WRONG,"not done yet"); 28055f44c854SHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 28065f44c854SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 28075f44c854SHong Zhang 28085f44c854SHong Zhang /* initialization */ 28095f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 28105f44c854SHong Zhang 28115f44c854SHong Zhang /* jl: linked list for storing indices of the pivot rows 28125f44c854SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 28135f44c854SHong Zhang ierr = PetscMalloc((2*am+1)*sizeof(PetscInt)+2*am*sizeof(PetscInt**),&jl);CHKERRQ(ierr); 28145f44c854SHong Zhang il = jl + am; 28155f44c854SHong Zhang uj_ptr = (PetscInt**)(il + am); 28165f44c854SHong Zhang uj_lvl_ptr = (PetscInt**)(uj_ptr + am); 28175f44c854SHong Zhang for (i=0; i<am; i++){ 28185f44c854SHong Zhang jl[i] = am; il[i] = 0; 28195f44c854SHong Zhang } 28205f44c854SHong Zhang 28215f44c854SHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 28225f44c854SHong Zhang nlnk = am + 1; 28235f44c854SHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 28245f44c854SHong Zhang 28255f44c854SHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 28265f44c854SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 28275f44c854SHong Zhang current_space = free_space; 28285f44c854SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr); 28295f44c854SHong Zhang current_space_lvl = free_space_lvl; 28305f44c854SHong Zhang 28315f44c854SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 28325f44c854SHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 28335f44c854SHong Zhang nzk = 0; 28345f44c854SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 28355f44c854SHong Zhang if (!ncols) SETERRQ2(PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k); 28365f44c854SHong Zhang ncols_upper = 0; 28375f44c854SHong Zhang for (j=0; j<ncols; j++){ 28385f44c854SHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 28395f44c854SHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 28405f44c854SHong Zhang ajtmp[ncols_upper] = i; 28415f44c854SHong Zhang ncols_upper++; 28425f44c854SHong Zhang } 28435f44c854SHong Zhang } 28445f44c854SHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 28455f44c854SHong Zhang nzk += nlnk; 28465f44c854SHong Zhang 28475f44c854SHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 28485f44c854SHong Zhang prow = jl[k]; /* 1st pivot row */ 28495f44c854SHong Zhang 28505f44c854SHong Zhang while (prow < k){ 28515f44c854SHong Zhang nextprow = jl[prow]; 28525f44c854SHong Zhang 28535f44c854SHong Zhang /* merge prow into k-th row */ 28545f44c854SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 28555f44c854SHong Zhang jmax = ui[prow+1]; 28565f44c854SHong Zhang ncols = jmax-jmin; 28575f44c854SHong Zhang i = jmin - ui[prow]; 28585f44c854SHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 28595f44c854SHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 28605f44c854SHong Zhang j = *(uj - 1); 28615f44c854SHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 28625f44c854SHong Zhang nzk += nlnk; 28635f44c854SHong Zhang 28645f44c854SHong Zhang /* update il and jl for prow */ 28655f44c854SHong Zhang if (jmin < jmax){ 28665f44c854SHong Zhang il[prow] = jmin; 28675f44c854SHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 28685f44c854SHong Zhang } 28695f44c854SHong Zhang prow = nextprow; 28705f44c854SHong Zhang } 28715f44c854SHong Zhang 28725f44c854SHong Zhang /* if free space is not available, make more free space */ 28735f44c854SHong Zhang if (current_space->local_remaining<nzk) { 28745f44c854SHong Zhang i = am - k + 1; /* num of unfactored rows */ 28755f44c854SHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 28765f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 28775f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 28785f44c854SHong Zhang reallocs++; 28795f44c854SHong Zhang } 28805f44c854SHong Zhang 28815f44c854SHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 28825f44c854SHong Zhang if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 28835f44c854SHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 28845f44c854SHong Zhang 28855f44c854SHong Zhang /* add the k-th row into il and jl */ 28865f44c854SHong Zhang if (nzk > 1){ 28875f44c854SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 28885f44c854SHong Zhang jl[k] = jl[i]; jl[i] = k; 28895f44c854SHong Zhang il[k] = ui[k] + 1; 28905f44c854SHong Zhang } 28915f44c854SHong Zhang uj_ptr[k] = current_space->array; 28925f44c854SHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 28935f44c854SHong Zhang 28945f44c854SHong Zhang current_space->array += nzk; 28955f44c854SHong Zhang current_space->local_used += nzk; 28965f44c854SHong Zhang current_space->local_remaining -= nzk; 28975f44c854SHong Zhang 28985f44c854SHong Zhang current_space_lvl->array += nzk; 28995f44c854SHong Zhang current_space_lvl->local_used += nzk; 29005f44c854SHong Zhang current_space_lvl->local_remaining -= nzk; 29015f44c854SHong Zhang 29025f44c854SHong Zhang ui[k+1] = ui[k] + nzk; 29035f44c854SHong Zhang } 29045f44c854SHong Zhang 29055f44c854SHong Zhang #if defined(PETSC_USE_INFO) 29065f44c854SHong Zhang if (ai[am] != 0) { 29075f44c854SHong Zhang PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]); 29085f44c854SHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 29095f44c854SHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 29105f44c854SHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 29115f44c854SHong Zhang } else { 29125f44c854SHong Zhang ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 29135f44c854SHong Zhang } 29145f44c854SHong Zhang #endif 29155f44c854SHong Zhang 29165f44c854SHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 29175f44c854SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 29185f44c854SHong Zhang ierr = PetscFree(jl);CHKERRQ(ierr); 29195f44c854SHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 29205f44c854SHong Zhang 29215f44c854SHong Zhang /* destroy list of free space and other temporary array(s) */ 29225f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2923783ef271SHong Zhang ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); 29245f44c854SHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 29255f44c854SHong Zhang ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 29265f44c854SHong Zhang 29275f44c854SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 29285f44c854SHong Zhang 29295f44c854SHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 29305f44c854SHong Zhang 29315f44c854SHong Zhang b = (Mat_SeqSBAIJ*)(fact)->data; 29325f44c854SHong Zhang b->singlemalloc = PETSC_FALSE; 29335f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 29345f44c854SHong Zhang b->j = uj; 29355f44c854SHong Zhang b->i = ui; 29365f44c854SHong Zhang b->diag = udiag; 29377f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 29385f44c854SHong Zhang b->ilen = 0; 29395f44c854SHong Zhang b->imax = 0; 29405f44c854SHong Zhang b->row = perm; 29415f44c854SHong Zhang b->col = perm; 29425f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 29435f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 29445f44c854SHong Zhang b->icol = iperm; 29455f44c854SHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 29465f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 29475f44c854SHong Zhang ierr = PetscLogObjectMemory((fact),(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 29485f44c854SHong Zhang b->maxnz = b->nz = ui[am]; 29495f44c854SHong Zhang b->free_a = PETSC_TRUE; 29505f44c854SHong Zhang b->free_ij = PETSC_TRUE; 29515f44c854SHong Zhang 29525f44c854SHong Zhang (fact)->info.factor_mallocs = reallocs; 29535f44c854SHong Zhang (fact)->info.fill_ratio_given = fill; 29545f44c854SHong Zhang if (ai[am] != 0) { 29555f44c854SHong Zhang (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 29565f44c854SHong Zhang } else { 29575f44c854SHong Zhang (fact)->info.fill_ratio_needed = 0.0; 29585f44c854SHong Zhang } 29595f44c854SHong Zhang (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_newdatastruct; 29605f44c854SHong Zhang PetscFunctionReturn(0); 29615f44c854SHong Zhang } 29625f44c854SHong Zhang 29635f44c854SHong Zhang #undef __FUNCT__ 29645f44c854SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ" 29655f44c854SHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 29665f44c854SHong Zhang { 29675f44c854SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 29685f44c854SHong Zhang Mat_SeqSBAIJ *b; 29695f44c854SHong Zhang PetscErrorCode ierr; 29705f44c854SHong Zhang PetscTruth perm_identity,missing; 29715f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 29725f44c854SHong Zhang const PetscInt *rip,*riip; 29735f44c854SHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 29745f44c854SHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 29755f44c854SHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 29765f44c854SHong Zhang PetscReal fill=info->fill,levels=info->levels; 29775f44c854SHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 29785f44c854SHong Zhang PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 29795f44c854SHong Zhang PetscBT lnkbt; 29805f44c854SHong Zhang IS iperm; 29815f44c854SHong Zhang PetscTruth newdatastruct=PETSC_FALSE; 29825f44c854SHong Zhang 29835f44c854SHong Zhang PetscFunctionBegin; 29845f44c854SHong Zhang ierr = PetscOptionsGetTruth(PETSC_NULL,"-icc_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr); 29855f44c854SHong Zhang if(newdatastruct){ 29865f44c854SHong Zhang ierr = MatICCFactorSymbolic_SeqAIJ_newdatastruct(fact,A,perm,info);CHKERRQ(ierr); 29875f44c854SHong Zhang PetscFunctionReturn(0); 29885f44c854SHong Zhang } 29895f44c854SHong Zhang 29905f44c854SHong Zhang if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 29915f44c854SHong Zhang ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 29925f44c854SHong Zhang if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 29935f44c854SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 29945f44c854SHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 29955f44c854SHong Zhang 29965f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 29975f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 29985f44c854SHong Zhang ui[0] = 0; 29995f44c854SHong Zhang 30005f44c854SHong Zhang /* ICC(0) without matrix ordering: simply copies fill pattern */ 30015f44c854SHong Zhang if (!levels && perm_identity) { 30025f44c854SHong Zhang 30035f44c854SHong Zhang for (i=0; i<am; i++) { 30045f44c854SHong Zhang ui[i+1] = ui[i] + ai[i+1] - a->diag[i]; 30055f44c854SHong Zhang udiag[i] = ui[i]; 3006ed59401aSHong Zhang } 300739e3d630SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 300839e3d630SHong Zhang cols = uj; 3009ed59401aSHong Zhang for (i=0; i<am; i++) { 3010ed59401aSHong Zhang aj = a->j + a->diag[i]; 301139e3d630SHong Zhang ncols = ui[i+1] - ui[i]; 301239e3d630SHong Zhang for (j=0; j<ncols; j++) *cols++ = *aj++; 3013ed59401aSHong Zhang } 301439e3d630SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 3015b635d36bSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 3016b635d36bSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 3017b635d36bSHong Zhang 30187a48dd6fSHong Zhang /* initialization */ 30195a8e39fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 30207a48dd6fSHong Zhang 30217a48dd6fSHong Zhang /* jl: linked list for storing indices of the pivot rows 30227a48dd6fSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 30231393bc97SHong Zhang ierr = PetscMalloc((2*am+1)*sizeof(PetscInt)+2*am*sizeof(PetscInt**),&jl);CHKERRQ(ierr); 30247a48dd6fSHong Zhang il = jl + am; 30257a48dd6fSHong Zhang uj_ptr = (PetscInt**)(il + am); 30267a48dd6fSHong Zhang uj_lvl_ptr = (PetscInt**)(uj_ptr + am); 30277a48dd6fSHong Zhang for (i=0; i<am; i++){ 30287a48dd6fSHong Zhang jl[i] = am; il[i] = 0; 30297a48dd6fSHong Zhang } 30307a48dd6fSHong Zhang 30317a48dd6fSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 30327a48dd6fSHong Zhang nlnk = am + 1; 30332ed133dbSHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 30347a48dd6fSHong Zhang 30357a48dd6fSHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 3036a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 30377a48dd6fSHong Zhang current_space = free_space; 3038a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr); 30397a48dd6fSHong Zhang current_space_lvl = free_space_lvl; 30407a48dd6fSHong Zhang 30417a48dd6fSHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 30427a48dd6fSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 30437a48dd6fSHong Zhang nzk = 0; 3044622e2028SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 30453b4a8b6dSBarry Smith if (!ncols) SETERRQ2(PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k); 3046622e2028SHong Zhang ncols_upper = 0; 3047622e2028SHong Zhang for (j=0; j<ncols; j++){ 3048b635d36bSHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 3049b635d36bSHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 30505a8e39fbSHong Zhang ajtmp[ncols_upper] = i; 3051622e2028SHong Zhang ncols_upper++; 3052622e2028SHong Zhang } 3053622e2028SHong Zhang } 3054b635d36bSHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 30557a48dd6fSHong Zhang nzk += nlnk; 30567a48dd6fSHong Zhang 30577a48dd6fSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 30587a48dd6fSHong Zhang prow = jl[k]; /* 1st pivot row */ 30597a48dd6fSHong Zhang 30607a48dd6fSHong Zhang while (prow < k){ 30617a48dd6fSHong Zhang nextprow = jl[prow]; 30627a48dd6fSHong Zhang 30637a48dd6fSHong Zhang /* merge prow into k-th row */ 30647a48dd6fSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 30657a48dd6fSHong Zhang jmax = ui[prow+1]; 30667a48dd6fSHong Zhang ncols = jmax-jmin; 3067ed59401aSHong Zhang i = jmin - ui[prow]; 3068ed59401aSHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 30695a8e39fbSHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 30705a8e39fbSHong Zhang j = *(uj - 1); 30715a8e39fbSHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 30727a48dd6fSHong Zhang nzk += nlnk; 30737a48dd6fSHong Zhang 30747a48dd6fSHong Zhang /* update il and jl for prow */ 30757a48dd6fSHong Zhang if (jmin < jmax){ 30767a48dd6fSHong Zhang il[prow] = jmin; 30777a48dd6fSHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 30787a48dd6fSHong Zhang } 30797a48dd6fSHong Zhang prow = nextprow; 30807a48dd6fSHong Zhang } 30817a48dd6fSHong Zhang 30827a48dd6fSHong Zhang /* if free space is not available, make more free space */ 30837a48dd6fSHong Zhang if (current_space->local_remaining<nzk) { 30847a48dd6fSHong Zhang i = am - k + 1; /* num of unfactored rows */ 30857a48dd6fSHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 3086a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 3087a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 30887a48dd6fSHong Zhang reallocs++; 30897a48dd6fSHong Zhang } 30907a48dd6fSHong Zhang 30917a48dd6fSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 3092b635d36bSHong Zhang if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 30932ed133dbSHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 30947a48dd6fSHong Zhang 30957a48dd6fSHong Zhang /* add the k-th row into il and jl */ 309639e3d630SHong Zhang if (nzk > 1){ 30977a48dd6fSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 30987a48dd6fSHong Zhang jl[k] = jl[i]; jl[i] = k; 30997a48dd6fSHong Zhang il[k] = ui[k] + 1; 31007a48dd6fSHong Zhang } 31017a48dd6fSHong Zhang uj_ptr[k] = current_space->array; 31027a48dd6fSHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 31037a48dd6fSHong Zhang 31047a48dd6fSHong Zhang current_space->array += nzk; 31057a48dd6fSHong Zhang current_space->local_used += nzk; 31067a48dd6fSHong Zhang current_space->local_remaining -= nzk; 31077a48dd6fSHong Zhang 31087a48dd6fSHong Zhang current_space_lvl->array += nzk; 31097a48dd6fSHong Zhang current_space_lvl->local_used += nzk; 31107a48dd6fSHong Zhang current_space_lvl->local_remaining -= nzk; 31117a48dd6fSHong Zhang 31127a48dd6fSHong Zhang ui[k+1] = ui[k] + nzk; 31137a48dd6fSHong Zhang } 31147a48dd6fSHong Zhang 31156cf91177SBarry Smith #if defined(PETSC_USE_INFO) 31167a48dd6fSHong Zhang if (ai[am] != 0) { 311739e3d630SHong Zhang PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]); 3118ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 3119ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 3120ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 31217a48dd6fSHong Zhang } else { 3122ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 31237a48dd6fSHong Zhang } 312463ba0a88SBarry Smith #endif 31257a48dd6fSHong Zhang 31267a48dd6fSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 3127b635d36bSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 31287a48dd6fSHong Zhang ierr = PetscFree(jl);CHKERRQ(ierr); 31295a8e39fbSHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 31307a48dd6fSHong Zhang 31317a48dd6fSHong Zhang /* destroy list of free space and other temporary array(s) */ 31327a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 3133a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 31342ed133dbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 3135a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 31367a48dd6fSHong Zhang 313739e3d630SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 313839e3d630SHong Zhang 31397a48dd6fSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 31407a48dd6fSHong Zhang 3141719d5645SBarry Smith b = (Mat_SeqSBAIJ*)(fact)->data; 31427a48dd6fSHong Zhang b->singlemalloc = PETSC_FALSE; 31437a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 31447a48dd6fSHong Zhang b->j = uj; 31457a48dd6fSHong Zhang b->i = ui; 31465f44c854SHong Zhang b->diag = udiag; 31477f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 31487a48dd6fSHong Zhang b->ilen = 0; 31497a48dd6fSHong Zhang b->imax = 0; 31507a48dd6fSHong Zhang b->row = perm; 3151b635d36bSHong Zhang b->col = perm; 3152b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 3153b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 3154b635d36bSHong Zhang b->icol = iperm; 31557a48dd6fSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 31567a48dd6fSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 3157719d5645SBarry Smith ierr = PetscLogObjectMemory((fact),(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 31587a48dd6fSHong Zhang b->maxnz = b->nz = ui[am]; 3159e6b907acSBarry Smith b->free_a = PETSC_TRUE; 3160e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 31617a48dd6fSHong Zhang 3162719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 3163719d5645SBarry Smith (fact)->info.fill_ratio_given = fill; 31647a48dd6fSHong Zhang if (ai[am] != 0) { 3165719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 31667a48dd6fSHong Zhang } else { 3167719d5645SBarry Smith (fact)->info.fill_ratio_needed = 0.0; 31687a48dd6fSHong Zhang } 3169719d5645SBarry Smith (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 3170a6175056SHong Zhang PetscFunctionReturn(0); 3171a6175056SHong Zhang } 3172d5d45c9bSBarry Smith 3173f76d2b81SHong Zhang #undef __FUNCT__ 3174f76d2b81SHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ" 31750481f469SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 3176f76d2b81SHong Zhang { 3177f76d2b81SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 317810c27e3dSHong Zhang Mat_SeqSBAIJ *b; 3179dfbe8321SBarry Smith PetscErrorCode ierr; 3180f76d2b81SHong Zhang PetscTruth perm_identity; 318110c27e3dSHong Zhang PetscReal fill = info->fill; 31825d0c19d7SBarry Smith const PetscInt *rip,*riip; 31835d0c19d7SBarry Smith PetscInt i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 318410c27e3dSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 31852ed133dbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr; 3186a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 318710c27e3dSHong Zhang PetscBT lnkbt; 31882ed133dbSHong Zhang IS iperm; 3189f76d2b81SHong Zhang 3190f76d2b81SHong Zhang PetscFunctionBegin; 3191d0f46423SBarry Smith if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 31922ed133dbSHong Zhang /* check whether perm is the identity mapping */ 3193f76d2b81SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 31942ed133dbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 31952ed133dbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 31969bfd6278SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 319710c27e3dSHong Zhang 319810c27e3dSHong Zhang /* initialization */ 31991393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 320010c27e3dSHong Zhang ui[0] = 0; 320110c27e3dSHong Zhang 320210c27e3dSHong Zhang /* jl: linked list for storing indices of the pivot rows 32031393bc97SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 32041393bc97SHong Zhang ierr = PetscMalloc((3*am+1)*sizeof(PetscInt)+am*sizeof(PetscInt**),&jl);CHKERRQ(ierr); 32051393bc97SHong Zhang il = jl + am; 32061393bc97SHong Zhang cols = il + am; 32071393bc97SHong Zhang ui_ptr = (PetscInt**)(cols + am); 32081393bc97SHong Zhang for (i=0; i<am; i++){ 32091393bc97SHong Zhang jl[i] = am; il[i] = 0; 3210f76d2b81SHong Zhang } 321110c27e3dSHong Zhang 321210c27e3dSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 32131393bc97SHong Zhang nlnk = am + 1; 32141393bc97SHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 321510c27e3dSHong Zhang 32161393bc97SHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 3217a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 321810c27e3dSHong Zhang current_space = free_space; 321910c27e3dSHong Zhang 32201393bc97SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 322110c27e3dSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 322210c27e3dSHong Zhang nzk = 0; 32232ed133dbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 32243b4a8b6dSBarry Smith if (!ncols) SETERRQ2(PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k); 32252ed133dbSHong Zhang ncols_upper = 0; 32262ed133dbSHong Zhang for (j=0; j<ncols; j++){ 32279bfd6278SHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 32282ed133dbSHong Zhang if (i >= k){ /* only take upper triangular entry */ 32292ed133dbSHong Zhang cols[ncols_upper] = i; 32302ed133dbSHong Zhang ncols_upper++; 32312ed133dbSHong Zhang } 32322ed133dbSHong Zhang } 32331393bc97SHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 323410c27e3dSHong Zhang nzk += nlnk; 323510c27e3dSHong Zhang 323610c27e3dSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 323710c27e3dSHong Zhang prow = jl[k]; /* 1st pivot row */ 323810c27e3dSHong Zhang 323910c27e3dSHong Zhang while (prow < k){ 324010c27e3dSHong Zhang nextprow = jl[prow]; 324110c27e3dSHong Zhang /* merge prow into k-th row */ 32421393bc97SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 324310c27e3dSHong Zhang jmax = ui[prow+1]; 324410c27e3dSHong Zhang ncols = jmax-jmin; 32451393bc97SHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 32465a8e39fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 324710c27e3dSHong Zhang nzk += nlnk; 324810c27e3dSHong Zhang 324910c27e3dSHong Zhang /* update il and jl for prow */ 325010c27e3dSHong Zhang if (jmin < jmax){ 325110c27e3dSHong Zhang il[prow] = jmin; 32522ed133dbSHong Zhang j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow; 325310c27e3dSHong Zhang } 325410c27e3dSHong Zhang prow = nextprow; 325510c27e3dSHong Zhang } 325610c27e3dSHong Zhang 325710c27e3dSHong Zhang /* if free space is not available, make more free space */ 325810c27e3dSHong Zhang if (current_space->local_remaining<nzk) { 32591393bc97SHong Zhang i = am - k + 1; /* num of unfactored rows */ 326010c27e3dSHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 3261a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 326210c27e3dSHong Zhang reallocs++; 326310c27e3dSHong Zhang } 326410c27e3dSHong Zhang 326510c27e3dSHong Zhang /* copy data into free space, then initialize lnk */ 32661393bc97SHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 326710c27e3dSHong Zhang 326810c27e3dSHong Zhang /* add the k-th row into il and jl */ 326910c27e3dSHong Zhang if (nzk-1 > 0){ 32701393bc97SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 327110c27e3dSHong Zhang jl[k] = jl[i]; jl[i] = k; 327210c27e3dSHong Zhang il[k] = ui[k] + 1; 327310c27e3dSHong Zhang } 32742ed133dbSHong Zhang ui_ptr[k] = current_space->array; 327510c27e3dSHong Zhang current_space->array += nzk; 327610c27e3dSHong Zhang current_space->local_used += nzk; 327710c27e3dSHong Zhang current_space->local_remaining -= nzk; 327810c27e3dSHong Zhang 327910c27e3dSHong Zhang ui[k+1] = ui[k] + nzk; 328010c27e3dSHong Zhang } 328110c27e3dSHong Zhang 32826cf91177SBarry Smith #if defined(PETSC_USE_INFO) 32831393bc97SHong Zhang if (ai[am] != 0) { 32841393bc97SHong Zhang PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]); 3285ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 3286ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 3287ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 328810c27e3dSHong Zhang } else { 3289ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 329010c27e3dSHong Zhang } 329163ba0a88SBarry Smith #endif 329210c27e3dSHong Zhang 329310c27e3dSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 32949bfd6278SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 329510c27e3dSHong Zhang ierr = PetscFree(jl);CHKERRQ(ierr); 329610c27e3dSHong Zhang 329710c27e3dSHong Zhang /* destroy list of free space and other temporary array(s) */ 32981393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 3299a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 330010c27e3dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 330110c27e3dSHong Zhang 330210c27e3dSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 330310c27e3dSHong Zhang 3304719d5645SBarry Smith b = (Mat_SeqSBAIJ*)(fact)->data; 330510c27e3dSHong Zhang b->singlemalloc = PETSC_FALSE; 3306e6b907acSBarry Smith b->free_a = PETSC_TRUE; 3307e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 33081393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 330910c27e3dSHong Zhang b->j = uj; 331010c27e3dSHong Zhang b->i = ui; 331110c27e3dSHong Zhang b->diag = 0; 331210c27e3dSHong Zhang b->ilen = 0; 331310c27e3dSHong Zhang b->imax = 0; 331410c27e3dSHong Zhang b->row = perm; 33159bfd6278SHong Zhang b->col = perm; 33169bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 33179bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 33189bfd6278SHong Zhang b->icol = iperm; 331910c27e3dSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 33201393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 3321719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 33221393bc97SHong Zhang b->maxnz = b->nz = ui[am]; 332310c27e3dSHong Zhang 3324719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 3325719d5645SBarry Smith (fact)->info.fill_ratio_given = fill; 33261393bc97SHong Zhang if (ai[am] != 0) { 3327719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 332810c27e3dSHong Zhang } else { 3329719d5645SBarry Smith (fact)->info.fill_ratio_needed = 0.0; 333010c27e3dSHong Zhang } 3331719d5645SBarry Smith (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 3332f76d2b81SHong Zhang PetscFunctionReturn(0); 3333f76d2b81SHong Zhang } 3334599ef60dSHong Zhang 3335599ef60dSHong Zhang #undef __FUNCT__ 3336894a2336SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_newdatastruct" 3337894a2336SHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_newdatastruct(Mat A,Vec bb,Vec xx) 3338599ef60dSHong Zhang { 33391d098868SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 33401d098868SHong Zhang PetscErrorCode ierr; 33411d098868SHong Zhang PetscInt n = A->rmap->n; 33428fc3a347SHong Zhang const PetscInt *ai = a->i,*aj = a->j,*vi; 33431d098868SHong Zhang PetscScalar *x,sum; 33441d098868SHong Zhang const PetscScalar *b; 33451d098868SHong Zhang const MatScalar *aa = a->a,*v; 33468fc3a347SHong Zhang PetscInt i,nz; 33471d098868SHong Zhang 33481d098868SHong Zhang PetscFunctionBegin; 33491d098868SHong Zhang if (!n) PetscFunctionReturn(0); 33501d098868SHong Zhang 33511d098868SHong Zhang ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 33521d098868SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 33531d098868SHong Zhang 33541d098868SHong Zhang /* forward solve the lower triangular */ 33551d098868SHong Zhang x[0] = b[0]; 33561d098868SHong Zhang v = aa; 33571d098868SHong Zhang vi = aj; 33581d098868SHong Zhang for (i=1; i<n; i++) { 33591d098868SHong Zhang nz = ai[i+1] - ai[i]; 33601d098868SHong Zhang sum = b[i]; 3361003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 3362e0e43300SBarry Smith v += nz; 3363e0e43300SBarry Smith vi += nz; 33641d098868SHong Zhang x[i] = sum; 33651d098868SHong Zhang } 33661d098868SHong Zhang 33671d098868SHong Zhang /* backward solve the upper triangular */ 33688fc3a347SHong Zhang v = aa + ai[n+1]; 33698fc3a347SHong Zhang vi = aj + ai[n+1]; 33701d098868SHong Zhang for (i=n-1; i>=0; i--){ 33718fc3a347SHong Zhang nz = ai[2*n-i +1] - ai[2*n-i]-1; 33721d098868SHong Zhang sum = x[i]; 3373003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 3374e0e43300SBarry Smith v += nz; 33758fc3a347SHong Zhang vi += nz; vi++; 3376fe31b70dSHong Zhang x[i] = *v++ *sum; /* x[i]=aa[adiag[i]]*sum; v++; */ 33771d098868SHong Zhang } 33781d098868SHong Zhang 33791d098868SHong Zhang ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 33801d098868SHong Zhang ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 33811d098868SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 33821d098868SHong Zhang PetscFunctionReturn(0); 33831d098868SHong Zhang } 33841d098868SHong Zhang 33851d098868SHong Zhang #undef __FUNCT__ 3386813a1f2bSShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_newdatastruct_v2" 3387813a1f2bSShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_newdatastruct_v2(Mat A,Vec bb,Vec xx) 3388813a1f2bSShri Abhyankar { 3389813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3390813a1f2bSShri Abhyankar PetscErrorCode ierr; 3391813a1f2bSShri Abhyankar PetscInt n = A->rmap->n; 3392813a1f2bSShri Abhyankar const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag,*vi; 3393813a1f2bSShri Abhyankar PetscScalar *x,sum; 3394813a1f2bSShri Abhyankar const PetscScalar *b; 3395813a1f2bSShri Abhyankar const MatScalar *aa = a->a,*v; 3396813a1f2bSShri Abhyankar PetscInt i,nz; 3397813a1f2bSShri Abhyankar 3398813a1f2bSShri Abhyankar PetscFunctionBegin; 3399813a1f2bSShri Abhyankar if (!n) PetscFunctionReturn(0); 3400813a1f2bSShri Abhyankar 3401813a1f2bSShri Abhyankar ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 3402813a1f2bSShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3403813a1f2bSShri Abhyankar 3404813a1f2bSShri Abhyankar /* forward solve the lower triangular */ 3405813a1f2bSShri Abhyankar x[0] = b[0]; 3406813a1f2bSShri Abhyankar v = aa; 3407813a1f2bSShri Abhyankar vi = aj; 3408813a1f2bSShri Abhyankar for (i=1; i<n; i++) { 3409813a1f2bSShri Abhyankar nz = ai[i+1] - ai[i]; 3410813a1f2bSShri Abhyankar sum = b[i]; 3411813a1f2bSShri Abhyankar PetscSparseDenseMinusDot(sum,x,v,vi,nz); 3412813a1f2bSShri Abhyankar v += nz; 3413813a1f2bSShri Abhyankar vi += nz; 3414813a1f2bSShri Abhyankar x[i] = sum; 3415813a1f2bSShri Abhyankar } 3416813a1f2bSShri Abhyankar 3417813a1f2bSShri Abhyankar /* backward solve the upper triangular */ 3418813a1f2bSShri Abhyankar /* v = aa + ai[n+1]; 3419813a1f2bSShri Abhyankar vi = aj + ai[n+1]; */ 3420813a1f2bSShri Abhyankar v = aa + adiag[n-1]; 3421813a1f2bSShri Abhyankar vi = aj + adiag[n-1]; 342262fbd6afSShri Abhyankar for (i=n-1; i>=0; i--){ 3423813a1f2bSShri Abhyankar nz = adiag[i] - adiag[i+1]-1; 3424813a1f2bSShri Abhyankar sum = x[i]; 3425813a1f2bSShri Abhyankar PetscSparseDenseMinusDot(sum,x,v,vi,nz); 3426813a1f2bSShri Abhyankar v += nz; 3427813a1f2bSShri Abhyankar vi += nz; vi++; 3428813a1f2bSShri Abhyankar x[i] = *v++ *sum; /* x[i]=aa[adiag[i]]*sum; v++; */ 3429813a1f2bSShri Abhyankar } 3430813a1f2bSShri Abhyankar 3431813a1f2bSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 3432813a1f2bSShri Abhyankar ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 3433813a1f2bSShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3434813a1f2bSShri Abhyankar PetscFunctionReturn(0); 3435813a1f2bSShri Abhyankar } 3436813a1f2bSShri Abhyankar 3437813a1f2bSShri Abhyankar #undef __FUNCT__ 3438894a2336SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_newdatastruct" 3439894a2336SHong Zhang PetscErrorCode MatSolve_SeqAIJ_newdatastruct(Mat A,Vec bb,Vec xx) 34401d098868SHong Zhang { 34411d098868SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 34421d098868SHong Zhang IS iscol = a->col,isrow = a->row; 34431d098868SHong Zhang PetscErrorCode ierr; 3444894a2336SHong Zhang PetscInt i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,nz,k; 34451d098868SHong Zhang const PetscInt *rout,*cout,*r,*c; 3446894a2336SHong Zhang PetscScalar *x,*tmp,*tmps,sum; 34471d098868SHong Zhang const PetscScalar *b; 34481d098868SHong Zhang const MatScalar *aa = a->a,*v; 34491d098868SHong Zhang 34501d098868SHong Zhang PetscFunctionBegin; 34511d098868SHong Zhang if (!n) PetscFunctionReturn(0); 34521d098868SHong Zhang 34531d098868SHong Zhang ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 34541d098868SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 34551d098868SHong Zhang tmp = a->solve_work; 34561d098868SHong Zhang 34571d098868SHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 3458894a2336SHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 34591d098868SHong Zhang 34601d098868SHong Zhang /* forward solve the lower triangular */ 34611d098868SHong Zhang tmp[0] = b[*r++]; 34621d098868SHong Zhang tmps = tmp; 34631d098868SHong Zhang v = aa; 34641d098868SHong Zhang vi = aj; 34651d098868SHong Zhang for (i=1; i<n; i++) { 34661d098868SHong Zhang nz = ai[i+1] - ai[i]; 3467894a2336SHong Zhang sum = b[*r++]; 3468894a2336SHong Zhang PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 3469894a2336SHong Zhang tmp[i] = sum; 34701d098868SHong Zhang v += nz; vi += nz; 34711d098868SHong Zhang } 34721d098868SHong Zhang 34731d098868SHong Zhang /* backward solve the upper triangular */ 3474894a2336SHong Zhang k = n+1; 3475894a2336SHong Zhang v = aa + ai[k]; /* 1st entry of U(n-1,:) */ 3476894a2336SHong Zhang vi = aj + ai[k]; 34771d098868SHong Zhang for (i=n-1; i>=0; i--){ 3478894a2336SHong Zhang k = 2*n-i; 3479894a2336SHong Zhang nz = ai[k +1] - ai[k] - 1; 3480894a2336SHong Zhang sum = tmp[i]; 3481894a2336SHong Zhang PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 3482894a2336SHong Zhang x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */ 34831d098868SHong Zhang v += nz+1; vi += nz+1; 34841d098868SHong Zhang } 34851d098868SHong Zhang 34861d098868SHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 34871d098868SHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 34881d098868SHong Zhang ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 34891d098868SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 34901d098868SHong Zhang ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr); 34911d098868SHong Zhang PetscFunctionReturn(0); 34921d098868SHong Zhang } 34931d098868SHong Zhang 34941d098868SHong Zhang #undef __FUNCT__ 3495f268cba8SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_newdatastruct_v2" 3496f268cba8SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_newdatastruct_v2(Mat A,Vec bb,Vec xx) 3497f268cba8SShri Abhyankar { 3498f268cba8SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3499f268cba8SShri Abhyankar IS iscol = a->col,isrow = a->row; 3500f268cba8SShri Abhyankar PetscErrorCode ierr; 3501f268cba8SShri Abhyankar PetscInt i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,*adiag = a->diag,nz; 3502f268cba8SShri Abhyankar const PetscInt *rout,*cout,*r,*c; 3503f268cba8SShri Abhyankar PetscScalar *x,*tmp,sum; 3504f268cba8SShri Abhyankar const PetscScalar *b; 3505f268cba8SShri Abhyankar const MatScalar *aa = a->a,*v; 3506f268cba8SShri Abhyankar 3507f268cba8SShri Abhyankar PetscFunctionBegin; 3508f268cba8SShri Abhyankar if (!n) PetscFunctionReturn(0); 3509f268cba8SShri Abhyankar 3510f268cba8SShri Abhyankar ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 3511f268cba8SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3512f268cba8SShri Abhyankar tmp = a->solve_work; 3513f268cba8SShri Abhyankar 3514f268cba8SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 3515f268cba8SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 3516f268cba8SShri Abhyankar 3517f268cba8SShri Abhyankar /* forward solve the lower triangular */ 3518f268cba8SShri Abhyankar /* tmp[0] = b[*r++]; */ 3519f268cba8SShri Abhyankar tmp[0] = b[r[0]]; 3520f268cba8SShri Abhyankar v = aa; 3521f268cba8SShri Abhyankar vi = aj; 3522f268cba8SShri Abhyankar for (i=1; i<n; i++) { 3523f268cba8SShri Abhyankar nz = ai[i+1] - ai[i]; 3524f268cba8SShri Abhyankar sum = b[r[i]]; 3525f268cba8SShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 3526f268cba8SShri Abhyankar tmp[i] = sum; 3527f268cba8SShri Abhyankar v += nz; vi += nz; 3528f268cba8SShri Abhyankar } 3529f268cba8SShri Abhyankar 3530f268cba8SShri Abhyankar /* backward solve the upper triangular */ 3531f268cba8SShri Abhyankar /* v = aa + ai[k]; *//* 1st entry of U(n-1,:) */ 3532f268cba8SShri Abhyankar /* vi = aj + ai[k]; */ 3533f268cba8SShri Abhyankar v = aa + adiag[n-1]; 3534f268cba8SShri Abhyankar vi = aj + adiag[n-1]; 3535f268cba8SShri Abhyankar for (i=n-1; i>=0; i--){ 3536f268cba8SShri Abhyankar /* nz = ai[k +1] - ai[k] - 1;*/ 3537f268cba8SShri Abhyankar nz = adiag[i]-adiag[i+1]-1; 3538f268cba8SShri Abhyankar sum = tmp[i]; 3539f268cba8SShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 3540f268cba8SShri Abhyankar x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */ 3541f268cba8SShri Abhyankar v += nz+1; vi += nz+1; 3542f268cba8SShri Abhyankar } 3543f268cba8SShri Abhyankar 3544f268cba8SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 3545f268cba8SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 3546f268cba8SShri Abhyankar ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 3547f268cba8SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3548f268cba8SShri Abhyankar ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr); 3549f268cba8SShri Abhyankar PetscFunctionReturn(0); 3550f268cba8SShri Abhyankar } 3551f268cba8SShri Abhyankar 3552f268cba8SShri Abhyankar #undef __FUNCT__ 35531d098868SHong Zhang #define __FUNCT__ "MatILUDTFactor_SeqAIJ" 35541d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact) 35551d098868SHong Zhang { 35561d098868SHong Zhang Mat B = *fact; 3557599ef60dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b; 3558599ef60dSHong Zhang IS isicol; 3559599ef60dSHong Zhang PetscErrorCode ierr; 35601d098868SHong Zhang const PetscInt *r,*ic; 35611d098868SHong Zhang PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag; 3562dc3a2fd3SHong Zhang PetscInt *bi,*bj,*bdiag,*bdiag_rev; 3563f61a948aSLisandro Dalcin PetscInt row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au; 35641d098868SHong Zhang PetscInt nlnk,*lnk; 35651d098868SHong Zhang PetscBT lnkbt; 35661d098868SHong Zhang PetscTruth row_identity,icol_identity,both_identity; 35678aee2decSHong Zhang MatScalar *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp; 35681d098868SHong Zhang const PetscInt *ics; 35691d098868SHong Zhang PetscInt j,nz,*pj,*bjtmp,k,ncut,*jtmp; 3570f61a948aSLisandro Dalcin PetscReal dt=info->dt,dtcol=info->dtcol,shift=info->shiftinblocks; 3571f61a948aSLisandro Dalcin PetscInt dtcount=(PetscInt)info->dtcount,nnz_max; 3572599ef60dSHong Zhang PetscTruth missing; 3573599ef60dSHong Zhang 3574599ef60dSHong Zhang PetscFunctionBegin; 3575f61a948aSLisandro Dalcin 3576f61a948aSLisandro Dalcin if (dt == PETSC_DEFAULT) dt = 0.005; 3577f61a948aSLisandro Dalcin if (dtcol == PETSC_DEFAULT) dtcol = 0.01; /* XXX unused! */ 3578f61a948aSLisandro Dalcin if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax); 3579f61a948aSLisandro Dalcin 35801d098868SHong Zhang /* ------- symbolic factorization, can be reused ---------*/ 35811d098868SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 35821d098868SHong Zhang if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 35831d098868SHong Zhang adiag=a->diag; 3584599ef60dSHong Zhang 3585599ef60dSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 3586599ef60dSHong Zhang 3587599ef60dSHong Zhang /* bdiag is location of diagonal in factor */ 3588a6226a1aSHong Zhang ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 3589a6226a1aSHong Zhang bdiag_rev = bdiag + n+1; 3590599ef60dSHong Zhang 35911d098868SHong Zhang /* allocate row pointers bi */ 35928fc3a347SHong Zhang ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 35931d098868SHong Zhang 35941d098868SHong Zhang /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */ 3595393d3a68SHong Zhang if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */ 35961d098868SHong Zhang nnz_max = ai[n]+2*n*dtcount+2; 35978fc3a347SHong Zhang 35988fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 35998fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(MatScalar),&ba);CHKERRQ(ierr); 3600599ef60dSHong Zhang 3601599ef60dSHong Zhang /* put together the new matrix */ 3602599ef60dSHong Zhang ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 3603599ef60dSHong Zhang ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 3604599ef60dSHong Zhang b = (Mat_SeqAIJ*)(B)->data; 3605599ef60dSHong Zhang b->free_a = PETSC_TRUE; 3606599ef60dSHong Zhang b->free_ij = PETSC_TRUE; 3607599ef60dSHong Zhang b->singlemalloc = PETSC_FALSE; 3608599ef60dSHong Zhang b->a = ba; 3609599ef60dSHong Zhang b->j = bj; 3610599ef60dSHong Zhang b->i = bi; 3611599ef60dSHong Zhang b->diag = bdiag; 3612599ef60dSHong Zhang b->ilen = 0; 3613599ef60dSHong Zhang b->imax = 0; 3614599ef60dSHong Zhang b->row = isrow; 3615599ef60dSHong Zhang b->col = iscol; 3616599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 3617599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 3618599ef60dSHong Zhang b->icol = isicol; 3619599ef60dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 3620599ef60dSHong Zhang 36211d098868SHong Zhang ierr = PetscLogObjectMemory(B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 36221d098868SHong Zhang b->maxnz = nnz_max; 3623599ef60dSHong Zhang 3624599ef60dSHong Zhang (B)->factor = MAT_FACTOR_ILUDT; 3625599ef60dSHong Zhang (B)->info.factor_mallocs = 0; 36261d098868SHong Zhang (B)->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]); 36271d098868SHong Zhang CHKMEMQ; 36281d098868SHong Zhang /* ------- end of symbolic factorization ---------*/ 3629599ef60dSHong Zhang 3630599ef60dSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 3631599ef60dSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 3632599ef60dSHong Zhang ics = ic; 3633599ef60dSHong Zhang 3634599ef60dSHong Zhang /* linked list for storing column indices of the active row */ 3635599ef60dSHong Zhang nlnk = n + 1; 3636599ef60dSHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 36371d098868SHong Zhang 36381d098868SHong Zhang /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */ 36391d098868SHong Zhang ierr = PetscMalloc((2*n+1)*sizeof(PetscInt),&im);CHKERRQ(ierr); 36401d098868SHong Zhang jtmp = im + n; 36411d098868SHong Zhang /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */ 36428369b28dSHong Zhang ierr = PetscMalloc((2*n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 36438aee2decSHong Zhang ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 36448369b28dSHong Zhang vtmp = rtmp + n; 3645599ef60dSHong Zhang 3646599ef60dSHong Zhang bi[0] = 0; 36478fc3a347SHong Zhang bdiag[0] = nnz_max-1; /* location of diag[0] in factor B */ 3648dc3a2fd3SHong Zhang bdiag_rev[n] = bdiag[0]; 36498fc3a347SHong Zhang bi[2*n+1] = bdiag[0]+1; /* endof bj and ba array */ 3650599ef60dSHong Zhang for (i=0; i<n; i++) { 3651599ef60dSHong Zhang /* copy initial fill into linked list */ 3652599ef60dSHong Zhang nzi = 0; /* nonzeros for active row i */ 36538369b28dSHong Zhang nzi = ai[r[i]+1] - ai[r[i]]; 36548369b28dSHong Zhang if (!nzi) SETERRQ2(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 36558369b28dSHong Zhang nzi_al = adiag[r[i]] - ai[r[i]]; 36568369b28dSHong Zhang nzi_au = ai[r[i]+1] - adiag[r[i]] -1; 3657599ef60dSHong Zhang ajtmp = aj + ai[r[i]]; 36588369b28dSHong Zhang ierr = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3659599ef60dSHong Zhang 3660599ef60dSHong Zhang /* load in initial (unfactored row) */ 36611d098868SHong Zhang aatmp = a->a + ai[r[i]]; 36628369b28dSHong Zhang for (j=0; j<nzi; j++) { 36631d098868SHong Zhang rtmp[ics[*ajtmp++]] = *aatmp++; 3664599ef60dSHong Zhang } 3665599ef60dSHong Zhang 3666599ef60dSHong Zhang /* add pivot rows into linked list */ 3667599ef60dSHong Zhang row = lnk[n]; 3668599ef60dSHong Zhang while (row < i ) { 36691d098868SHong Zhang nzi_bl = bi[row+1] - bi[row] + 1; 36701d098868SHong Zhang bjtmp = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */ 36711d098868SHong Zhang ierr = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr); 3672599ef60dSHong Zhang nzi += nlnk; 3673599ef60dSHong Zhang row = lnk[row]; 3674599ef60dSHong Zhang } 3675599ef60dSHong Zhang 36768369b28dSHong Zhang /* copy data from lnk into jtmp, then initialize lnk */ 36778369b28dSHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr); 3678599ef60dSHong Zhang 3679599ef60dSHong Zhang /* numerical factorization */ 36808369b28dSHong Zhang bjtmp = jtmp; 3681599ef60dSHong Zhang row = *bjtmp++; /* 1st pivot row */ 3682599ef60dSHong Zhang while ( row < i ) { 3683599ef60dSHong Zhang pc = rtmp + row; 36843c2a7987SHong Zhang pv = ba + bdiag[row]; /* 1./(diag of the pivot row) */ 36853c2a7987SHong Zhang multiplier = (*pc) * (*pv); 3686599ef60dSHong Zhang *pc = multiplier; 36873c2a7987SHong Zhang if (PetscAbsScalar(*pc) > dt){ /* apply tolerance dropping rule */ 36881d098868SHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 36891d098868SHong Zhang pv = ba + bdiag[row+1] + 1; 36901d098868SHong Zhang /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */ 36911d098868SHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 36921d098868SHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 3693dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 3694599ef60dSHong Zhang } 3695599ef60dSHong Zhang row = *bjtmp++; 3696599ef60dSHong Zhang } 3697599ef60dSHong Zhang 36988369b28dSHong Zhang /* copy sparse rtmp into contiguous vtmp; separate L and U part */ 36998aee2decSHong Zhang diag_tmp = rtmp[i]; /* save diagonal value - may not needed?? */ 37008369b28dSHong Zhang nzi_bl = 0; j = 0; 37018369b28dSHong Zhang while (jtmp[j] < i){ /* Note: jtmp is sorted */ 37028aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 37038369b28dSHong Zhang nzi_bl++; j++; 37048369b28dSHong Zhang } 37058369b28dSHong Zhang nzi_bu = nzi - nzi_bl -1; 37068369b28dSHong Zhang while (j < nzi){ 37078aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 37088369b28dSHong Zhang j++; 37098369b28dSHong Zhang } 37108369b28dSHong Zhang 3711599ef60dSHong Zhang bjtmp = bj + bi[i]; 3712599ef60dSHong Zhang batmp = ba + bi[i]; 37138369b28dSHong Zhang /* apply level dropping rule to L part */ 37148369b28dSHong Zhang ncut = nzi_al + dtcount; 37158369b28dSHong Zhang if (ncut < nzi_bl){ 37168369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr); 37178369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr); 3718599ef60dSHong Zhang } else { 37198369b28dSHong Zhang ncut = nzi_bl; 37208369b28dSHong Zhang } 37218369b28dSHong Zhang for (j=0; j<ncut; j++){ 37228369b28dSHong Zhang bjtmp[j] = jtmp[j]; 37238369b28dSHong Zhang batmp[j] = vtmp[j]; 37246da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */ 37258369b28dSHong Zhang } 37261d098868SHong Zhang bi[i+1] = bi[i] + ncut; 37278369b28dSHong Zhang nzi = ncut + 1; 37288369b28dSHong Zhang 37298369b28dSHong Zhang /* apply level dropping rule to U part */ 37308369b28dSHong Zhang ncut = nzi_au + dtcount; 37318369b28dSHong Zhang if (ncut < nzi_bu){ 37328369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr); 37338369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr); 37348369b28dSHong Zhang } else { 37358369b28dSHong Zhang ncut = nzi_bu; 37368369b28dSHong Zhang } 37378369b28dSHong Zhang nzi += ncut; 37381d098868SHong Zhang 37391d098868SHong Zhang /* mark bdiagonal */ 37401d098868SHong Zhang bdiag[i+1] = bdiag[i] - (ncut + 1); 3741dc3a2fd3SHong Zhang bdiag_rev[n-i-1] = bdiag[i+1]; 37428fc3a347SHong Zhang bi[2*n - i] = bi[2*n - i +1] - (ncut + 1); 37431d098868SHong Zhang bjtmp = bj + bdiag[i]; 37441d098868SHong Zhang batmp = ba + bdiag[i]; 37451d098868SHong Zhang *bjtmp = i; 37468aee2decSHong Zhang *batmp = diag_tmp; /* rtmp[i]; */ 3747c9c72f8fSHong Zhang if (*batmp == 0.0) { 3748c9c72f8fSHong Zhang *batmp = dt+shift; 3749bd1bc851SBarry Smith /* printf(" row %d add shift %g\n",i,shift); */ 3750c9c72f8fSHong Zhang } 3751b78a477dSHong Zhang *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */ 37526da515a1SHong Zhang /* printf(" (%d,%g),",*bjtmp,*batmp); */ 37531d098868SHong Zhang 37541d098868SHong Zhang bjtmp = bj + bdiag[i+1]+1; 37551d098868SHong Zhang batmp = ba + bdiag[i+1]+1; 37568369b28dSHong Zhang for (k=0; k<ncut; k++){ 37571d098868SHong Zhang bjtmp[k] = jtmp[nzi_bl+1+k]; 37581d098868SHong Zhang batmp[k] = vtmp[nzi_bl+1+k]; 37596da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */ 3760599ef60dSHong Zhang } 37616da515a1SHong Zhang /* printf("\n"); */ 3762599ef60dSHong Zhang 3763599ef60dSHong Zhang im[i] = nzi; /* used by PetscLLAddSortedLU() */ 3764599ef60dSHong Zhang /* 37651d098868SHong Zhang printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]); 37668369b28dSHong Zhang printf(" ----------------------------\n"); 3767599ef60dSHong Zhang */ 37688369b28dSHong Zhang } /* for (i=0; i<n; i++) */ 37691d098868SHong Zhang /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */ 37701d098868SHong Zhang if (bi[n] >= bdiag[n]) SETERRQ2(PETSC_ERR_ARG_SIZ,"end of L array %d cannot >= the beginning of U array %d",bi[n],bdiag[n]); 3771599ef60dSHong Zhang 3772599ef60dSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3773599ef60dSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3774599ef60dSHong Zhang 3775599ef60dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 3776599ef60dSHong Zhang ierr = PetscFree(im);CHKERRQ(ierr); 3777599ef60dSHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 3778599ef60dSHong Zhang 3779599ef60dSHong Zhang ierr = PetscLogFlops(B->cmap->n);CHKERRQ(ierr); 37801d098868SHong Zhang b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n]; 3781599ef60dSHong Zhang 3782599ef60dSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 3783599ef60dSHong Zhang ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr); 3784599ef60dSHong Zhang both_identity = (PetscTruth) (row_identity && icol_identity); 3785599ef60dSHong Zhang if (row_identity && icol_identity) { 3786894a2336SHong Zhang B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct; 3787599ef60dSHong Zhang } else { 3788894a2336SHong Zhang B->ops->solve = MatSolve_SeqAIJ_newdatastruct; 3789599ef60dSHong Zhang } 3790599ef60dSHong Zhang 3791b78a477dSHong Zhang B->ops->lufactorsymbolic = MatILUDTFactorSymbolic_SeqAIJ; 3792b78a477dSHong Zhang B->ops->lufactornumeric = MatILUDTFactorNumeric_SeqAIJ; 37931d098868SHong Zhang B->ops->solveadd = 0; 37941d098868SHong Zhang B->ops->solvetranspose = 0; 37951d098868SHong Zhang B->ops->solvetransposeadd = 0; 37961d098868SHong Zhang B->ops->matsolve = 0; 3797599ef60dSHong Zhang B->assembled = PETSC_TRUE; 3798599ef60dSHong Zhang B->preallocated = PETSC_TRUE; 3799599ef60dSHong Zhang PetscFunctionReturn(0); 3800599ef60dSHong Zhang } 3801599ef60dSHong Zhang 38023c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */ 38033c2a7987SHong Zhang #undef __FUNCT__ 38043c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorSymbolic_SeqAIJ" 38053c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info) 38063c2a7987SHong Zhang { 38073c2a7987SHong Zhang PetscErrorCode ierr; 38083c2a7987SHong Zhang 38093c2a7987SHong Zhang PetscFunctionBegin; 38103c2a7987SHong Zhang ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr); 38113c2a7987SHong Zhang 38123c2a7987SHong Zhang fact->ops->lufactornumeric = MatILUDTFactorNumeric_SeqAIJ; 38133c2a7987SHong Zhang PetscFunctionReturn(0); 38143c2a7987SHong Zhang } 38153c2a7987SHong Zhang 38163c2a7987SHong Zhang /* 38173c2a7987SHong Zhang same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors 38183c2a7987SHong Zhang - intend to replace existing MatLUFactorNumeric_SeqAIJ() 38193c2a7987SHong Zhang */ 38203c2a7987SHong Zhang #undef __FUNCT__ 38213c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorNumeric_SeqAIJ" 38223c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info) 38233c2a7987SHong Zhang { 38243c2a7987SHong Zhang Mat C=fact; 38253c2a7987SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 38263c2a7987SHong Zhang IS isrow = b->row,isicol = b->icol; 38273c2a7987SHong Zhang PetscErrorCode ierr; 38283c2a7987SHong Zhang const PetscInt *r,*ic,*ics; 3829b78a477dSHong Zhang PetscInt i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 3830b78a477dSHong Zhang PetscInt *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj; 38313c2a7987SHong Zhang MatScalar *rtmp,*pc,multiplier,*v,*pv,*aa=a->a; 3832b78a477dSHong Zhang PetscReal dt=info->dt,shift=info->shiftinblocks; 3833b78a477dSHong Zhang PetscTruth row_identity, col_identity; 38343c2a7987SHong Zhang 38353c2a7987SHong Zhang PetscFunctionBegin; 38363c2a7987SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 38373c2a7987SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 3838b78a477dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 38393c2a7987SHong Zhang ics = ic; 38403c2a7987SHong Zhang 38413c2a7987SHong Zhang for (i=0; i<n; i++){ 3842b78a477dSHong Zhang /* initialize rtmp array */ 3843b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of nozeros in L(i,:) */ 38443c2a7987SHong Zhang bjtmp = bj + bi[i]; 3845b78a477dSHong Zhang for (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0; 3846b78a477dSHong Zhang rtmp[i] = 0.0; 3847b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */ 3848b78a477dSHong Zhang bjtmp = bj + bdiag[i+1] + 1; 3849b78a477dSHong Zhang for (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0; 38503c2a7987SHong Zhang 3851b78a477dSHong Zhang /* load in initial unfactored row of A */ 38526da515a1SHong Zhang /* printf("row %d\n",i); */ 38533c2a7987SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 38543c2a7987SHong Zhang ajtmp = aj + ai[r[i]]; 38553c2a7987SHong Zhang v = aa + ai[r[i]]; 38563c2a7987SHong Zhang for (j=0; j<nz; j++) { 3857b78a477dSHong Zhang rtmp[ics[*ajtmp++]] = v[j]; 38586da515a1SHong Zhang /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */ 38593c2a7987SHong Zhang } 38606da515a1SHong Zhang /* printf("\n"); */ 38613c2a7987SHong Zhang 3862b78a477dSHong Zhang /* numerical factorization */ 3863b78a477dSHong Zhang bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */ 3864b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of entries in L(i,:) */ 3865b78a477dSHong Zhang k = 0; 3866b78a477dSHong Zhang while (k < nzl){ 38673c2a7987SHong Zhang row = *bjtmp++; 38686da515a1SHong Zhang /* printf(" prow %d\n",row); */ 38693c2a7987SHong Zhang pc = rtmp + row; 3870b78a477dSHong Zhang pv = b->a + bdiag[row]; /* 1./(diag of the pivot row) */ 3871b78a477dSHong Zhang multiplier = (*pc) * (*pv); 38723c2a7987SHong Zhang *pc = multiplier; 3873b78a477dSHong Zhang if (PetscAbsScalar(multiplier) > dt){ 3874b78a477dSHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 3875b78a477dSHong Zhang pv = b->a + bdiag[row+1] + 1; 3876b78a477dSHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 3877b78a477dSHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 38786da515a1SHong Zhang /* ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); */ 38793c2a7987SHong Zhang } 3880b78a477dSHong Zhang k++; 38813c2a7987SHong Zhang } 38823c2a7987SHong Zhang 3883b78a477dSHong Zhang /* finished row so stick it into b->a */ 3884b78a477dSHong Zhang /* L-part */ 3885b78a477dSHong Zhang pv = b->a + bi[i] ; 3886b78a477dSHong Zhang pj = bj + bi[i] ; 3887b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; 3888b78a477dSHong Zhang for (j=0; j<nzl; j++) { 3889b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 38906da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 38913c2a7987SHong Zhang } 3892b78a477dSHong Zhang 3893b78a477dSHong Zhang /* diagonal: invert diagonal entries for simplier triangular solves */ 3894b78a477dSHong Zhang if (rtmp[i] == 0.0) rtmp[i] = dt+shift; 3895b78a477dSHong Zhang b->a[bdiag[i]] = 1.0/rtmp[i]; 38966da515a1SHong Zhang /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */ 3897b78a477dSHong Zhang 3898b78a477dSHong Zhang /* U-part */ 3899b78a477dSHong Zhang pv = b->a + bdiag[i+1] + 1; 3900b78a477dSHong Zhang pj = bj + bdiag[i+1] + 1; 3901b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1] - 1; 3902b78a477dSHong Zhang for (j=0; j<nzu; j++) { 3903b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 39046da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 3905b78a477dSHong Zhang } 39066da515a1SHong Zhang /* printf("\n"); */ 3907b78a477dSHong Zhang } 3908b78a477dSHong Zhang 39093c2a7987SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 39103c2a7987SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 39113c2a7987SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3912b78a477dSHong Zhang 39133c2a7987SHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 39143c2a7987SHong Zhang ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 39153c2a7987SHong Zhang if (row_identity && col_identity) { 3916894a2336SHong Zhang C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct; 39173c2a7987SHong Zhang } else { 3918894a2336SHong Zhang C->ops->solve = MatSolve_SeqAIJ_newdatastruct; 39193c2a7987SHong Zhang } 3920b78a477dSHong Zhang C->ops->solveadd = 0; 3921b78a477dSHong Zhang C->ops->solvetranspose = 0; 3922b78a477dSHong Zhang C->ops->solvetransposeadd = 0; 3923b78a477dSHong Zhang C->ops->matsolve = 0; 39243c2a7987SHong Zhang C->assembled = PETSC_TRUE; 39253c2a7987SHong Zhang C->preallocated = PETSC_TRUE; 39263c2a7987SHong Zhang ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 39273c2a7987SHong Zhang PetscFunctionReturn(0); 39283c2a7987SHong Zhang } 3929