1be1d678aSKris Buschelman #define PETSCMAT_DLL 2289bc588SBarry Smith 3b377110cSBarry Smith 47c4f633dSBarry Smith #include "../src/mat/impls/aij/seq/aij.h" 51393bc97SHong Zhang #include "petscbt.h" 67c4f633dSBarry Smith #include "../src/mat/utils/freespace.h" 73b2fbd54SBarry Smith 8f6a26d55SMatthew Knepley EXTERN_C_BEGIN 94a2ae208SSatish Balay #undef __FUNCT__ 104a2ae208SSatish Balay #define __FUNCT__ "MatOrdering_Flow_SeqAIJ" 118fa24674SBarry Smith /* 128fa24674SBarry Smith Computes an ordering to get most of the large numerical values in the lower triangular part of the matrix 138fa24674SBarry Smith */ 14dfbe8321SBarry Smith PetscErrorCode MatOrdering_Flow_SeqAIJ(Mat mat,const MatOrderingType type,IS *irow,IS *icol) 153b2fbd54SBarry Smith { 168fa24674SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)mat->data; 178fa24674SBarry Smith PetscErrorCode ierr; 18889a8dedSBarry Smith PetscInt i,j,jj,k, kk,n = mat->rmap->n, current = 0, newcurrent = 0,*order; 198fa24674SBarry Smith const PetscInt *ai = a->i, *aj = a->j; 208fa24674SBarry Smith const PetscScalar *aa = a->a; 218fa24674SBarry Smith PetscTruth *done; 22889a8dedSBarry Smith PetscReal best,past = 0,future; 233a40ed3dSBarry Smith 248fa24674SBarry Smith PetscFunctionBegin; 258fa24674SBarry Smith /* pick initial row */ 268fa24674SBarry Smith best = -1; 278fa24674SBarry Smith for (i=0; i<n; i++) { 288fa24674SBarry Smith future = 0; 298fa24674SBarry Smith for (j=ai[i]; j<ai[i+1]; j++) { 308fa24674SBarry Smith if (aj[j] != i) future += PetscAbsScalar(aa[j]); else past = PetscAbsScalar(aa[j]); 318fa24674SBarry Smith } 328fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 338fa24674SBarry Smith if (past/future > best) { 348fa24674SBarry Smith best = past/future; 358fa24674SBarry Smith current = i; 368fa24674SBarry Smith } 378fa24674SBarry Smith } 388fa24674SBarry Smith 398fa24674SBarry Smith ierr = PetscMalloc(n*sizeof(PetscTruth),&done);CHKERRQ(ierr); 408fa24674SBarry Smith ierr = PetscMalloc(n*sizeof(PetscInt),&order);CHKERRQ(ierr); 418fa24674SBarry Smith ierr = PetscMemzero(done,n*sizeof(PetscTruth));CHKERRQ(ierr); 428fa24674SBarry Smith order[0] = current; 438fa24674SBarry Smith for (i=0; i<n-1; i++) { 448fa24674SBarry Smith done[current] = PETSC_TRUE; 458fa24674SBarry Smith best = -1; 468fa24674SBarry Smith /* loop over all neighbors of current pivot */ 478fa24674SBarry Smith for (j=ai[current]; j<ai[current+1]; j++) { 488fa24674SBarry Smith jj = aj[j]; 498fa24674SBarry Smith if (done[jj]) continue; 508fa24674SBarry Smith /* loop over columns of potential next row computing weights for below and above diagonal */ 518fa24674SBarry Smith past = future = 0.0; 528fa24674SBarry Smith for (k=ai[jj]; k<ai[jj+1]; k++) { 538fa24674SBarry Smith kk = aj[k]; 548fa24674SBarry Smith if (done[kk]) past += PetscAbsScalar(aa[k]); 558fa24674SBarry Smith else if (kk != jj) future += PetscAbsScalar(aa[k]); 568fa24674SBarry Smith } 578fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 588fa24674SBarry Smith if (past/future > best) { 598fa24674SBarry Smith best = past/future; 608fa24674SBarry Smith newcurrent = jj; 618fa24674SBarry Smith } 628fa24674SBarry Smith } 638fa24674SBarry Smith if (best == -1) { /* no neighbors to select from so select best of all that remain */ 648fa24674SBarry Smith best = -1; 658fa24674SBarry Smith for (k=0; k<n; k++) { 668fa24674SBarry Smith if (done[k]) continue; 678fa24674SBarry Smith future = 0; 688fa24674SBarry Smith past = 0; 698fa24674SBarry Smith for (j=ai[k]; j<ai[k+1]; j++) { 708fa24674SBarry Smith kk = aj[j]; 718fa24674SBarry Smith if (done[kk]) past += PetscAbsScalar(aa[j]); 728fa24674SBarry Smith else if (kk != k) future += PetscAbsScalar(aa[j]); 738fa24674SBarry Smith } 748fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 758fa24674SBarry Smith if (past/future > best) { 768fa24674SBarry Smith best = past/future; 778fa24674SBarry Smith newcurrent = k; 788fa24674SBarry Smith } 798fa24674SBarry Smith } 808fa24674SBarry Smith } 818fa24674SBarry Smith if (current == newcurrent) SETERRQ(PETSC_ERR_PLIB,"newcurrent cannot be current"); 828fa24674SBarry Smith current = newcurrent; 838fa24674SBarry Smith order[i+1] = current; 848fa24674SBarry Smith } 858fa24674SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,n,order,irow);CHKERRQ(ierr); 868fa24674SBarry Smith *icol = *irow; 878fa24674SBarry Smith ierr = PetscObjectReference((PetscObject)*irow);CHKERRQ(ierr); 888fa24674SBarry Smith ierr = PetscFree(done);CHKERRQ(ierr); 898fa24674SBarry Smith ierr = PetscFree(order);CHKERRQ(ierr); 90d64ed03dSBarry Smith PetscFunctionReturn(0); 913b2fbd54SBarry Smith } 92f6a26d55SMatthew Knepley EXTERN_C_END 933b2fbd54SBarry Smith 94e631078cSBarry Smith EXTERN_C_BEGIN 954a2ae208SSatish Balay #undef __FUNCT__ 96db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqaij_petsc" 97db4efbfdSBarry Smith PetscErrorCode MatGetFactorAvailable_seqaij_petsc(Mat A,MatFactorType ftype,PetscTruth *flg) 98db4efbfdSBarry Smith { 99db4efbfdSBarry Smith PetscFunctionBegin; 100db4efbfdSBarry Smith *flg = PETSC_TRUE; 101db4efbfdSBarry Smith PetscFunctionReturn(0); 102db4efbfdSBarry Smith } 103db4efbfdSBarry Smith EXTERN_C_END 104db4efbfdSBarry Smith 105db4efbfdSBarry Smith EXTERN_C_BEGIN 106db4efbfdSBarry Smith #undef __FUNCT__ 107b24902e0SBarry Smith #define __FUNCT__ "MatGetFactor_seqaij_petsc" 108b24902e0SBarry Smith PetscErrorCode MatGetFactor_seqaij_petsc(Mat A,MatFactorType ftype,Mat *B) 109b24902e0SBarry Smith { 110d0f46423SBarry Smith PetscInt n = A->rmap->n; 111b24902e0SBarry Smith PetscErrorCode ierr; 112b24902e0SBarry Smith 113b24902e0SBarry Smith PetscFunctionBegin; 114b24902e0SBarry Smith ierr = MatCreate(((PetscObject)A)->comm,B);CHKERRQ(ierr); 115b24902e0SBarry Smith ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr); 116599ef60dSHong Zhang if (ftype == MAT_FACTOR_LU || ftype == MAT_FACTOR_ILU || ftype == MAT_FACTOR_ILUDT){ 117b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQAIJ);CHKERRQ(ierr); 118db4efbfdSBarry Smith (*B)->ops->ilufactorsymbolic = MatILUFactorSymbolic_SeqAIJ; 11935bd34faSBarry Smith (*B)->ops->lufactorsymbolic = MatLUFactorSymbolic_SeqAIJ; 1201da05e5aSHong Zhang (*B)->ops->iludtfactor = MatILUDTFactor_SeqAIJ; 121b24902e0SBarry Smith } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 122b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr); 123b24902e0SBarry Smith ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 1245c9eb25fSBarry Smith (*B)->ops->iccfactorsymbolic = MatICCFactorSymbolic_SeqAIJ; 1255c9eb25fSBarry Smith (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ; 126b24902e0SBarry Smith } else SETERRQ(PETSC_ERR_SUP,"Factor type not supported"); 127db4efbfdSBarry Smith (*B)->factor = ftype; 128b24902e0SBarry Smith PetscFunctionReturn(0); 129b24902e0SBarry Smith } 130e631078cSBarry Smith EXTERN_C_END 131b24902e0SBarry Smith 132b24902e0SBarry Smith #undef __FUNCT__ 133b9617806SBarry Smith #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ" 1340481f469SBarry Smith PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 135289bc588SBarry Smith { 136416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 137289bc588SBarry Smith IS isicol; 1386849ba73SBarry Smith PetscErrorCode ierr; 1395d0c19d7SBarry Smith const PetscInt *r,*ic; 1405d0c19d7SBarry Smith PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 1411393bc97SHong Zhang PetscInt *bi,*bj,*ajtmp; 1421393bc97SHong Zhang PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 1439e046878SBarry Smith PetscReal f; 1444875ba38SHong Zhang PetscInt nlnk,*lnk,k,**bi_ptr; 145a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1461393bc97SHong Zhang PetscBT lnkbt; 147*ce3d78c0SShri Abhyankar PetscTruth newdatastruct=PETSC_FALSE; 148289bc588SBarry Smith 1493a40ed3dSBarry Smith PetscFunctionBegin; 150*ce3d78c0SShri Abhyankar ierr = PetscOptionsGetTruth(PETSC_NULL,"-lu_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr); 151*ce3d78c0SShri Abhyankar if(newdatastruct){ 152*ce3d78c0SShri Abhyankar ierr = MatLUFactorSymbolic_SeqAIJ_newdatastruct(B,A,isrow,iscol,info);CHKERRQ(ierr); 153*ce3d78c0SShri Abhyankar PetscFunctionReturn(0); 154*ce3d78c0SShri Abhyankar } 155*ce3d78c0SShri Abhyankar 156d0f46423SBarry Smith if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square"); 1574c49b128SBarry Smith ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 158ac121b3dSBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 159ac121b3dSBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 160289bc588SBarry Smith 161289bc588SBarry Smith /* get new row pointers */ 1621393bc97SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 1631393bc97SHong Zhang bi[0] = 0; 1641393bc97SHong Zhang 1651393bc97SHong Zhang /* bdiag is location of diagonal in factor */ 1661393bc97SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 1671393bc97SHong Zhang bdiag[0] = 0; 1681393bc97SHong Zhang 1691393bc97SHong Zhang /* linked list for storing column indices of the active row */ 1701393bc97SHong Zhang nlnk = n + 1; 1711393bc97SHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1721393bc97SHong Zhang 17335baf27eSSatish Balay ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 1741393bc97SHong Zhang 1751393bc97SHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 176d3d32019SBarry Smith f = info->fill; 177a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 1781393bc97SHong Zhang current_space = free_space; 179289bc588SBarry Smith 180289bc588SBarry Smith for (i=0; i<n; i++) { 1811393bc97SHong Zhang /* copy previous fill into linked list */ 182289bc588SBarry Smith nzi = 0; 1831393bc97SHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 1843b4a8b6dSBarry 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); 1851393bc97SHong Zhang ajtmp = aj + ai[r[i]]; 186afcc9904SHong Zhang ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1871393bc97SHong Zhang nzi += nlnk; 1881393bc97SHong Zhang 1891393bc97SHong Zhang /* add pivot rows into linked list */ 1901393bc97SHong Zhang row = lnk[n]; 1911393bc97SHong Zhang while (row < i) { 192d1170560SHong Zhang nzbd = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */ 193d1170560SHong Zhang ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 194d1170560SHong Zhang ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 1951393bc97SHong Zhang nzi += nlnk; 1961393bc97SHong Zhang row = lnk[row]; 197289bc588SBarry Smith } 1981393bc97SHong Zhang bi[i+1] = bi[i] + nzi; 1991393bc97SHong Zhang im[i] = nzi; 2001393bc97SHong Zhang 2011393bc97SHong Zhang /* mark bdiag */ 2021393bc97SHong Zhang nzbd = 0; 2031393bc97SHong Zhang nnz = nzi; 2041393bc97SHong Zhang k = lnk[n]; 2051393bc97SHong Zhang while (nnz-- && k < i){ 2061393bc97SHong Zhang nzbd++; 2071393bc97SHong Zhang k = lnk[k]; 2081393bc97SHong Zhang } 2091393bc97SHong Zhang bdiag[i] = bi[i] + nzbd; 2101393bc97SHong Zhang 2111393bc97SHong Zhang /* if free space is not available, make more free space */ 2121393bc97SHong Zhang if (current_space->local_remaining<nzi) { 2131393bc97SHong Zhang nnz = (n - i)*nzi; /* estimated and max additional space needed */ 214a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 2151393bc97SHong Zhang reallocs++; 2161393bc97SHong Zhang } 2171393bc97SHong Zhang 2181393bc97SHong Zhang /* copy data into free space, then initialize lnk */ 2191393bc97SHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 2201393bc97SHong Zhang bi_ptr[i] = current_space->array; 2211393bc97SHong Zhang current_space->array += nzi; 2221393bc97SHong Zhang current_space->local_used += nzi; 2231393bc97SHong Zhang current_space->local_remaining -= nzi; 224289bc588SBarry Smith } 2256cf91177SBarry Smith #if defined(PETSC_USE_INFO) 226464e8d44SSatish Balay if (ai[n] != 0) { 2271393bc97SHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 228ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 229ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 230ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 231ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 23205bf559cSSatish Balay } else { 233ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 23405bf559cSSatish Balay } 23563ba0a88SBarry Smith #endif 2362fb3ed76SBarry Smith 237898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 238898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 2391987afe7SBarry Smith 2401393bc97SHong Zhang /* destroy list of free space and other temporary array(s) */ 2411393bc97SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 242a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); 2431393bc97SHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 24435baf27eSSatish Balay ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 245289bc588SBarry Smith 246289bc588SBarry Smith /* put together the new matrix */ 247719d5645SBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 248719d5645SBarry Smith ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 249719d5645SBarry Smith b = (Mat_SeqAIJ*)(B)->data; 250e6b907acSBarry Smith b->free_a = PETSC_TRUE; 251e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 2527c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 2531393bc97SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 2541393bc97SHong Zhang b->j = bj; 2551393bc97SHong Zhang b->i = bi; 2561393bc97SHong Zhang b->diag = bdiag; 257416022c9SBarry Smith b->ilen = 0; 258416022c9SBarry Smith b->imax = 0; 259416022c9SBarry Smith b->row = isrow; 260416022c9SBarry Smith b->col = iscol; 261c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 262c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 26382bf6240SBarry Smith b->icol = isicol; 26487828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2651393bc97SHong Zhang 2661393bc97SHong Zhang /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 267719d5645SBarry Smith ierr = PetscLogObjectMemory(B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 2681393bc97SHong Zhang b->maxnz = b->nz = bi[n] ; 2697fd98bd6SLois Curfman McInnes 270719d5645SBarry Smith (B)->factor = MAT_FACTOR_LU; 271719d5645SBarry Smith (B)->info.factor_mallocs = reallocs; 272719d5645SBarry Smith (B)->info.fill_ratio_given = f; 273703deb49SSatish Balay 2749f7d0b68SHong Zhang if (ai[n]) { 275719d5645SBarry Smith (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 276eea2913aSSatish Balay } else { 277719d5645SBarry Smith (B)->info.fill_ratio_needed = 0.0; 278eea2913aSSatish Balay } 279719d5645SBarry Smith (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 280719d5645SBarry Smith (B)->ops->solve = MatSolve_SeqAIJ; 281719d5645SBarry Smith (B)->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 282db4efbfdSBarry Smith /* switch to inodes if appropriate */ 283719d5645SBarry Smith ierr = MatLUFactorSymbolic_Inode(B,A,isrow,iscol,info);CHKERRQ(ierr); 2843a40ed3dSBarry Smith PetscFunctionReturn(0); 285289bc588SBarry Smith } 2861393bc97SHong Zhang 287*ce3d78c0SShri Abhyankar #undef __FUNCT__ 288*ce3d78c0SShri Abhyankar #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ_newdatastruct" 289*ce3d78c0SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ_newdatastruct(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 290*ce3d78c0SShri Abhyankar { 291*ce3d78c0SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 292*ce3d78c0SShri Abhyankar IS isicol; 293*ce3d78c0SShri Abhyankar PetscErrorCode ierr; 294*ce3d78c0SShri Abhyankar const PetscInt *r,*ic; 295*ce3d78c0SShri Abhyankar PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 296*ce3d78c0SShri Abhyankar PetscInt *bi,*bj,*ajtmp; 297*ce3d78c0SShri Abhyankar PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 298*ce3d78c0SShri Abhyankar PetscReal f; 299*ce3d78c0SShri Abhyankar PetscInt nlnk,*lnk,k,**bi_ptr; 300*ce3d78c0SShri Abhyankar PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 301*ce3d78c0SShri Abhyankar PetscBT lnkbt; 302*ce3d78c0SShri Abhyankar 303*ce3d78c0SShri Abhyankar PetscFunctionBegin; 304*ce3d78c0SShri Abhyankar if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square"); 305*ce3d78c0SShri Abhyankar ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 306*ce3d78c0SShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 307*ce3d78c0SShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 308*ce3d78c0SShri Abhyankar 309*ce3d78c0SShri Abhyankar /* get new row pointers */ 310*ce3d78c0SShri Abhyankar ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 311*ce3d78c0SShri Abhyankar bi[0] = 0; 312*ce3d78c0SShri Abhyankar 313*ce3d78c0SShri Abhyankar /* bdiag is location of diagonal in factor */ 314*ce3d78c0SShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 315*ce3d78c0SShri Abhyankar bdiag[0] = 0; 316*ce3d78c0SShri Abhyankar 317*ce3d78c0SShri Abhyankar /* linked list for storing column indices of the active row */ 318*ce3d78c0SShri Abhyankar nlnk = n + 1; 319*ce3d78c0SShri Abhyankar ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 320*ce3d78c0SShri Abhyankar 321*ce3d78c0SShri Abhyankar ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 322*ce3d78c0SShri Abhyankar 323*ce3d78c0SShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 324*ce3d78c0SShri Abhyankar f = info->fill; 325*ce3d78c0SShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 326*ce3d78c0SShri Abhyankar current_space = free_space; 327*ce3d78c0SShri Abhyankar 328*ce3d78c0SShri Abhyankar for (i=0; i<n; i++) { 329*ce3d78c0SShri Abhyankar /* copy previous fill into linked list */ 330*ce3d78c0SShri Abhyankar nzi = 0; 331*ce3d78c0SShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 332*ce3d78c0SShri 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); 333*ce3d78c0SShri Abhyankar ajtmp = aj + ai[r[i]]; 334*ce3d78c0SShri Abhyankar ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 335*ce3d78c0SShri Abhyankar nzi += nlnk; 336*ce3d78c0SShri Abhyankar 337*ce3d78c0SShri Abhyankar /* add pivot rows into linked list */ 338*ce3d78c0SShri Abhyankar row = lnk[n]; 339*ce3d78c0SShri Abhyankar while (row < i) { 340*ce3d78c0SShri Abhyankar nzbd = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */ 341*ce3d78c0SShri Abhyankar ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 342*ce3d78c0SShri Abhyankar ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 343*ce3d78c0SShri Abhyankar nzi += nlnk; 344*ce3d78c0SShri Abhyankar row = lnk[row]; 345*ce3d78c0SShri Abhyankar } 346*ce3d78c0SShri Abhyankar bi[i+1] = bi[i] + nzi; 347*ce3d78c0SShri Abhyankar im[i] = nzi; 348*ce3d78c0SShri Abhyankar 349*ce3d78c0SShri Abhyankar /* mark bdiag */ 350*ce3d78c0SShri Abhyankar nzbd = 0; 351*ce3d78c0SShri Abhyankar nnz = nzi; 352*ce3d78c0SShri Abhyankar k = lnk[n]; 353*ce3d78c0SShri Abhyankar while (nnz-- && k < i){ 354*ce3d78c0SShri Abhyankar nzbd++; 355*ce3d78c0SShri Abhyankar k = lnk[k]; 356*ce3d78c0SShri Abhyankar } 357*ce3d78c0SShri Abhyankar bdiag[i] = bi[i] + nzbd; 358*ce3d78c0SShri Abhyankar 359*ce3d78c0SShri Abhyankar /* if free space is not available, make more free space */ 360*ce3d78c0SShri Abhyankar if (current_space->local_remaining<nzi) { 361*ce3d78c0SShri Abhyankar nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */ 362*ce3d78c0SShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 363*ce3d78c0SShri Abhyankar reallocs++; 364*ce3d78c0SShri Abhyankar } 365*ce3d78c0SShri Abhyankar 366*ce3d78c0SShri Abhyankar /* copy data into free space, then initialize lnk */ 367*ce3d78c0SShri Abhyankar ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 368*ce3d78c0SShri Abhyankar bi_ptr[i] = current_space->array; 369*ce3d78c0SShri Abhyankar current_space->array += nzi; 370*ce3d78c0SShri Abhyankar current_space->local_used += nzi; 371*ce3d78c0SShri Abhyankar current_space->local_remaining -= nzi; 372*ce3d78c0SShri Abhyankar } 373*ce3d78c0SShri Abhyankar #if defined(PETSC_USE_INFO) 374*ce3d78c0SShri Abhyankar if (ai[n] != 0) { 375*ce3d78c0SShri Abhyankar PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 376*ce3d78c0SShri Abhyankar ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 377*ce3d78c0SShri Abhyankar ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 378*ce3d78c0SShri Abhyankar ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 379*ce3d78c0SShri Abhyankar ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 380*ce3d78c0SShri Abhyankar } else { 381*ce3d78c0SShri Abhyankar ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 382*ce3d78c0SShri Abhyankar } 383*ce3d78c0SShri Abhyankar #endif 384*ce3d78c0SShri Abhyankar 385*ce3d78c0SShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 386*ce3d78c0SShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 387*ce3d78c0SShri Abhyankar 388*ce3d78c0SShri Abhyankar /* destroy list of free space and other temporary array(s) */ 389*ce3d78c0SShri Abhyankar ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 390*ce3d78c0SShri Abhyankar ierr = PetscFreeSpaceContiguous_newdatastruct(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 391*ce3d78c0SShri Abhyankar ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 392*ce3d78c0SShri Abhyankar ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 393*ce3d78c0SShri Abhyankar 394*ce3d78c0SShri Abhyankar /* put together the new matrix */ 395*ce3d78c0SShri Abhyankar ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 396*ce3d78c0SShri Abhyankar ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 397*ce3d78c0SShri Abhyankar b = (Mat_SeqAIJ*)(B)->data; 398*ce3d78c0SShri Abhyankar b->free_a = PETSC_TRUE; 399*ce3d78c0SShri Abhyankar b->free_ij = PETSC_TRUE; 400*ce3d78c0SShri Abhyankar b->singlemalloc = PETSC_FALSE; 401*ce3d78c0SShri Abhyankar ierr = PetscMalloc((bi[2*n+1])*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 402*ce3d78c0SShri Abhyankar b->j = bj; 403*ce3d78c0SShri Abhyankar b->i = bi; 404*ce3d78c0SShri Abhyankar b->diag = bdiag; 405*ce3d78c0SShri Abhyankar b->ilen = 0; 406*ce3d78c0SShri Abhyankar b->imax = 0; 407*ce3d78c0SShri Abhyankar b->row = isrow; 408*ce3d78c0SShri Abhyankar b->col = iscol; 409*ce3d78c0SShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 410*ce3d78c0SShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 411*ce3d78c0SShri Abhyankar b->icol = isicol; 412*ce3d78c0SShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 413*ce3d78c0SShri Abhyankar 414*ce3d78c0SShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 415*ce3d78c0SShri Abhyankar ierr = PetscLogObjectMemory(B,bi[2*n+1]*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 416*ce3d78c0SShri Abhyankar b->maxnz = b->nz = bi[2*n+1] ; 417*ce3d78c0SShri Abhyankar 418*ce3d78c0SShri Abhyankar (B)->factor = MAT_FACTOR_LU; 419*ce3d78c0SShri Abhyankar (B)->info.factor_mallocs = reallocs; 420*ce3d78c0SShri Abhyankar (B)->info.fill_ratio_given = f; 421*ce3d78c0SShri Abhyankar 422*ce3d78c0SShri Abhyankar if (ai[n]) { 423*ce3d78c0SShri Abhyankar (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 424*ce3d78c0SShri Abhyankar } else { 425*ce3d78c0SShri Abhyankar (B)->info.fill_ratio_needed = 0.0; 426*ce3d78c0SShri Abhyankar } 427*ce3d78c0SShri Abhyankar (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct; 428*ce3d78c0SShri Abhyankar (B)->ops->solve = MatSolve_SeqAIJ; 429*ce3d78c0SShri Abhyankar (B)->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 430*ce3d78c0SShri Abhyankar /* switch to inodes if appropriate */ 431*ce3d78c0SShri Abhyankar /* ierr = MatLUFactorSymbolic_Inode(B,A,isrow,iscol,info);CHKERRQ(ierr); */ 432*ce3d78c0SShri Abhyankar PetscFunctionReturn(0); 433*ce3d78c0SShri Abhyankar } 434*ce3d78c0SShri Abhyankar 435*ce3d78c0SShri Abhyankar 4365b5bc046SBarry Smith /* 4375b5bc046SBarry Smith Trouble in factorization, should we dump the original matrix? 4385b5bc046SBarry Smith */ 4395b5bc046SBarry Smith #undef __FUNCT__ 4405b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix" 4415b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A) 4425b5bc046SBarry Smith { 4435b5bc046SBarry Smith PetscErrorCode ierr; 44490d69ab7SBarry Smith PetscTruth flg = PETSC_FALSE; 4455b5bc046SBarry Smith 4465b5bc046SBarry Smith PetscFunctionBegin; 44790d69ab7SBarry Smith ierr = PetscOptionsGetTruth(PETSC_NULL,"-mat_factor_dump_on_error",&flg,PETSC_NULL);CHKERRQ(ierr); 4485b5bc046SBarry Smith if (flg) { 4495b5bc046SBarry Smith PetscViewer viewer; 4505b5bc046SBarry Smith char filename[PETSC_MAX_PATH_LEN]; 4515b5bc046SBarry Smith 4525b5bc046SBarry Smith ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr); 4537adad957SLisandro Dalcin ierr = PetscViewerBinaryOpen(((PetscObject)A)->comm,filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr); 4545b5bc046SBarry Smith ierr = MatView(A,viewer);CHKERRQ(ierr); 4555b5bc046SBarry Smith ierr = PetscViewerDestroy(viewer);CHKERRQ(ierr); 4565b5bc046SBarry Smith } 4575b5bc046SBarry Smith PetscFunctionReturn(0); 4585b5bc046SBarry Smith } 4595b5bc046SBarry Smith 460db4efbfdSBarry Smith extern PetscErrorCode MatSolve_Inode(Mat,Vec,Vec); 461db4efbfdSBarry Smith 4620a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 463c9c72f8fSHong Zhang extern PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_iludt(Mat,Vec,Vec); 464c9c72f8fSHong Zhang extern PetscErrorCode MatSolve_SeqAIJ_iludt(Mat,Vec,Vec); 465c9c72f8fSHong Zhang 466c9c72f8fSHong Zhang #undef __FUNCT__ 467c9c72f8fSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_newdatastruct" 468c9c72f8fSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_newdatastruct(Mat B,Mat A,const MatFactorInfo *info) 469c9c72f8fSHong Zhang { 470c9c72f8fSHong Zhang Mat C=B; 471c9c72f8fSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 472c9c72f8fSHong Zhang IS isrow = b->row,isicol = b->icol; 473c9c72f8fSHong Zhang PetscErrorCode ierr; 474c9c72f8fSHong Zhang const PetscInt *r,*ic,*ics; 475c9c72f8fSHong Zhang PetscInt i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 476c9c72f8fSHong Zhang PetscInt *ajtmp,*bjtmp,nz,nzL,row,*bdiag=b->diag,*pj; 477c9c72f8fSHong Zhang MatScalar *rtmp,*pc,multiplier,*v,*pv,*aa=a->a; 478c9c72f8fSHong Zhang PetscReal shift=info->shiftinblocks; 479c9c72f8fSHong Zhang PetscTruth row_identity, col_identity; 480c9c72f8fSHong Zhang 481c9c72f8fSHong Zhang PetscFunctionBegin; 482c9c72f8fSHong Zhang /* printf("MatLUFactorNumeric_SeqAIJ_newdatastruct is called ...\n"); */ 483c9c72f8fSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 484c9c72f8fSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 485c9c72f8fSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 486c9c72f8fSHong Zhang ics = ic; 487c9c72f8fSHong Zhang 488c9c72f8fSHong Zhang for (i=0; i<n; i++){ 489c9c72f8fSHong Zhang /* zero rtmp */ 490c9c72f8fSHong Zhang /* L part */ 491c9c72f8fSHong Zhang nz = bi[i+1] - bi[i]; 492c9c72f8fSHong Zhang bjtmp = bj + bi[i]; 493c9c72f8fSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 494c9c72f8fSHong Zhang 495c9c72f8fSHong Zhang /* U part */ 4968fc3a347SHong Zhang nz = bi[2*n-i+1] - bi[2*n-i]; 4978fc3a347SHong Zhang bjtmp = bj + bi[2*n-i]; 498c9c72f8fSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 499c9c72f8fSHong Zhang 500c9c72f8fSHong Zhang /* load in initial (unfactored row) */ 501c9c72f8fSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 502c9c72f8fSHong Zhang ajtmp = aj + ai[r[i]]; 503c9c72f8fSHong Zhang v = aa + ai[r[i]]; 504c9c72f8fSHong Zhang for (j=0; j<nz; j++) { 505c9c72f8fSHong Zhang rtmp[ics[ajtmp[j]]] = v[j]; 506c9c72f8fSHong Zhang } 507c9c72f8fSHong Zhang if (rtmp[ics[r[i]]] == 0.0){ 508c9c72f8fSHong Zhang rtmp[ics[r[i]]] += shift; /* shift the diagonal of the matrix */ 509c9c72f8fSHong Zhang /* printf("row %d, shift %g\n",i,shift); */ 510c9c72f8fSHong Zhang } 511c9c72f8fSHong Zhang 512c9c72f8fSHong Zhang /* elimination */ 513c9c72f8fSHong Zhang bjtmp = bj + bi[i]; 514c9c72f8fSHong Zhang row = *bjtmp++; 515c9c72f8fSHong Zhang nzL = bi[i+1] - bi[i]; 516c9c72f8fSHong Zhang k = 0; 517c9c72f8fSHong Zhang while (k < nzL) { 518c9c72f8fSHong Zhang pc = rtmp + row; 519c9c72f8fSHong Zhang if (*pc != 0.0) { 520c9c72f8fSHong Zhang pv = b->a + bdiag[row]; 521c9c72f8fSHong Zhang multiplier = *pc * (*pv); 522c9c72f8fSHong Zhang *pc = multiplier; 5238fc3a347SHong Zhang pj = b->j + bi[2*n-row]; /* begining of U(row,:) */ 5248fc3a347SHong Zhang pv = b->a + bi[2*n-row]; 5258fc3a347SHong Zhang nz = bi[2*n-row+1] - bi[2*n-row] - 1; /* num of entries in U(row,:), excluding diag */ 526c9c72f8fSHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 527c9c72f8fSHong Zhang ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 528c9c72f8fSHong Zhang } 529c9c72f8fSHong Zhang row = *bjtmp++; k++; 530c9c72f8fSHong Zhang } 531c9c72f8fSHong Zhang 532c9c72f8fSHong Zhang /* finished row so stick it into b->a */ 533c9c72f8fSHong Zhang /* L part */ 534c9c72f8fSHong Zhang pv = b->a + bi[i] ; 535c9c72f8fSHong Zhang pj = b->j + bi[i] ; 536c9c72f8fSHong Zhang nz = bi[i+1] - bi[i]; 537c9c72f8fSHong Zhang for (j=0; j<nz; j++) { 538c9c72f8fSHong Zhang pv[j] = rtmp[pj[j]]; 539c9c72f8fSHong Zhang } 540c9c72f8fSHong Zhang 541c9c72f8fSHong Zhang /* Mark diagonal and invert diagonal for simplier triangular solves */ 542c9c72f8fSHong Zhang pv = b->a + bdiag[i]; 543c9c72f8fSHong Zhang pj = b->j + bdiag[i]; 544c9c72f8fSHong Zhang /* if (*pj != i)SETERRQ2(PETSC_ERR_SUP,"row %d != *pj %d",i,*pj) */ 545c9c72f8fSHong Zhang *pv = 1.0/rtmp[*pj]; 546c9c72f8fSHong Zhang 547c9c72f8fSHong Zhang /* U part */ 5488fc3a347SHong Zhang pv = b->a + bi[2*n-i]; 5498fc3a347SHong Zhang pj = b->j + bi[2*n-i]; 5508fc3a347SHong Zhang nz = bi[2*n-i+1] - bi[2*n-i] - 1; 5518fc3a347SHong Zhang for (j=0; j<nz; j++) pv[j] = rtmp[pj[j]]; 552c9c72f8fSHong Zhang } 553c9c72f8fSHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 554c9c72f8fSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 555c9c72f8fSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 556c9c72f8fSHong Zhang 557c9c72f8fSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 558c9c72f8fSHong Zhang ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 559c9c72f8fSHong Zhang if (row_identity && col_identity) { 560c9c72f8fSHong Zhang C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_iludt; 561c9c72f8fSHong Zhang } else { 562c9c72f8fSHong Zhang C->ops->solve = MatSolve_SeqAIJ_iludt; 563c9c72f8fSHong Zhang } 564c9c72f8fSHong Zhang 565c9c72f8fSHong Zhang C->ops->solveadd = 0; 566c9c72f8fSHong Zhang C->ops->solvetranspose = 0; 567c9c72f8fSHong Zhang C->ops->solvetransposeadd = 0; 568c9c72f8fSHong Zhang C->ops->matsolve = 0; 569c9c72f8fSHong Zhang C->assembled = PETSC_TRUE; 570c9c72f8fSHong Zhang C->preallocated = PETSC_TRUE; 571c9c72f8fSHong Zhang ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 572c9c72f8fSHong Zhang PetscFunctionReturn(0); 573c9c72f8fSHong Zhang } 574c9c72f8fSHong Zhang 5754a2ae208SSatish Balay #undef __FUNCT__ 5764a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ" 5770481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 578289bc588SBarry Smith { 579719d5645SBarry Smith Mat C=B; 580416022c9SBarry Smith Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 58182bf6240SBarry Smith IS isrow = b->row,isicol = b->icol; 5826849ba73SBarry Smith PetscErrorCode ierr; 5835d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 584d278edc7SBarry Smith PetscInt nz,row,i,j,n=A->rmap->n,diag; 585d278edc7SBarry Smith const PetscInt *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 586d278edc7SBarry Smith const PetscInt *ajtmp,*bjtmp,*diag_offset = b->diag,*pj; 587d278edc7SBarry Smith MatScalar *pv,*rtmp,*pc,multiplier,d; 588d278edc7SBarry Smith const MatScalar *v,*aa=a->a; 58933f9893dSHong Zhang PetscReal rs=0.0; 590b3bf805bSHong Zhang LUShift_Ctx sctx; 59142898933SHong Zhang PetscInt newshift,*ddiag; 592289bc588SBarry Smith 5933a40ed3dSBarry Smith PetscFunctionBegin; 59478b31e54SBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 59578b31e54SBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 596b78a477dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 5979f7d0b68SHong Zhang ics = ic; 598289bc588SBarry Smith 599ada7143aSSatish Balay sctx.shift_top = 0; 600ada7143aSSatish Balay sctx.nshift_max = 0; 601ada7143aSSatish Balay sctx.shift_lo = 0; 602ada7143aSSatish Balay sctx.shift_hi = 0; 6030481f469SBarry Smith sctx.shift_fraction = 0; 604ada7143aSSatish Balay 605118f5789SBarry Smith /* if both shift schemes are chosen by user, only use info->shiftpd */ 6061a890ab1SHong Zhang if (info->shiftpd) { /* set sctx.shift_top=max{rs} */ 60742898933SHong Zhang ddiag = a->diag; 6089f7d0b68SHong Zhang sctx.shift_top = info->zeropivot; 6096cc28720Svictorle for (i=0; i<n; i++) { 6109f95998fSHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 6119f7d0b68SHong Zhang d = (aa)[ddiag[i]]; 6121a890ab1SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 6139f7d0b68SHong Zhang v = aa+ai[i]; 6149f7d0b68SHong Zhang nz = ai[i+1] - ai[i]; 615e24cfe64SHong Zhang for (j=0; j<nz; j++) 6161a890ab1SHong Zhang rs += PetscAbsScalar(v[j]); 6171a890ab1SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 6186cc28720Svictorle } 619118f5789SBarry Smith sctx.shift_top *= 1.1; 620d2276718SHong Zhang sctx.nshift_max = 5; 621d2276718SHong Zhang sctx.shift_lo = 0.; 622d2276718SHong Zhang sctx.shift_hi = 1.; 623a0a356efSHong Zhang } 624a0a356efSHong Zhang 6259f7d0b68SHong Zhang sctx.shift_amount = 0.0; 626a0a356efSHong Zhang sctx.nshift = 0; 627085256b3SBarry Smith do { 628d2276718SHong Zhang sctx.lushift = PETSC_FALSE; 629289bc588SBarry Smith for (i=0; i<n; i++){ 630b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 631b3bf805bSHong Zhang bjtmp = bj + bi[i]; 6329f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 633289bc588SBarry Smith 634289bc588SBarry Smith /* load in initial (unfactored row) */ 6359f7d0b68SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 6369f7d0b68SHong Zhang ajtmp = aj + ai[r[i]]; 6379f7d0b68SHong Zhang v = aa + ai[r[i]]; 638085256b3SBarry Smith for (j=0; j<nz; j++) { 639b3bf805bSHong Zhang rtmp[ics[ajtmp[j]]] = v[j]; 640085256b3SBarry Smith } 641d2276718SHong Zhang rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 642bd1bc851SBarry Smith /* if (sctx.shift_amount > 0.0) printf("row %d, shift %g\n",i,sctx.shift_amount); */ 643289bc588SBarry Smith 644b3bf805bSHong Zhang row = *bjtmp++; 645f2582111SSatish Balay while (row < i) { 6468c37ef55SBarry Smith pc = rtmp + row; 6478c37ef55SBarry Smith if (*pc != 0.0) { 648010a20caSHong Zhang pv = b->a + diag_offset[row]; 649010a20caSHong Zhang pj = b->j + diag_offset[row] + 1; 65035aab85fSBarry Smith multiplier = *pc / *pv++; 6518c37ef55SBarry Smith *pc = multiplier; 652b3bf805bSHong Zhang nz = bi[row+1] - diag_offset[row] - 1; 6539f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 654dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 655289bc588SBarry Smith } 656b3bf805bSHong Zhang row = *bjtmp++; 657289bc588SBarry Smith } 658416022c9SBarry Smith /* finished row so stick it into b->a */ 659b3bf805bSHong Zhang pv = b->a + bi[i] ; 660b3bf805bSHong Zhang pj = b->j + bi[i] ; 661b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 662b3bf805bSHong Zhang diag = diag_offset[i] - bi[i]; 663e57ff13aSBarry Smith rs = 0.0; 664d3d32019SBarry Smith for (j=0; j<nz; j++) { 6659f7d0b68SHong Zhang pv[j] = rtmp[pj[j]]; 6669f7d0b68SHong Zhang rs += PetscAbsScalar(pv[j]); 667d3d32019SBarry Smith } 668e57ff13aSBarry Smith rs -= PetscAbsScalar(pv[diag]); 669d2276718SHong Zhang 670b3bf805bSHong Zhang /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */ 671d2276718SHong Zhang sctx.rs = rs; 672d2276718SHong Zhang sctx.pv = pv[diag]; 673c6b1f410SHong Zhang ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); 6745b5bc046SBarry Smith if (newshift == 1) break; 67535aab85fSBarry Smith } 676d2276718SHong Zhang 6770481f469SBarry Smith if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 6786cc28720Svictorle /* 6796c037d1bSvictorle * if no shift in this attempt & shifting & started shifting & can refine, 6806cc28720Svictorle * then try lower shift 6816cc28720Svictorle */ 6820481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 6830481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 6840481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 685d2276718SHong Zhang sctx.lushift = PETSC_TRUE; 686d2276718SHong Zhang sctx.nshift++; 6876cc28720Svictorle } 688d2276718SHong Zhang } while (sctx.lushift); 6890f11f9aeSSatish Balay 69035aab85fSBarry Smith /* invert diagonal entries for simplier triangular solves */ 69135aab85fSBarry Smith for (i=0; i<n; i++) { 692010a20caSHong Zhang b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]]; 69335aab85fSBarry Smith } 694606d414cSSatish Balay ierr = PetscFree(rtmp);CHKERRQ(ierr); 69578b31e54SBarry Smith ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 69678b31e54SBarry Smith ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 697db4efbfdSBarry Smith if (b->inode.use) { 698db4efbfdSBarry Smith C->ops->solve = MatSolve_Inode; 699db4efbfdSBarry Smith } else { 7008d8c7c43SBarry Smith PetscTruth row_identity, col_identity; 7018d8c7c43SBarry Smith ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 7028d8c7c43SBarry Smith ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 7038d8c7c43SBarry Smith if (row_identity && col_identity) { 7048d8c7c43SBarry Smith C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 7058d8c7c43SBarry Smith } else { 706db4efbfdSBarry Smith C->ops->solve = MatSolve_SeqAIJ; 707db4efbfdSBarry Smith } 7088d8c7c43SBarry Smith } 709db4efbfdSBarry Smith C->ops->solveadd = MatSolveAdd_SeqAIJ; 710db4efbfdSBarry Smith C->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 711db4efbfdSBarry Smith C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 712de26497bSBarry Smith C->ops->matsolve = MatMatSolve_SeqAIJ; 713c456f294SBarry Smith C->assembled = PETSC_TRUE; 7145c9eb25fSBarry Smith C->preallocated = PETSC_TRUE; 715d0f46423SBarry Smith ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 716d2276718SHong Zhang if (sctx.nshift){ 717e738e422SBarry Smith if (info->shiftpd) { 7180481f469SBarry 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); 719e738e422SBarry Smith } else if (info->shiftnz) { 720e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 7216cc28720Svictorle } 7220697b6caSHong Zhang } 7233a40ed3dSBarry Smith PetscFunctionReturn(0); 724289bc588SBarry Smith } 7250889b5dcSvictorle 726137fb511SHong Zhang /* 727137fb511SHong Zhang This routine implements inplace ILU(0) with row or/and column permutations. 728137fb511SHong Zhang Input: 729137fb511SHong Zhang A - original matrix 730137fb511SHong Zhang Output; 731137fb511SHong Zhang A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i] 732137fb511SHong Zhang a->j (col index) is permuted by the inverse of colperm, then sorted 733137fb511SHong Zhang a->a reordered accordingly with a->j 734137fb511SHong Zhang a->diag (ptr to diagonal elements) is updated. 735137fb511SHong Zhang */ 736137fb511SHong Zhang #undef __FUNCT__ 737137fb511SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_InplaceWithPerm" 7380481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info) 739137fb511SHong Zhang { 740b051339dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 741b051339dSHong Zhang IS isrow = a->row,isicol = a->icol; 742137fb511SHong Zhang PetscErrorCode ierr; 7435d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 7445d0c19d7SBarry Smith PetscInt i,j,n=A->rmap->n,*ai=a->i,*aj=a->j; 7455d0c19d7SBarry Smith PetscInt *ajtmp,nz,row; 746b051339dSHong Zhang PetscInt *diag = a->diag,nbdiag,*pj; 747a77337e4SBarry Smith PetscScalar *rtmp,*pc,multiplier,d; 748a77337e4SBarry Smith MatScalar *v,*pv; 749137fb511SHong Zhang PetscReal rs; 750137fb511SHong Zhang LUShift_Ctx sctx; 751b051339dSHong Zhang PetscInt newshift; 752137fb511SHong Zhang 753137fb511SHong Zhang PetscFunctionBegin; 754719d5645SBarry Smith if (A != B) SETERRQ(PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address"); 755137fb511SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 756137fb511SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 757137fb511SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr); 758137fb511SHong Zhang ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 759b051339dSHong Zhang ics = ic; 760137fb511SHong Zhang 761137fb511SHong Zhang sctx.shift_top = 0; 762137fb511SHong Zhang sctx.nshift_max = 0; 763137fb511SHong Zhang sctx.shift_lo = 0; 764137fb511SHong Zhang sctx.shift_hi = 0; 7650481f469SBarry Smith sctx.shift_fraction = 0; 766137fb511SHong Zhang 767137fb511SHong Zhang /* if both shift schemes are chosen by user, only use info->shiftpd */ 768137fb511SHong Zhang if (info->shiftpd) { /* set sctx.shift_top=max{rs} */ 769137fb511SHong Zhang sctx.shift_top = 0; 770137fb511SHong Zhang for (i=0; i<n; i++) { 771137fb511SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 772b051339dSHong Zhang d = (a->a)[diag[i]]; 773137fb511SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 774b051339dSHong Zhang v = a->a+ai[i]; 775b051339dSHong Zhang nz = ai[i+1] - ai[i]; 776137fb511SHong Zhang for (j=0; j<nz; j++) 777137fb511SHong Zhang rs += PetscAbsScalar(v[j]); 778137fb511SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 779137fb511SHong Zhang } 780137fb511SHong Zhang if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot; 781137fb511SHong Zhang sctx.shift_top *= 1.1; 782137fb511SHong Zhang sctx.nshift_max = 5; 783137fb511SHong Zhang sctx.shift_lo = 0.; 784137fb511SHong Zhang sctx.shift_hi = 1.; 785137fb511SHong Zhang } 786137fb511SHong Zhang 787137fb511SHong Zhang sctx.shift_amount = 0; 788137fb511SHong Zhang sctx.nshift = 0; 789137fb511SHong Zhang do { 790137fb511SHong Zhang sctx.lushift = PETSC_FALSE; 791137fb511SHong Zhang for (i=0; i<n; i++){ 792b051339dSHong Zhang /* load in initial unfactored row */ 793b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 794b051339dSHong Zhang ajtmp = aj + ai[r[i]]; 795b051339dSHong Zhang v = a->a + ai[r[i]]; 796137fb511SHong Zhang /* sort permuted ajtmp and values v accordingly */ 797b051339dSHong Zhang for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]]; 798137fb511SHong Zhang ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr); 799137fb511SHong Zhang 800b051339dSHong Zhang diag[r[i]] = ai[r[i]]; 801137fb511SHong Zhang for (j=0; j<nz; j++) { 802137fb511SHong Zhang rtmp[ajtmp[j]] = v[j]; 803b051339dSHong Zhang if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */ 804137fb511SHong Zhang } 805137fb511SHong Zhang rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 806137fb511SHong Zhang 807b051339dSHong Zhang row = *ajtmp++; 808137fb511SHong Zhang while (row < i) { 809137fb511SHong Zhang pc = rtmp + row; 810137fb511SHong Zhang if (*pc != 0.0) { 811b051339dSHong Zhang pv = a->a + diag[r[row]]; 812b051339dSHong Zhang pj = aj + diag[r[row]] + 1; 813137fb511SHong Zhang 814137fb511SHong Zhang multiplier = *pc / *pv++; 815137fb511SHong Zhang *pc = multiplier; 816b051339dSHong Zhang nz = ai[r[row]+1] - diag[r[row]] - 1; 817b051339dSHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 818dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 819137fb511SHong Zhang } 820b051339dSHong Zhang row = *ajtmp++; 821137fb511SHong Zhang } 822b051339dSHong Zhang /* finished row so overwrite it onto a->a */ 823b051339dSHong Zhang pv = a->a + ai[r[i]] ; 824b051339dSHong Zhang pj = aj + ai[r[i]] ; 825b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 826b051339dSHong Zhang nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */ 827137fb511SHong Zhang 828137fb511SHong Zhang rs = 0.0; 829137fb511SHong Zhang for (j=0; j<nz; j++) { 830b051339dSHong Zhang pv[j] = rtmp[pj[j]]; 831b051339dSHong Zhang if (j != nbdiag) rs += PetscAbsScalar(pv[j]); 832137fb511SHong Zhang } 833137fb511SHong Zhang 834137fb511SHong Zhang /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */ 835137fb511SHong Zhang sctx.rs = rs; 836b051339dSHong Zhang sctx.pv = pv[nbdiag]; 837c6b1f410SHong Zhang ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); 838137fb511SHong Zhang if (newshift == 1) break; 839137fb511SHong Zhang } 840137fb511SHong Zhang 8410481f469SBarry Smith if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 842137fb511SHong Zhang /* 843137fb511SHong Zhang * if no shift in this attempt & shifting & started shifting & can refine, 844137fb511SHong Zhang * then try lower shift 845137fb511SHong Zhang */ 8460481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 8470481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 8480481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 849137fb511SHong Zhang sctx.lushift = PETSC_TRUE; 850137fb511SHong Zhang sctx.nshift++; 851137fb511SHong Zhang } 852137fb511SHong Zhang } while (sctx.lushift); 853137fb511SHong Zhang 854137fb511SHong Zhang /* invert diagonal entries for simplier triangular solves */ 855137fb511SHong Zhang for (i=0; i<n; i++) { 856b051339dSHong Zhang a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]]; 857137fb511SHong Zhang } 858137fb511SHong Zhang 859137fb511SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 860137fb511SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 861137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 862b051339dSHong Zhang A->ops->solve = MatSolve_SeqAIJ_InplaceWithPerm; 863db4efbfdSBarry Smith A->ops->solveadd = MatSolveAdd_SeqAIJ; 864db4efbfdSBarry Smith A->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 865db4efbfdSBarry Smith A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 866b051339dSHong Zhang A->assembled = PETSC_TRUE; 8675c9eb25fSBarry Smith A->preallocated = PETSC_TRUE; 868d0f46423SBarry Smith ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr); 869137fb511SHong Zhang if (sctx.nshift){ 870e738e422SBarry Smith if (info->shiftpd) { 8710481f469SBarry 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); 872e738e422SBarry Smith } else if (info->shiftnz) { 873e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 874137fb511SHong Zhang } 875137fb511SHong Zhang } 876137fb511SHong Zhang PetscFunctionReturn(0); 877137fb511SHong Zhang } 878137fb511SHong Zhang 8790a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 8804a2ae208SSatish Balay #undef __FUNCT__ 8814a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ" 8820481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info) 883da3a660dSBarry Smith { 884dfbe8321SBarry Smith PetscErrorCode ierr; 885416022c9SBarry Smith Mat C; 886416022c9SBarry Smith 8873a40ed3dSBarry Smith PetscFunctionBegin; 88843244d56SBarry Smith ierr = MatGetFactor(A,MAT_SOLVER_PETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr); 889719d5645SBarry Smith ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr); 890719d5645SBarry Smith ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr); 891db4efbfdSBarry Smith A->ops->solve = C->ops->solve; 892db4efbfdSBarry Smith A->ops->solvetranspose = C->ops->solvetranspose; 893273d9f13SBarry Smith ierr = MatHeaderCopy(A,C);CHKERRQ(ierr); 89452e6d16bSBarry Smith ierr = PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr); 8953a40ed3dSBarry Smith PetscFunctionReturn(0); 896da3a660dSBarry Smith } 8970a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 8981d098868SHong Zhang 899889a8dedSBarry Smith 9004a2ae208SSatish Balay #undef __FUNCT__ 9014a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ" 902dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx) 9038c37ef55SBarry Smith { 904416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 905416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 9066849ba73SBarry Smith PetscErrorCode ierr; 9075d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 9085d0c19d7SBarry Smith PetscInt nz; 9095d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 910dd6ea824SBarry Smith PetscScalar *x,*tmp,*tmps,sum; 911d9fead3dSBarry Smith const PetscScalar *b; 912dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 9138c37ef55SBarry Smith 9143a40ed3dSBarry Smith PetscFunctionBegin; 9153a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 91691bf9adeSBarry Smith 917d9fead3dSBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 9181ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 919416022c9SBarry Smith tmp = a->solve_work; 9208c37ef55SBarry Smith 9213b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 9223b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 9238c37ef55SBarry Smith 9248c37ef55SBarry Smith /* forward solve the lower triangular */ 9258c37ef55SBarry Smith tmp[0] = b[*r++]; 926010a20caSHong Zhang tmps = tmp; 9278c37ef55SBarry Smith for (i=1; i<n; i++) { 928010a20caSHong Zhang v = aa + ai[i] ; 929010a20caSHong Zhang vi = aj + ai[i] ; 93053e7425aSBarry Smith nz = a->diag[i] - ai[i]; 93153e7425aSBarry Smith sum = b[*r++]; 932003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 9338c37ef55SBarry Smith tmp[i] = sum; 9348c37ef55SBarry Smith } 9358c37ef55SBarry Smith 9368c37ef55SBarry Smith /* backward solve the upper triangular */ 9378c37ef55SBarry Smith for (i=n-1; i>=0; i--){ 938010a20caSHong Zhang v = aa + a->diag[i] + 1; 939010a20caSHong Zhang vi = aj + a->diag[i] + 1; 940416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 9418c37ef55SBarry Smith sum = tmp[i]; 942003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 943010a20caSHong Zhang x[*c--] = tmp[i] = sum*aa[a->diag[i]]; 9448c37ef55SBarry Smith } 9458c37ef55SBarry Smith 9463b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 9473b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 948d9fead3dSBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 9491ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 950dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 9513a40ed3dSBarry Smith PetscFunctionReturn(0); 9528c37ef55SBarry Smith } 953026e39d0SSatish Balay 9542bebee5dSHong Zhang #undef __FUNCT__ 9552bebee5dSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ" 9562bebee5dSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X) 9572bebee5dSHong Zhang { 9582bebee5dSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 9592bebee5dSHong Zhang IS iscol = a->col,isrow = a->row; 9602bebee5dSHong Zhang PetscErrorCode ierr; 9615d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 9625d0c19d7SBarry Smith PetscInt nz,neq; 9635d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 964dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 965dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 966db4efbfdSBarry Smith PetscTruth bisdense,xisdense; 9672bebee5dSHong Zhang 9682bebee5dSHong Zhang PetscFunctionBegin; 9692bebee5dSHong Zhang if (!n) PetscFunctionReturn(0); 9702bebee5dSHong Zhang 971db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr); 972db4efbfdSBarry Smith if (!bisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix"); 973db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr); 974db4efbfdSBarry Smith if (!xisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix"); 975db4efbfdSBarry Smith 9762bebee5dSHong Zhang ierr = MatGetArray(B,&b);CHKERRQ(ierr); 9772bebee5dSHong Zhang ierr = MatGetArray(X,&x);CHKERRQ(ierr); 9782bebee5dSHong Zhang 9792bebee5dSHong Zhang tmp = a->solve_work; 9802bebee5dSHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 9812bebee5dSHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 9822bebee5dSHong Zhang 983a36b22ccSSatish Balay for (neq=0; neq<B->cmap->n; neq++){ 9842bebee5dSHong Zhang /* forward solve the lower triangular */ 9852bebee5dSHong Zhang tmp[0] = b[r[0]]; 9862bebee5dSHong Zhang tmps = tmp; 9872bebee5dSHong Zhang for (i=1; i<n; i++) { 9882bebee5dSHong Zhang v = aa + ai[i] ; 9892bebee5dSHong Zhang vi = aj + ai[i] ; 9902bebee5dSHong Zhang nz = a->diag[i] - ai[i]; 9912bebee5dSHong Zhang sum = b[r[i]]; 992003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 9932bebee5dSHong Zhang tmp[i] = sum; 9942bebee5dSHong Zhang } 9952bebee5dSHong Zhang /* backward solve the upper triangular */ 9962bebee5dSHong Zhang for (i=n-1; i>=0; i--){ 9972bebee5dSHong Zhang v = aa + a->diag[i] + 1; 9982bebee5dSHong Zhang vi = aj + a->diag[i] + 1; 9992bebee5dSHong Zhang nz = ai[i+1] - a->diag[i] - 1; 10002bebee5dSHong Zhang sum = tmp[i]; 1001003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10022bebee5dSHong Zhang x[c[i]] = tmp[i] = sum*aa[a->diag[i]]; 10032bebee5dSHong Zhang } 10042bebee5dSHong Zhang 10052bebee5dSHong Zhang b += n; 10062bebee5dSHong Zhang x += n; 10072bebee5dSHong Zhang } 10082bebee5dSHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10092bebee5dSHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 10102bebee5dSHong Zhang ierr = MatRestoreArray(B,&b);CHKERRQ(ierr); 10112bebee5dSHong Zhang ierr = MatRestoreArray(X,&x);CHKERRQ(ierr); 1012dc0b31edSSatish Balay ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr); 10132bebee5dSHong Zhang PetscFunctionReturn(0); 10142bebee5dSHong Zhang } 10152bebee5dSHong Zhang 1016137fb511SHong Zhang #undef __FUNCT__ 1017137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm" 1018137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx) 1019137fb511SHong Zhang { 1020137fb511SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1021137fb511SHong Zhang IS iscol = a->col,isrow = a->row; 1022137fb511SHong Zhang PetscErrorCode ierr; 10235d0c19d7SBarry Smith const PetscInt *r,*c,*rout,*cout; 10245d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 10255d0c19d7SBarry Smith PetscInt nz,row; 1026dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1027dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1028137fb511SHong Zhang 1029137fb511SHong Zhang PetscFunctionBegin; 1030137fb511SHong Zhang if (!n) PetscFunctionReturn(0); 1031137fb511SHong Zhang 1032137fb511SHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 1033137fb511SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1034137fb511SHong Zhang tmp = a->solve_work; 1035137fb511SHong Zhang 1036137fb511SHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1037137fb511SHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1038137fb511SHong Zhang 1039137fb511SHong Zhang /* forward solve the lower triangular */ 1040137fb511SHong Zhang tmp[0] = b[*r++]; 1041137fb511SHong Zhang tmps = tmp; 1042137fb511SHong Zhang for (row=1; row<n; row++) { 1043137fb511SHong Zhang i = rout[row]; /* permuted row */ 1044137fb511SHong Zhang v = aa + ai[i] ; 1045137fb511SHong Zhang vi = aj + ai[i] ; 1046137fb511SHong Zhang nz = a->diag[i] - ai[i]; 1047137fb511SHong Zhang sum = b[*r++]; 1048003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1049137fb511SHong Zhang tmp[row] = sum; 1050137fb511SHong Zhang } 1051137fb511SHong Zhang 1052137fb511SHong Zhang /* backward solve the upper triangular */ 1053137fb511SHong Zhang for (row=n-1; row>=0; row--){ 1054137fb511SHong Zhang i = rout[row]; /* permuted row */ 1055137fb511SHong Zhang v = aa + a->diag[i] + 1; 1056137fb511SHong Zhang vi = aj + a->diag[i] + 1; 1057137fb511SHong Zhang nz = ai[i+1] - a->diag[i] - 1; 1058137fb511SHong Zhang sum = tmp[row]; 1059003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1060137fb511SHong Zhang x[*c--] = tmp[row] = sum*aa[a->diag[i]]; 1061137fb511SHong Zhang } 1062137fb511SHong Zhang 1063137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1064137fb511SHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1065137fb511SHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 1066137fb511SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1067dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 1068137fb511SHong Zhang PetscFunctionReturn(0); 1069137fb511SHong Zhang } 1070137fb511SHong Zhang 1071930ae53cSBarry Smith /* ----------------------------------------------------------- */ 1072a4005a5dSBarry Smith #include "../src/mat/impls/aij/seq/ftn-kernels/fsolve.h" 10734a2ae208SSatish Balay #undef __FUNCT__ 10744a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering" 1075dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx) 1076930ae53cSBarry Smith { 1077930ae53cSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 10786849ba73SBarry Smith PetscErrorCode ierr; 1079d0f46423SBarry Smith PetscInt n = A->rmap->n; 1080b377110cSBarry Smith const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag; 1081356650c2SBarry Smith PetscScalar *x; 1082356650c2SBarry Smith const PetscScalar *b; 1083dd6ea824SBarry Smith const MatScalar *aa = a->a; 1084aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 1085356650c2SBarry Smith PetscInt adiag_i,i,nz,ai_i; 1086b377110cSBarry Smith const PetscInt *vi; 1087dd6ea824SBarry Smith const MatScalar *v; 1088dd6ea824SBarry Smith PetscScalar sum; 1089d85afc42SSatish Balay #endif 1090930ae53cSBarry Smith 10913a40ed3dSBarry Smith PetscFunctionBegin; 10923a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 1093930ae53cSBarry Smith 1094356650c2SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 10951ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1096930ae53cSBarry Smith 1097aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 10981c4feecaSSatish Balay fortransolveaij_(&n,x,ai,aj,adiag,aa,b); 10991c4feecaSSatish Balay #else 1100930ae53cSBarry Smith /* forward solve the lower triangular */ 1101930ae53cSBarry Smith x[0] = b[0]; 1102930ae53cSBarry Smith for (i=1; i<n; i++) { 1103930ae53cSBarry Smith ai_i = ai[i]; 1104930ae53cSBarry Smith v = aa + ai_i; 1105930ae53cSBarry Smith vi = aj + ai_i; 1106930ae53cSBarry Smith nz = adiag[i] - ai_i; 1107930ae53cSBarry Smith sum = b[i]; 1108003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1109930ae53cSBarry Smith x[i] = sum; 1110930ae53cSBarry Smith } 1111930ae53cSBarry Smith 1112930ae53cSBarry Smith /* backward solve the upper triangular */ 1113930ae53cSBarry Smith for (i=n-1; i>=0; i--){ 1114930ae53cSBarry Smith adiag_i = adiag[i]; 1115d708749eSBarry Smith v = aa + adiag_i + 1; 1116d708749eSBarry Smith vi = aj + adiag_i + 1; 1117930ae53cSBarry Smith nz = ai[i+1] - adiag_i - 1; 1118930ae53cSBarry Smith sum = x[i]; 1119003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1120930ae53cSBarry Smith x[i] = sum*aa[adiag_i]; 1121930ae53cSBarry Smith } 11221c4feecaSSatish Balay #endif 1123dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 1124356650c2SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 11251ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 11263a40ed3dSBarry Smith PetscFunctionReturn(0); 1127930ae53cSBarry Smith } 1128930ae53cSBarry Smith 11294a2ae208SSatish Balay #undef __FUNCT__ 11304a2ae208SSatish Balay #define __FUNCT__ "MatSolveAdd_SeqAIJ" 1131dfbe8321SBarry Smith PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx) 1132da3a660dSBarry Smith { 1133416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1134416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 11356849ba73SBarry Smith PetscErrorCode ierr; 11365d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 11375d0c19d7SBarry Smith PetscInt nz; 11385d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 1139dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,sum; 1140dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1141da3a660dSBarry Smith 11423a40ed3dSBarry Smith PetscFunctionBegin; 114378b31e54SBarry Smith if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 1144da3a660dSBarry Smith 11451ebc52fbSHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 11461ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1147416022c9SBarry Smith tmp = a->solve_work; 1148da3a660dSBarry Smith 11493b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 11503b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1151da3a660dSBarry Smith 1152da3a660dSBarry Smith /* forward solve the lower triangular */ 1153da3a660dSBarry Smith tmp[0] = b[*r++]; 1154da3a660dSBarry Smith for (i=1; i<n; i++) { 1155010a20caSHong Zhang v = aa + ai[i] ; 1156010a20caSHong Zhang vi = aj + ai[i] ; 1157416022c9SBarry Smith nz = a->diag[i] - ai[i]; 1158da3a660dSBarry Smith sum = b[*r++]; 1159010a20caSHong Zhang while (nz--) sum -= *v++ * tmp[*vi++ ]; 1160da3a660dSBarry Smith tmp[i] = sum; 1161da3a660dSBarry Smith } 1162da3a660dSBarry Smith 1163da3a660dSBarry Smith /* backward solve the upper triangular */ 1164da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1165010a20caSHong Zhang v = aa + a->diag[i] + 1; 1166010a20caSHong Zhang vi = aj + a->diag[i] + 1; 1167416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 1168da3a660dSBarry Smith sum = tmp[i]; 1169010a20caSHong Zhang while (nz--) sum -= *v++ * tmp[*vi++ ]; 1170010a20caSHong Zhang tmp[i] = sum*aa[a->diag[i]]; 1171da3a660dSBarry Smith x[*c--] += tmp[i]; 1172da3a660dSBarry Smith } 1173da3a660dSBarry Smith 11743b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 11753b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 11761ebc52fbSHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 11771ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1178dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 1179898183ecSLois Curfman McInnes 11803a40ed3dSBarry Smith PetscFunctionReturn(0); 1181da3a660dSBarry Smith } 1182da3a660dSBarry Smith /* -------------------------------------------------------------------*/ 11834a2ae208SSatish Balay #undef __FUNCT__ 11844a2ae208SSatish Balay #define __FUNCT__ "MatSolveTranspose_SeqAIJ" 1185dfbe8321SBarry Smith PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx) 1186da3a660dSBarry Smith { 1187416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 11882235e667SBarry Smith IS iscol = a->col,isrow = a->row; 11896849ba73SBarry Smith PetscErrorCode ierr; 11905d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 11915d0c19d7SBarry Smith PetscInt i,n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 11925d0c19d7SBarry Smith PetscInt nz,*diag = a->diag; 1193dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,s1; 1194dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1195da3a660dSBarry Smith 11963a40ed3dSBarry Smith PetscFunctionBegin; 11971ebc52fbSHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 11981ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1199416022c9SBarry Smith tmp = a->solve_work; 1200da3a660dSBarry Smith 12012235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 12022235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1203da3a660dSBarry Smith 1204da3a660dSBarry Smith /* copy the b into temp work space according to permutation */ 12052235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1206da3a660dSBarry Smith 1207da3a660dSBarry Smith /* forward solve the U^T */ 1208da3a660dSBarry Smith for (i=0; i<n; i++) { 1209010a20caSHong Zhang v = aa + diag[i] ; 1210010a20caSHong Zhang vi = aj + diag[i] + 1; 1211f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 1212c38d4ed2SBarry Smith s1 = tmp[i]; 1213ef66eb69SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 1214da3a660dSBarry Smith while (nz--) { 1215010a20caSHong Zhang tmp[*vi++ ] -= (*v++)*s1; 1216da3a660dSBarry Smith } 1217c38d4ed2SBarry Smith tmp[i] = s1; 1218da3a660dSBarry Smith } 1219da3a660dSBarry Smith 1220da3a660dSBarry Smith /* backward solve the L^T */ 1221da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1222010a20caSHong Zhang v = aa + diag[i] - 1 ; 1223010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1224f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 1225f1af5d2fSBarry Smith s1 = tmp[i]; 1226da3a660dSBarry Smith while (nz--) { 1227010a20caSHong Zhang tmp[*vi-- ] -= (*v--)*s1; 1228da3a660dSBarry Smith } 1229da3a660dSBarry Smith } 1230da3a660dSBarry Smith 1231da3a660dSBarry Smith /* copy tmp into x according to permutation */ 1232da3a660dSBarry Smith for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1233da3a660dSBarry Smith 12342235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 12352235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 12361ebc52fbSHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 12371ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1238da3a660dSBarry Smith 1239dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 12403a40ed3dSBarry Smith PetscFunctionReturn(0); 1241da3a660dSBarry Smith } 1242da3a660dSBarry Smith 12434a2ae208SSatish Balay #undef __FUNCT__ 12444a2ae208SSatish Balay #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ" 1245dfbe8321SBarry Smith PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx) 1246da3a660dSBarry Smith { 1247416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 12482235e667SBarry Smith IS iscol = a->col,isrow = a->row; 12496849ba73SBarry Smith PetscErrorCode ierr; 12505d0c19d7SBarry Smith const PetscInt *r,*c,*rout,*cout; 12515d0c19d7SBarry Smith PetscInt i,n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 12525d0c19d7SBarry Smith PetscInt nz,*diag = a->diag; 1253dd6ea824SBarry Smith PetscScalar *x,*b,*tmp; 1254dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 12556abc6512SBarry Smith 12563a40ed3dSBarry Smith PetscFunctionBegin; 125723bc6035SMatthew Knepley if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 12586abc6512SBarry Smith 12591ebc52fbSHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 12601ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1261416022c9SBarry Smith tmp = a->solve_work; 12626abc6512SBarry Smith 12632235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 12642235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 12656abc6512SBarry Smith 12666abc6512SBarry Smith /* copy the b into temp work space according to permutation */ 12672235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 12686abc6512SBarry Smith 12696abc6512SBarry Smith /* forward solve the U^T */ 12706abc6512SBarry Smith for (i=0; i<n; i++) { 1271010a20caSHong Zhang v = aa + diag[i] ; 1272010a20caSHong Zhang vi = aj + diag[i] + 1; 1273f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 12746abc6512SBarry Smith tmp[i] *= *v++; 12756abc6512SBarry Smith while (nz--) { 1276010a20caSHong Zhang tmp[*vi++ ] -= (*v++)*tmp[i]; 12776abc6512SBarry Smith } 12786abc6512SBarry Smith } 12796abc6512SBarry Smith 12806abc6512SBarry Smith /* backward solve the L^T */ 12816abc6512SBarry Smith for (i=n-1; i>=0; i--){ 1282010a20caSHong Zhang v = aa + diag[i] - 1 ; 1283010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1284f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 12856abc6512SBarry Smith while (nz--) { 1286010a20caSHong Zhang tmp[*vi-- ] -= (*v--)*tmp[i]; 12876abc6512SBarry Smith } 12886abc6512SBarry Smith } 12896abc6512SBarry Smith 12906abc6512SBarry Smith /* copy tmp into x according to permutation */ 12916abc6512SBarry Smith for (i=0; i<n; i++) x[r[i]] += tmp[i]; 12926abc6512SBarry Smith 12932235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 12942235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 12951ebc52fbSHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 12961ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 12976abc6512SBarry Smith 1298dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 12993a40ed3dSBarry Smith PetscFunctionReturn(0); 1300da3a660dSBarry Smith } 13019e25ed09SBarry Smith /* ----------------------------------------------------------------*/ 13023c5fc038SBarry Smith EXTERN PetscErrorCode Mat_CheckInode(Mat,PetscTruth); 1303f77e22a1SHong Zhang EXTERN PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,Mat,MatDuplicateOption,PetscTruth); 130408480c60SBarry Smith 13058fc3a347SHong Zhang /* 13068fc3a347SHong Zhang ilu(0) with natural ordering under new data structure. 13078fc3a347SHong Zhang Factored arrays bj and ba are stored as 13088fc3a347SHong Zhang L(0,:), L(1,:), ...,L(n-1,:), U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:) 13098fc3a347SHong Zhang 13108fc3a347SHong Zhang bi=fact->i is an array of size 2n+2, in which 13118fc3a347SHong Zhang bi+ 13128fc3a347SHong Zhang bi[i] -> 1st entry of L(i,:),i=0,...,i-1 1313f77e22a1SHong Zhang bi[n] -> points to L(n-1,:)+1 13148fc3a347SHong Zhang bi[n+1] -> 1st entry of U(n-1,:) 13158fc3a347SHong Zhang bi[2n-i] -> 1st entry of U(i,:) 13168fc3a347SHong Zhang bi[2n-i+1] -> end of U(i,:)+1, the 1st entry of U(i-1,:) 1317f77e22a1SHong Zhang bi[2n] -> 1st entry of U(0,:) 1318f77e22a1SHong Zhang bi[2n+1] -> points to U(0,:)+1 13198fc3a347SHong Zhang 13208fc3a347SHong Zhang U(i,:) contains diag[i] as its last entry, i.e., 13218fc3a347SHong Zhang U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 13228fc3a347SHong Zhang */ 1323c9c72f8fSHong Zhang #undef __FUNCT__ 13248fc3a347SHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct" 13258fc3a347SHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1326c9c72f8fSHong Zhang { 1327c9c72f8fSHong Zhang 1328c9c72f8fSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1329c9c72f8fSHong Zhang PetscErrorCode ierr; 1330c9c72f8fSHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag; 13318fc3a347SHong Zhang PetscInt i,j,nz,*bi,*bj,*bdiag; 1332c9c72f8fSHong Zhang 1333c9c72f8fSHong Zhang PetscFunctionBegin; 1334f77e22a1SHong Zhang ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr); 1335c9c72f8fSHong Zhang b = (Mat_SeqAIJ*)(fact)->data; 1336c9c72f8fSHong Zhang 1337f77e22a1SHong Zhang /* allocate matrix arrays for new data structure */ 1338f77e22a1SHong Zhang ierr = PetscMalloc3(ai[n]+1,PetscScalar,&b->a,ai[n]+1,PetscInt,&b->j,2*n+2,PetscInt,&b->i);CHKERRQ(ierr); 1339f77e22a1SHong Zhang ierr = PetscLogObjectMemory(fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(2*n+2)*sizeof(PetscInt));CHKERRQ(ierr); 1340f77e22a1SHong Zhang b->singlemalloc = PETSC_TRUE; 1341f77e22a1SHong Zhang if (!b->diag){ 13428fc3a347SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&b->diag);CHKERRQ(ierr); 1343f77e22a1SHong Zhang } 1344c9c72f8fSHong Zhang bdiag = b->diag; 1345c9c72f8fSHong Zhang 1346c9c72f8fSHong Zhang if (n > 0) { 134716a2bf60SHong Zhang ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr); 1348c9c72f8fSHong Zhang } 1349c9c72f8fSHong Zhang 1350c9c72f8fSHong Zhang /* set bi and bj with new data structure */ 1351c9c72f8fSHong Zhang bi = b->i; 1352c9c72f8fSHong Zhang bj = b->j; 1353c9c72f8fSHong Zhang 1354c9c72f8fSHong Zhang /* L part */ 1355c9c72f8fSHong Zhang bi[0] = 0; 1356c9c72f8fSHong Zhang for (i=0; i<n; i++){ 1357c9c72f8fSHong Zhang nz = adiag[i] - ai[i]; 1358c9c72f8fSHong Zhang bi[i+1] = bi[i] + nz; 1359c9c72f8fSHong Zhang aj = a->j + ai[i]; 1360c9c72f8fSHong Zhang for (j=0; j<nz; j++){ 1361c9c72f8fSHong Zhang *bj = aj[j]; bj++; 1362c9c72f8fSHong Zhang } 1363c9c72f8fSHong Zhang } 1364c9c72f8fSHong Zhang 1365c9c72f8fSHong Zhang /* U part */ 13668fc3a347SHong Zhang bi[n+1] = bi[n]; 1367c9c72f8fSHong Zhang for (i=n-1; i>=0; i--){ 1368c9c72f8fSHong Zhang nz = ai[i+1] - adiag[i] - 1; 13698fc3a347SHong Zhang bi[2*n-i+1] = bi[2*n-i] + nz + 1; 1370c9c72f8fSHong Zhang aj = a->j + adiag[i] + 1; 1371c9c72f8fSHong Zhang for (j=0; j<nz; j++){ 1372c9c72f8fSHong Zhang *bj = aj[j]; bj++; 1373c9c72f8fSHong Zhang } 1374c9c72f8fSHong Zhang /* diag[i] */ 1375c9c72f8fSHong Zhang *bj = i; bj++; 13768fc3a347SHong Zhang bdiag[i] = bi[2*n-i+1]-1; 1377c9c72f8fSHong Zhang } 1378c9c72f8fSHong Zhang PetscFunctionReturn(0); 1379c9c72f8fSHong Zhang } 1380c9c72f8fSHong Zhang 13814a2ae208SSatish Balay #undef __FUNCT__ 13826516239fSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_newdatastruct" 13836516239fSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_newdatastruct(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 13849e25ed09SBarry Smith { 1385416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 13869e25ed09SBarry Smith IS isicol; 13876849ba73SBarry Smith PetscErrorCode ierr; 13885d0c19d7SBarry Smith const PetscInt *r,*ic; 13895d0c19d7SBarry Smith PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,d; 13905a8e39fbSHong Zhang PetscInt *bi,*cols,nnz,*cols_lvl; 1391052f0c41SBarry Smith PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 13925a8e39fbSHong Zhang PetscInt i,levels,diagonal_fill; 139377c4ece6SBarry Smith PetscTruth col_identity,row_identity; 1394329f5518SBarry Smith PetscReal f; 13955a8e39fbSHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 13965a8e39fbSHong Zhang PetscBT lnkbt; 13975a8e39fbSHong Zhang PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 1398a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1399a1a86e44SBarry Smith PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 140009f38230SBarry Smith PetscTruth missing; 14019e25ed09SBarry Smith 14023a40ed3dSBarry Smith PetscFunctionBegin; 14036516239fSHong Zhang //printf("MatILUFactorSymbolic_SeqAIJ_newdatastruct ...\n"); 1404d0f46423SBarry 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); 14056cf997b0SBarry Smith f = info->fill; 140638baddfdSBarry Smith levels = (PetscInt)info->levels; 140738baddfdSBarry Smith diagonal_fill = (PetscInt)info->diagonal_fill; 14084c49b128SBarry Smith ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 140982bf6240SBarry Smith 1410be0abb6dSBarry Smith ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 1411be0abb6dSBarry Smith ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 1412c9c72f8fSHong Zhang 14136516239fSHong Zhang if (!levels && row_identity && col_identity) { 14146516239fSHong Zhang /* special case: ilu(0) with natural ordering */ 14158fc3a347SHong Zhang ierr = MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct(fact,A,isrow,iscol,info);CHKERRQ(ierr); 1416c9c72f8fSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct; 14176516239fSHong Zhang 14186516239fSHong Zhang fact->factor = MAT_FACTOR_ILU; 14196516239fSHong Zhang (fact)->info.factor_mallocs = 0; 14206516239fSHong Zhang (fact)->info.fill_ratio_given = info->fill; 14216516239fSHong Zhang (fact)->info.fill_ratio_needed = 1.0; 14226516239fSHong Zhang b = (Mat_SeqAIJ*)(fact)->data; 14236516239fSHong Zhang ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 14246516239fSHong Zhang if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 14256516239fSHong Zhang b->row = isrow; 14266516239fSHong Zhang b->col = iscol; 14276516239fSHong Zhang b->icol = isicol; 14286516239fSHong Zhang ierr = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 14296516239fSHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 14306516239fSHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 14316516239fSHong Zhang /* ierr = MatILUFactorSymbolic_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); */ 14326516239fSHong Zhang PetscFunctionReturn(0); 14336516239fSHong Zhang } 14346516239fSHong Zhang 14356516239fSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 14366516239fSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 14376516239fSHong Zhang 14386516239fSHong Zhang /* get new row pointers */ 14396516239fSHong Zhang ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 14406516239fSHong Zhang bi[0] = 0; 14416516239fSHong Zhang /* bdiag is location of diagonal in factor */ 14426516239fSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 14436516239fSHong Zhang bdiag[0] = 0; 14446516239fSHong Zhang 14456516239fSHong Zhang ierr = PetscMalloc((2*n+1)*sizeof(PetscInt**),&bj_ptr);CHKERRQ(ierr); 14466516239fSHong Zhang bjlvl_ptr = (PetscInt**)(bj_ptr + n); 14476516239fSHong Zhang 14486516239fSHong Zhang /* create a linked list for storing column indices of the active row */ 14496516239fSHong Zhang nlnk = n + 1; 14506516239fSHong Zhang ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 14516516239fSHong Zhang 14526516239fSHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 14536516239fSHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 14546516239fSHong Zhang current_space = free_space; 14556516239fSHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 14566516239fSHong Zhang current_space_lvl = free_space_lvl; 14576516239fSHong Zhang 14586516239fSHong Zhang for (i=0; i<n; i++) { 14596516239fSHong Zhang nzi = 0; 14606516239fSHong Zhang /* copy current row into linked list */ 14616516239fSHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 14626516239fSHong 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); 14636516239fSHong Zhang cols = aj + ai[r[i]]; 14646516239fSHong Zhang lnk[i] = -1; /* marker to indicate if diagonal exists */ 14656516239fSHong Zhang ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 14666516239fSHong Zhang nzi += nlnk; 14676516239fSHong Zhang 14686516239fSHong Zhang /* make sure diagonal entry is included */ 14696516239fSHong Zhang if (diagonal_fill && lnk[i] == -1) { 14706516239fSHong Zhang fm = n; 14716516239fSHong Zhang while (lnk[fm] < i) fm = lnk[fm]; 14726516239fSHong Zhang lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 14736516239fSHong Zhang lnk[fm] = i; 14746516239fSHong Zhang lnk_lvl[i] = 0; 14756516239fSHong Zhang nzi++; dcount++; 14766516239fSHong Zhang } 14776516239fSHong Zhang 14786516239fSHong Zhang /* add pivot rows into the active row */ 14796516239fSHong Zhang nzbd = 0; 14806516239fSHong Zhang prow = lnk[n]; 14816516239fSHong Zhang while (prow < i) { 14826516239fSHong Zhang nnz = bdiag[prow]; 14836516239fSHong Zhang cols = bj_ptr[prow] + nnz + 1; 14846516239fSHong Zhang cols_lvl = bjlvl_ptr[prow] + nnz + 1; 14856516239fSHong Zhang nnz = bi[prow+1] - bi[prow] - nnz - 1; 14866516239fSHong Zhang ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 14876516239fSHong Zhang nzi += nlnk; 14886516239fSHong Zhang prow = lnk[prow]; 14896516239fSHong Zhang nzbd++; 14906516239fSHong Zhang } 14916516239fSHong Zhang bdiag[i] = nzbd; 14926516239fSHong Zhang bi[i+1] = bi[i] + nzi; 14936516239fSHong Zhang 14946516239fSHong Zhang /* if free space is not available, make more free space */ 14956516239fSHong Zhang if (current_space->local_remaining<nzi) { 14966516239fSHong Zhang nnz = 2*nzi*(n - i); /* estimated and max additional space needed */ 14976516239fSHong Zhang ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 14986516239fSHong Zhang ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 14996516239fSHong Zhang reallocs++; 15006516239fSHong Zhang } 15016516239fSHong Zhang 15026516239fSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 15036516239fSHong Zhang ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 15046516239fSHong Zhang bj_ptr[i] = current_space->array; 15056516239fSHong Zhang bjlvl_ptr[i] = current_space_lvl->array; 15066516239fSHong Zhang 15076516239fSHong Zhang /* make sure the active row i has diagonal entry */ 15086516239fSHong Zhang if (*(bj_ptr[i]+bdiag[i]) != i) { 15096516239fSHong Zhang SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\ 15106516239fSHong Zhang try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i); 15116516239fSHong Zhang } 15126516239fSHong Zhang 15136516239fSHong Zhang current_space->array += nzi; 15146516239fSHong Zhang current_space->local_used += nzi; 15156516239fSHong Zhang current_space->local_remaining -= nzi; 15166516239fSHong Zhang current_space_lvl->array += nzi; 15176516239fSHong Zhang current_space_lvl->local_used += nzi; 15186516239fSHong Zhang current_space_lvl->local_remaining -= nzi; 15196516239fSHong Zhang } 15206516239fSHong Zhang 15216516239fSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 15226516239fSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 15236516239fSHong Zhang 15246516239fSHong Zhang /* destroy list of free space and other temporary arrays */ 15256516239fSHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 15266516239fSHong Zhang 15276516239fSHong Zhang /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */ 15286516239fSHong Zhang ierr = PetscFreeSpaceContiguous_newdatastruct(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 15296516239fSHong Zhang 15306516239fSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 15316516239fSHong Zhang ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 15326516239fSHong Zhang ierr = PetscFree(bj_ptr);CHKERRQ(ierr); 15336516239fSHong Zhang 15346516239fSHong Zhang #if defined(PETSC_USE_INFO) 15356516239fSHong Zhang { 15366516239fSHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 15376516239fSHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 15386516239fSHong Zhang ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 15396516239fSHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 15406516239fSHong Zhang ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 15416516239fSHong Zhang if (diagonal_fill) { 15426516239fSHong Zhang ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 15436516239fSHong Zhang } 15446516239fSHong Zhang } 15456516239fSHong Zhang #endif 15466516239fSHong Zhang 15476516239fSHong Zhang /* put together the new matrix */ 15486516239fSHong Zhang ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 15496516239fSHong Zhang ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 15506516239fSHong Zhang b = (Mat_SeqAIJ*)(fact)->data; 15516516239fSHong Zhang b->free_a = PETSC_TRUE; 15526516239fSHong Zhang b->free_ij = PETSC_TRUE; 15536516239fSHong Zhang b->singlemalloc = PETSC_FALSE; 15546516239fSHong Zhang ierr = PetscMalloc( (bi[2*n+1] )*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 15556516239fSHong Zhang b->j = bj; 15566516239fSHong Zhang b->i = bi; 15576516239fSHong Zhang b->diag = bdiag; 15586516239fSHong Zhang b->ilen = 0; 15596516239fSHong Zhang b->imax = 0; 15606516239fSHong Zhang b->row = isrow; 15616516239fSHong Zhang b->col = iscol; 15626516239fSHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 15636516239fSHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 15646516239fSHong Zhang b->icol = isicol; 15656516239fSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 15666516239fSHong Zhang /* In b structure: Free imax, ilen, old a, old j. 15676516239fSHong Zhang Allocate bdiag, solve_work, new a, new j */ 15686516239fSHong Zhang ierr = PetscLogObjectMemory(fact,bi[2*n+1] * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 15696516239fSHong Zhang b->maxnz = b->nz = bi[2*n+1] ; 15706516239fSHong Zhang (fact)->info.factor_mallocs = reallocs; 15716516239fSHong Zhang (fact)->info.fill_ratio_given = f; 15726516239fSHong Zhang (fact)->info.fill_ratio_needed = ((PetscReal)bi[2*n+1])/((PetscReal)ai[n]); 15736516239fSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct; 15746516239fSHong Zhang /* ierr = MatILUFactorSymbolic_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); */ 15756516239fSHong Zhang PetscFunctionReturn(0); 15766516239fSHong Zhang } 15776516239fSHong Zhang 15786516239fSHong Zhang #undef __FUNCT__ 15796516239fSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ" 15806516239fSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 15816516239fSHong Zhang { 15826516239fSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 15836516239fSHong Zhang IS isicol; 15846516239fSHong Zhang PetscErrorCode ierr; 15856516239fSHong Zhang const PetscInt *r,*ic; 15866516239fSHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,d; 15876516239fSHong Zhang PetscInt *bi,*cols,nnz,*cols_lvl; 15886516239fSHong Zhang PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 15896516239fSHong Zhang PetscInt i,levels,diagonal_fill; 15906516239fSHong Zhang PetscTruth col_identity,row_identity; 15916516239fSHong Zhang PetscReal f; 15926516239fSHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 15936516239fSHong Zhang PetscBT lnkbt; 15946516239fSHong Zhang PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 15956516239fSHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 15966516239fSHong Zhang PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 15976516239fSHong Zhang PetscTruth missing; 15986516239fSHong Zhang PetscTruth newdatastruct=PETSC_FALSE; 15996516239fSHong Zhang 16006516239fSHong Zhang PetscFunctionBegin; 16016516239fSHong Zhang ierr = PetscOptionsGetTruth(PETSC_NULL,"-ilu_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr); 16026516239fSHong Zhang if (newdatastruct){ 16036516239fSHong Zhang ierr = MatILUFactorSymbolic_SeqAIJ_newdatastruct(fact,A,isrow,iscol,info);CHKERRQ(ierr); 16046516239fSHong Zhang PetscFunctionReturn(0); 16056516239fSHong Zhang } 16066516239fSHong Zhang 16076516239fSHong 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); 16086516239fSHong Zhang f = info->fill; 16096516239fSHong Zhang levels = (PetscInt)info->levels; 16106516239fSHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 16116516239fSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 16126516239fSHong Zhang 16136516239fSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 16146516239fSHong Zhang ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 16156516239fSHong Zhang if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */ 1616f77e22a1SHong Zhang ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr); 1617c9c72f8fSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 1618c9c72f8fSHong Zhang 1619719d5645SBarry Smith fact->factor = MAT_FACTOR_ILU; 1620719d5645SBarry Smith (fact)->info.factor_mallocs = 0; 1621719d5645SBarry Smith (fact)->info.fill_ratio_given = info->fill; 1622719d5645SBarry Smith (fact)->info.fill_ratio_needed = 1.0; 1623719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 16248ef3462fSBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 162509f38230SBarry Smith if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 162608480c60SBarry Smith b->row = isrow; 162708480c60SBarry Smith b->col = iscol; 162882bf6240SBarry Smith b->icol = isicol; 1629719d5645SBarry Smith ierr = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1630c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1631c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1632719d5645SBarry Smith ierr = MatILUFactorSymbolic_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); 16333a40ed3dSBarry Smith PetscFunctionReturn(0); 163408480c60SBarry Smith } 163508480c60SBarry Smith 1636898183ecSLois Curfman McInnes ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1637898183ecSLois Curfman McInnes ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 16389e25ed09SBarry Smith 16399e25ed09SBarry Smith /* get new row pointers */ 16405a8e39fbSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 16415a8e39fbSHong Zhang bi[0] = 0; 16425a8e39fbSHong Zhang /* bdiag is location of diagonal in factor */ 16435a8e39fbSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 16445a8e39fbSHong Zhang bdiag[0] = 0; 16456cf997b0SBarry Smith 16465a8e39fbSHong Zhang ierr = PetscMalloc((2*n+1)*sizeof(PetscInt**),&bj_ptr);CHKERRQ(ierr); 16475a8e39fbSHong Zhang bjlvl_ptr = (PetscInt**)(bj_ptr + n); 16485a8e39fbSHong Zhang 16495a8e39fbSHong Zhang /* create a linked list for storing column indices of the active row */ 16505a8e39fbSHong Zhang nlnk = n + 1; 16515a8e39fbSHong Zhang ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 16525a8e39fbSHong Zhang 16535a8e39fbSHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 1654a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 16555a8e39fbSHong Zhang current_space = free_space; 1656a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 16575a8e39fbSHong Zhang current_space_lvl = free_space_lvl; 16585a8e39fbSHong Zhang 16595a8e39fbSHong Zhang for (i=0; i<n; i++) { 16605a8e39fbSHong Zhang nzi = 0; 16616cf997b0SBarry Smith /* copy current row into linked list */ 16625a8e39fbSHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 16633b4a8b6dSBarry 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); 16645a8e39fbSHong Zhang cols = aj + ai[r[i]]; 16655a8e39fbSHong Zhang lnk[i] = -1; /* marker to indicate if diagonal exists */ 16665a8e39fbSHong Zhang ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 16675a8e39fbSHong Zhang nzi += nlnk; 16686cf997b0SBarry Smith 16696cf997b0SBarry Smith /* make sure diagonal entry is included */ 16705a8e39fbSHong Zhang if (diagonal_fill && lnk[i] == -1) { 16716cf997b0SBarry Smith fm = n; 16725a8e39fbSHong Zhang while (lnk[fm] < i) fm = lnk[fm]; 16735a8e39fbSHong Zhang lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 16745a8e39fbSHong Zhang lnk[fm] = i; 16755a8e39fbSHong Zhang lnk_lvl[i] = 0; 16765a8e39fbSHong Zhang nzi++; dcount++; 16776cf997b0SBarry Smith } 16786cf997b0SBarry Smith 16795a8e39fbSHong Zhang /* add pivot rows into the active row */ 16805a8e39fbSHong Zhang nzbd = 0; 16815a8e39fbSHong Zhang prow = lnk[n]; 16825a8e39fbSHong Zhang while (prow < i) { 16835a8e39fbSHong Zhang nnz = bdiag[prow]; 16845a8e39fbSHong Zhang cols = bj_ptr[prow] + nnz + 1; 16855a8e39fbSHong Zhang cols_lvl = bjlvl_ptr[prow] + nnz + 1; 16865a8e39fbSHong Zhang nnz = bi[prow+1] - bi[prow] - nnz - 1; 16875a8e39fbSHong Zhang ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 16885a8e39fbSHong Zhang nzi += nlnk; 16895a8e39fbSHong Zhang prow = lnk[prow]; 16905a8e39fbSHong Zhang nzbd++; 169156beaf04SBarry Smith } 16925a8e39fbSHong Zhang bdiag[i] = nzbd; 16935a8e39fbSHong Zhang bi[i+1] = bi[i] + nzi; 1694ecf371e4SBarry Smith 16955a8e39fbSHong Zhang /* if free space is not available, make more free space */ 16965a8e39fbSHong Zhang if (current_space->local_remaining<nzi) { 16975a8e39fbSHong Zhang nnz = nzi*(n - i); /* estimated and max additional space needed */ 1698a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 1699a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 17005a8e39fbSHong Zhang reallocs++; 17015a8e39fbSHong Zhang } 1702ecf371e4SBarry Smith 17035a8e39fbSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 17045a8e39fbSHong Zhang ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 17055a8e39fbSHong Zhang bj_ptr[i] = current_space->array; 17065a8e39fbSHong Zhang bjlvl_ptr[i] = current_space_lvl->array; 17075a8e39fbSHong Zhang 17085a8e39fbSHong Zhang /* make sure the active row i has diagonal entry */ 17095a8e39fbSHong Zhang if (*(bj_ptr[i]+bdiag[i]) != i) { 171077431f27SBarry Smith SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\ 1711d7ee0231SBarry Smith try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i); 17126cf997b0SBarry Smith } 17135a8e39fbSHong Zhang 17145a8e39fbSHong Zhang current_space->array += nzi; 17155a8e39fbSHong Zhang current_space->local_used += nzi; 17165a8e39fbSHong Zhang current_space->local_remaining -= nzi; 17175a8e39fbSHong Zhang current_space_lvl->array += nzi; 17185a8e39fbSHong Zhang current_space_lvl->local_used += nzi; 17195a8e39fbSHong Zhang current_space_lvl->local_remaining -= nzi; 17209e25ed09SBarry Smith } 17215a8e39fbSHong Zhang 1722898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 1723898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 17245a8e39fbSHong Zhang 17255a8e39fbSHong Zhang /* destroy list of free space and other temporary arrays */ 17265a8e39fbSHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 17276516239fSHong Zhang ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */ 17285a8e39fbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 1729a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 17305a8e39fbSHong Zhang ierr = PetscFree(bj_ptr);CHKERRQ(ierr); 17319e25ed09SBarry Smith 17326cf91177SBarry Smith #if defined(PETSC_USE_INFO) 1733f64065bbSBarry Smith { 17345a8e39fbSHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 1735ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 1736ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 1737ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 1738ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 1739335d9088SBarry Smith if (diagonal_fill) { 1740ae15b995SBarry Smith ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 1741335d9088SBarry Smith } 1742f64065bbSBarry Smith } 174363ba0a88SBarry Smith #endif 1744bbb6d6a8SBarry Smith 17459e25ed09SBarry Smith /* put together the new matrix */ 174607dbf95fSBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 1747719d5645SBarry Smith ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 1748719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 1749e6b907acSBarry Smith b->free_a = PETSC_TRUE; 1750e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 17517c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 1752052f0c41SBarry Smith ierr = PetscMalloc( (bi[n] )*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 17535a8e39fbSHong Zhang b->j = bj; 17545a8e39fbSHong Zhang b->i = bi; 17555a8e39fbSHong Zhang for (i=0; i<n; i++) bdiag[i] += bi[i]; 17565a8e39fbSHong Zhang b->diag = bdiag; 1757416022c9SBarry Smith b->ilen = 0; 1758416022c9SBarry Smith b->imax = 0; 1759416022c9SBarry Smith b->row = isrow; 1760416022c9SBarry Smith b->col = iscol; 1761c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1762c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 176382bf6240SBarry Smith b->icol = isicol; 176487828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1765416022c9SBarry Smith /* In b structure: Free imax, ilen, old a, old j. 17665a8e39fbSHong Zhang Allocate bdiag, solve_work, new a, new j */ 1767719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 17685a8e39fbSHong Zhang b->maxnz = b->nz = bi[n] ; 1769719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 1770719d5645SBarry Smith (fact)->info.fill_ratio_given = f; 1771719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 1772719d5645SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 1773719d5645SBarry Smith ierr = MatILUFactorSymbolic_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); 17743a40ed3dSBarry Smith PetscFunctionReturn(0); 17759e25ed09SBarry Smith } 1776d5d45c9bSBarry Smith 17777c4f633dSBarry Smith #include "../src/mat/impls/sbaij/seq/sbaij.h" 1778a6175056SHong Zhang #undef __FUNCT__ 1779a6175056SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ" 17800481f469SBarry Smith PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 1781a6175056SHong Zhang { 1782719d5645SBarry Smith Mat C = B; 17830968510aSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 17842ed133dbSHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 17859bfd6278SHong Zhang IS ip=b->row,iip = b->icol; 17862ed133dbSHong Zhang PetscErrorCode ierr; 17875d0c19d7SBarry Smith const PetscInt *rip,*riip; 17885d0c19d7SBarry Smith PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol; 17892ed133dbSHong Zhang PetscInt *ai=a->i,*aj=a->j; 17902ed133dbSHong Zhang PetscInt k,jmin,jmax,*jl,*il,col,nexti,ili,nz; 1791622e2028SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 1792540703acSHong Zhang PetscReal zeropivot,rs,shiftnz; 1793fbf22428SSatish Balay PetscReal shiftpd; 1794540703acSHong Zhang ChShift_Ctx sctx; 1795540703acSHong Zhang PetscInt newshift; 1796db4efbfdSBarry Smith PetscTruth perm_identity; 1797d5d45c9bSBarry Smith 1798a6175056SHong Zhang PetscFunctionBegin; 1799117f1891Stmunson 1800540703acSHong Zhang shiftnz = info->shiftnz; 1801540703acSHong Zhang shiftpd = info->shiftpd; 1802ee45ca4aSHong Zhang zeropivot = info->zeropivot; 1803ee45ca4aSHong Zhang 18042ed133dbSHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 18059bfd6278SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 18062ed133dbSHong Zhang 18072ed133dbSHong Zhang /* initialization */ 18082ed133dbSHong Zhang nz = (2*mbs+1)*sizeof(PetscInt)+mbs*sizeof(MatScalar); 18092ed133dbSHong Zhang ierr = PetscMalloc(nz,&il);CHKERRQ(ierr); 18102ed133dbSHong Zhang jl = il + mbs; 18112ed133dbSHong Zhang rtmp = (MatScalar*)(jl + mbs); 18122ed133dbSHong Zhang 1813540703acSHong Zhang sctx.shift_amount = 0; 1814540703acSHong Zhang sctx.nshift = 0; 18152ed133dbSHong Zhang do { 1816540703acSHong Zhang sctx.chshift = PETSC_FALSE; 18172ed133dbSHong Zhang for (i=0; i<mbs; i++) { 18182ed133dbSHong Zhang rtmp[i] = 0.0; jl[i] = mbs; il[0] = 0; 1819a6175056SHong Zhang } 18202ed133dbSHong Zhang 18212ed133dbSHong Zhang for (k = 0; k<mbs; k++){ 1822c05c3958SHong Zhang bval = ba + bi[k]; 18232ed133dbSHong Zhang /* initialize k-th row by the perm[k]-th row of A */ 18242ed133dbSHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 18252ed133dbSHong Zhang for (j = jmin; j < jmax; j++){ 18269bfd6278SHong Zhang col = riip[aj[j]]; 18272ed133dbSHong Zhang if (col >= k){ /* only take upper triangular entry */ 18282ed133dbSHong Zhang rtmp[col] = aa[j]; 1829c05c3958SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 18302ed133dbSHong Zhang } 18312ed133dbSHong Zhang } 183239e3d630SHong Zhang /* shift the diagonal of the matrix */ 1833540703acSHong Zhang if (sctx.nshift) rtmp[k] += sctx.shift_amount; 18342ed133dbSHong Zhang 18352ed133dbSHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 18362ed133dbSHong Zhang dk = rtmp[k]; 18372ed133dbSHong Zhang i = jl[k]; /* first row to be added to k_th row */ 18382ed133dbSHong Zhang 18392ed133dbSHong Zhang while (i < k){ 18402ed133dbSHong Zhang nexti = jl[i]; /* next row to be added to k_th row */ 18412ed133dbSHong Zhang 18422ed133dbSHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 18432ed133dbSHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 18442ed133dbSHong Zhang uikdi = - ba[ili]*ba[bi[i]]; /* diagonal(k) */ 18452ed133dbSHong Zhang dk += uikdi*ba[ili]; 18462ed133dbSHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 18472ed133dbSHong Zhang 18482ed133dbSHong Zhang /* add multiple of row i to k-th row */ 18492ed133dbSHong Zhang jmin = ili + 1; jmax = bi[i+1]; 18502ed133dbSHong Zhang if (jmin < jmax){ 18512ed133dbSHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 18522ed133dbSHong Zhang /* update il and jl for row i */ 18532ed133dbSHong Zhang il[i] = jmin; 18542ed133dbSHong Zhang j = bj[jmin]; jl[i] = jl[j]; jl[j] = i; 18552ed133dbSHong Zhang } 18562ed133dbSHong Zhang i = nexti; 18572ed133dbSHong Zhang } 18582ed133dbSHong Zhang 18590697b6caSHong Zhang /* shift the diagonals when zero pivot is detected */ 18600697b6caSHong Zhang /* compute rs=sum of abs(off-diagonal) */ 18610697b6caSHong Zhang rs = 0.0; 18623c9e8598SHong Zhang jmin = bi[k]+1; 18633c9e8598SHong Zhang nz = bi[k+1] - jmin; 18643c9e8598SHong Zhang bcol = bj + jmin; 18653c9e8598SHong Zhang while (nz--){ 186689f845c8SHong Zhang rs += PetscAbsScalar(rtmp[*bcol]); 186789f845c8SHong Zhang bcol++; 18683c9e8598SHong Zhang } 1869540703acSHong Zhang 1870540703acSHong Zhang sctx.rs = rs; 1871540703acSHong Zhang sctx.pv = dk; 18725b5bc046SBarry Smith ierr = MatCholeskyCheckShift_inline(info,sctx,k,newshift);CHKERRQ(ierr); 1873117f1891Stmunson 1874117f1891Stmunson if (newshift == 1) { 1875117f1891Stmunson if (!sctx.shift_amount) { 1876117f1891Stmunson sctx.shift_amount = 1e-5; 1877117f1891Stmunson } 1878117f1891Stmunson break; 1879117f1891Stmunson } 18803c9e8598SHong Zhang 18813c9e8598SHong Zhang /* copy data into U(k,:) */ 188239e3d630SHong Zhang ba[bi[k]] = 1.0/dk; /* U(k,k) */ 188339e3d630SHong Zhang jmin = bi[k]+1; jmax = bi[k+1]; 188439e3d630SHong Zhang if (jmin < jmax) { 188539e3d630SHong Zhang for (j=jmin; j<jmax; j++){ 188639e3d630SHong Zhang col = bj[j]; ba[j] = rtmp[col]; rtmp[col] = 0.0; 18873c9e8598SHong Zhang } 188839e3d630SHong Zhang /* add the k-th row into il and jl */ 18893c9e8598SHong Zhang il[k] = jmin; 18903c9e8598SHong Zhang i = bj[jmin]; jl[k] = jl[i]; jl[i] = k; 18913c9e8598SHong Zhang } 18923c9e8598SHong Zhang } 1893540703acSHong Zhang } while (sctx.chshift); 18943c9e8598SHong Zhang ierr = PetscFree(il);CHKERRQ(ierr); 18953c9e8598SHong Zhang 189639e3d630SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 18979bfd6278SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 1898db4efbfdSBarry Smith 1899db4efbfdSBarry Smith ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 1900db4efbfdSBarry Smith if (perm_identity){ 1901719d5645SBarry Smith (B)->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering; 1902719d5645SBarry Smith (B)->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering; 1903719d5645SBarry Smith (B)->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering; 1904719d5645SBarry Smith (B)->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering; 1905db4efbfdSBarry Smith } else { 1906719d5645SBarry Smith (B)->ops->solve = MatSolve_SeqSBAIJ_1; 1907719d5645SBarry Smith (B)->ops->solvetranspose = MatSolve_SeqSBAIJ_1; 1908719d5645SBarry Smith (B)->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1; 1909719d5645SBarry Smith (B)->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1; 1910db4efbfdSBarry Smith } 1911db4efbfdSBarry Smith 19123c9e8598SHong Zhang C->assembled = PETSC_TRUE; 19133c9e8598SHong Zhang C->preallocated = PETSC_TRUE; 1914d0f46423SBarry Smith ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 1915540703acSHong Zhang if (sctx.nshift){ 1916540703acSHong Zhang if (shiftnz) { 19171e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 1918540703acSHong Zhang } else if (shiftpd) { 19191e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 19200697b6caSHong Zhang } 19213c9e8598SHong Zhang } 19223c9e8598SHong Zhang PetscFunctionReturn(0); 19233c9e8598SHong Zhang } 1924a6175056SHong Zhang 1925a6175056SHong Zhang #undef __FUNCT__ 1926a6175056SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ" 19270481f469SBarry Smith PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 1928a6175056SHong Zhang { 19290968510aSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1930ed59401aSHong Zhang Mat_SeqSBAIJ *b; 1931dfbe8321SBarry Smith PetscErrorCode ierr; 193258ebbce7SBarry Smith PetscTruth perm_identity,missing; 19335d0c19d7SBarry Smith PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui; 19345d0c19d7SBarry Smith const PetscInt *rip,*riip; 1935ed59401aSHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 193658ebbce7SBarry Smith PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 19375a8e39fbSHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 1938ed59401aSHong Zhang PetscReal fill=info->fill,levels=info->levels; 1939a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1940a1a86e44SBarry Smith PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 19417a48dd6fSHong Zhang PetscBT lnkbt; 1942b635d36bSHong Zhang IS iperm; 1943a6175056SHong Zhang 1944a6175056SHong Zhang PetscFunctionBegin; 1945d0f46423SBarry 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); 194658ebbce7SBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 194758ebbce7SBarry Smith if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 1948653a6975SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 1949b635d36bSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 19507a48dd6fSHong Zhang 195139e3d630SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 195239e3d630SHong Zhang ui[0] = 0; 195339e3d630SHong Zhang 1954b635d36bSHong Zhang /* ICC(0) without matrix ordering: simply copies fill pattern */ 1955622e2028SHong Zhang if (!levels && perm_identity) { 195658ebbce7SBarry Smith 1957ed59401aSHong Zhang for (i=0; i<am; i++) { 195839e3d630SHong Zhang ui[i+1] = ui[i] + ai[i+1] - a->diag[i]; 1959ed59401aSHong Zhang } 196039e3d630SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 196139e3d630SHong Zhang cols = uj; 1962ed59401aSHong Zhang for (i=0; i<am; i++) { 1963ed59401aSHong Zhang aj = a->j + a->diag[i]; 196439e3d630SHong Zhang ncols = ui[i+1] - ui[i]; 196539e3d630SHong Zhang for (j=0; j<ncols; j++) *cols++ = *aj++; 1966ed59401aSHong Zhang } 196739e3d630SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 1968b635d36bSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 1969b635d36bSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 1970b635d36bSHong Zhang 19717a48dd6fSHong Zhang /* initialization */ 19725a8e39fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 19737a48dd6fSHong Zhang 19747a48dd6fSHong Zhang /* jl: linked list for storing indices of the pivot rows 19757a48dd6fSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 19761393bc97SHong Zhang ierr = PetscMalloc((2*am+1)*sizeof(PetscInt)+2*am*sizeof(PetscInt**),&jl);CHKERRQ(ierr); 19777a48dd6fSHong Zhang il = jl + am; 19787a48dd6fSHong Zhang uj_ptr = (PetscInt**)(il + am); 19797a48dd6fSHong Zhang uj_lvl_ptr = (PetscInt**)(uj_ptr + am); 19807a48dd6fSHong Zhang for (i=0; i<am; i++){ 19817a48dd6fSHong Zhang jl[i] = am; il[i] = 0; 19827a48dd6fSHong Zhang } 19837a48dd6fSHong Zhang 19847a48dd6fSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 19857a48dd6fSHong Zhang nlnk = am + 1; 19862ed133dbSHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 19877a48dd6fSHong Zhang 19887a48dd6fSHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 1989a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 19907a48dd6fSHong Zhang current_space = free_space; 1991a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr); 19927a48dd6fSHong Zhang current_space_lvl = free_space_lvl; 19937a48dd6fSHong Zhang 19947a48dd6fSHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 19957a48dd6fSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 19967a48dd6fSHong Zhang nzk = 0; 1997622e2028SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 19983b4a8b6dSBarry 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); 1999622e2028SHong Zhang ncols_upper = 0; 2000622e2028SHong Zhang for (j=0; j<ncols; j++){ 2001b635d36bSHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 2002b635d36bSHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 20035a8e39fbSHong Zhang ajtmp[ncols_upper] = i; 2004622e2028SHong Zhang ncols_upper++; 2005622e2028SHong Zhang } 2006622e2028SHong Zhang } 2007b635d36bSHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 20087a48dd6fSHong Zhang nzk += nlnk; 20097a48dd6fSHong Zhang 20107a48dd6fSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 20117a48dd6fSHong Zhang prow = jl[k]; /* 1st pivot row */ 20127a48dd6fSHong Zhang 20137a48dd6fSHong Zhang while (prow < k){ 20147a48dd6fSHong Zhang nextprow = jl[prow]; 20157a48dd6fSHong Zhang 20167a48dd6fSHong Zhang /* merge prow into k-th row */ 20177a48dd6fSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 20187a48dd6fSHong Zhang jmax = ui[prow+1]; 20197a48dd6fSHong Zhang ncols = jmax-jmin; 2020ed59401aSHong Zhang i = jmin - ui[prow]; 2021ed59401aSHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 20225a8e39fbSHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 20235a8e39fbSHong Zhang j = *(uj - 1); 20245a8e39fbSHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 20257a48dd6fSHong Zhang nzk += nlnk; 20267a48dd6fSHong Zhang 20277a48dd6fSHong Zhang /* update il and jl for prow */ 20287a48dd6fSHong Zhang if (jmin < jmax){ 20297a48dd6fSHong Zhang il[prow] = jmin; 20307a48dd6fSHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 20317a48dd6fSHong Zhang } 20327a48dd6fSHong Zhang prow = nextprow; 20337a48dd6fSHong Zhang } 20347a48dd6fSHong Zhang 20357a48dd6fSHong Zhang /* if free space is not available, make more free space */ 20367a48dd6fSHong Zhang if (current_space->local_remaining<nzk) { 20377a48dd6fSHong Zhang i = am - k + 1; /* num of unfactored rows */ 20387a48dd6fSHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 2039a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 2040a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 20417a48dd6fSHong Zhang reallocs++; 20427a48dd6fSHong Zhang } 20437a48dd6fSHong Zhang 20447a48dd6fSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 2045b635d36bSHong Zhang if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 20462ed133dbSHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 20477a48dd6fSHong Zhang 20487a48dd6fSHong Zhang /* add the k-th row into il and jl */ 204939e3d630SHong Zhang if (nzk > 1){ 20507a48dd6fSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 20517a48dd6fSHong Zhang jl[k] = jl[i]; jl[i] = k; 20527a48dd6fSHong Zhang il[k] = ui[k] + 1; 20537a48dd6fSHong Zhang } 20547a48dd6fSHong Zhang uj_ptr[k] = current_space->array; 20557a48dd6fSHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 20567a48dd6fSHong Zhang 20577a48dd6fSHong Zhang current_space->array += nzk; 20587a48dd6fSHong Zhang current_space->local_used += nzk; 20597a48dd6fSHong Zhang current_space->local_remaining -= nzk; 20607a48dd6fSHong Zhang 20617a48dd6fSHong Zhang current_space_lvl->array += nzk; 20627a48dd6fSHong Zhang current_space_lvl->local_used += nzk; 20637a48dd6fSHong Zhang current_space_lvl->local_remaining -= nzk; 20647a48dd6fSHong Zhang 20657a48dd6fSHong Zhang ui[k+1] = ui[k] + nzk; 20667a48dd6fSHong Zhang } 20677a48dd6fSHong Zhang 20686cf91177SBarry Smith #if defined(PETSC_USE_INFO) 20697a48dd6fSHong Zhang if (ai[am] != 0) { 207039e3d630SHong Zhang PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]); 2071ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 2072ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2073ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 20747a48dd6fSHong Zhang } else { 2075ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 20767a48dd6fSHong Zhang } 207763ba0a88SBarry Smith #endif 20787a48dd6fSHong Zhang 20797a48dd6fSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 2080b635d36bSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 20817a48dd6fSHong Zhang ierr = PetscFree(jl);CHKERRQ(ierr); 20825a8e39fbSHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 20837a48dd6fSHong Zhang 20847a48dd6fSHong Zhang /* destroy list of free space and other temporary array(s) */ 20857a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2086a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 20872ed133dbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2088a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 20897a48dd6fSHong Zhang 209039e3d630SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 209139e3d630SHong Zhang 20927a48dd6fSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 20937a48dd6fSHong Zhang 2094719d5645SBarry Smith b = (Mat_SeqSBAIJ*)(fact)->data; 20957a48dd6fSHong Zhang b->singlemalloc = PETSC_FALSE; 20967a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 20977a48dd6fSHong Zhang b->j = uj; 20987a48dd6fSHong Zhang b->i = ui; 20997a48dd6fSHong Zhang b->diag = 0; 21007a48dd6fSHong Zhang b->ilen = 0; 21017a48dd6fSHong Zhang b->imax = 0; 21027a48dd6fSHong Zhang b->row = perm; 2103b635d36bSHong Zhang b->col = perm; 2104b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2105b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2106b635d36bSHong Zhang b->icol = iperm; 21077a48dd6fSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 21087a48dd6fSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2109719d5645SBarry Smith ierr = PetscLogObjectMemory((fact),(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 21107a48dd6fSHong Zhang b->maxnz = b->nz = ui[am]; 2111e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2112e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 21137a48dd6fSHong Zhang 2114719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 2115719d5645SBarry Smith (fact)->info.fill_ratio_given = fill; 21167a48dd6fSHong Zhang if (ai[am] != 0) { 2117719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 21187a48dd6fSHong Zhang } else { 2119719d5645SBarry Smith (fact)->info.fill_ratio_needed = 0.0; 21207a48dd6fSHong Zhang } 2121719d5645SBarry Smith (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 2122a6175056SHong Zhang PetscFunctionReturn(0); 2123a6175056SHong Zhang } 2124d5d45c9bSBarry Smith 2125f76d2b81SHong Zhang #undef __FUNCT__ 2126f76d2b81SHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ" 21270481f469SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2128f76d2b81SHong Zhang { 2129f76d2b81SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 213010c27e3dSHong Zhang Mat_SeqSBAIJ *b; 2131dfbe8321SBarry Smith PetscErrorCode ierr; 2132f76d2b81SHong Zhang PetscTruth perm_identity; 213310c27e3dSHong Zhang PetscReal fill = info->fill; 21345d0c19d7SBarry Smith const PetscInt *rip,*riip; 21355d0c19d7SBarry Smith PetscInt i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 213610c27e3dSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 21372ed133dbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr; 2138a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 213910c27e3dSHong Zhang PetscBT lnkbt; 21402ed133dbSHong Zhang IS iperm; 2141f76d2b81SHong Zhang 2142f76d2b81SHong Zhang PetscFunctionBegin; 2143d0f46423SBarry 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); 21442ed133dbSHong Zhang /* check whether perm is the identity mapping */ 2145f76d2b81SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 21462ed133dbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 21472ed133dbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 21489bfd6278SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 214910c27e3dSHong Zhang 215010c27e3dSHong Zhang /* initialization */ 21511393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 215210c27e3dSHong Zhang ui[0] = 0; 215310c27e3dSHong Zhang 215410c27e3dSHong Zhang /* jl: linked list for storing indices of the pivot rows 21551393bc97SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 21561393bc97SHong Zhang ierr = PetscMalloc((3*am+1)*sizeof(PetscInt)+am*sizeof(PetscInt**),&jl);CHKERRQ(ierr); 21571393bc97SHong Zhang il = jl + am; 21581393bc97SHong Zhang cols = il + am; 21591393bc97SHong Zhang ui_ptr = (PetscInt**)(cols + am); 21601393bc97SHong Zhang for (i=0; i<am; i++){ 21611393bc97SHong Zhang jl[i] = am; il[i] = 0; 2162f76d2b81SHong Zhang } 216310c27e3dSHong Zhang 216410c27e3dSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 21651393bc97SHong Zhang nlnk = am + 1; 21661393bc97SHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 216710c27e3dSHong Zhang 21681393bc97SHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 2169a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 217010c27e3dSHong Zhang current_space = free_space; 217110c27e3dSHong Zhang 21721393bc97SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 217310c27e3dSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 217410c27e3dSHong Zhang nzk = 0; 21752ed133dbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 21763b4a8b6dSBarry 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); 21772ed133dbSHong Zhang ncols_upper = 0; 21782ed133dbSHong Zhang for (j=0; j<ncols; j++){ 21799bfd6278SHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 21802ed133dbSHong Zhang if (i >= k){ /* only take upper triangular entry */ 21812ed133dbSHong Zhang cols[ncols_upper] = i; 21822ed133dbSHong Zhang ncols_upper++; 21832ed133dbSHong Zhang } 21842ed133dbSHong Zhang } 21851393bc97SHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 218610c27e3dSHong Zhang nzk += nlnk; 218710c27e3dSHong Zhang 218810c27e3dSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 218910c27e3dSHong Zhang prow = jl[k]; /* 1st pivot row */ 219010c27e3dSHong Zhang 219110c27e3dSHong Zhang while (prow < k){ 219210c27e3dSHong Zhang nextprow = jl[prow]; 219310c27e3dSHong Zhang /* merge prow into k-th row */ 21941393bc97SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 219510c27e3dSHong Zhang jmax = ui[prow+1]; 219610c27e3dSHong Zhang ncols = jmax-jmin; 21971393bc97SHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 21985a8e39fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 219910c27e3dSHong Zhang nzk += nlnk; 220010c27e3dSHong Zhang 220110c27e3dSHong Zhang /* update il and jl for prow */ 220210c27e3dSHong Zhang if (jmin < jmax){ 220310c27e3dSHong Zhang il[prow] = jmin; 22042ed133dbSHong Zhang j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow; 220510c27e3dSHong Zhang } 220610c27e3dSHong Zhang prow = nextprow; 220710c27e3dSHong Zhang } 220810c27e3dSHong Zhang 220910c27e3dSHong Zhang /* if free space is not available, make more free space */ 221010c27e3dSHong Zhang if (current_space->local_remaining<nzk) { 22111393bc97SHong Zhang i = am - k + 1; /* num of unfactored rows */ 221210c27e3dSHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 2213a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 221410c27e3dSHong Zhang reallocs++; 221510c27e3dSHong Zhang } 221610c27e3dSHong Zhang 221710c27e3dSHong Zhang /* copy data into free space, then initialize lnk */ 22181393bc97SHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 221910c27e3dSHong Zhang 222010c27e3dSHong Zhang /* add the k-th row into il and jl */ 222110c27e3dSHong Zhang if (nzk-1 > 0){ 22221393bc97SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 222310c27e3dSHong Zhang jl[k] = jl[i]; jl[i] = k; 222410c27e3dSHong Zhang il[k] = ui[k] + 1; 222510c27e3dSHong Zhang } 22262ed133dbSHong Zhang ui_ptr[k] = current_space->array; 222710c27e3dSHong Zhang current_space->array += nzk; 222810c27e3dSHong Zhang current_space->local_used += nzk; 222910c27e3dSHong Zhang current_space->local_remaining -= nzk; 223010c27e3dSHong Zhang 223110c27e3dSHong Zhang ui[k+1] = ui[k] + nzk; 223210c27e3dSHong Zhang } 223310c27e3dSHong Zhang 22346cf91177SBarry Smith #if defined(PETSC_USE_INFO) 22351393bc97SHong Zhang if (ai[am] != 0) { 22361393bc97SHong Zhang PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]); 2237ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 2238ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2239ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 224010c27e3dSHong Zhang } else { 2241ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 224210c27e3dSHong Zhang } 224363ba0a88SBarry Smith #endif 224410c27e3dSHong Zhang 224510c27e3dSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 22469bfd6278SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 224710c27e3dSHong Zhang ierr = PetscFree(jl);CHKERRQ(ierr); 224810c27e3dSHong Zhang 224910c27e3dSHong Zhang /* destroy list of free space and other temporary array(s) */ 22501393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2251a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 225210c27e3dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 225310c27e3dSHong Zhang 225410c27e3dSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 225510c27e3dSHong Zhang 2256719d5645SBarry Smith b = (Mat_SeqSBAIJ*)(fact)->data; 225710c27e3dSHong Zhang b->singlemalloc = PETSC_FALSE; 2258e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2259e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 22601393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 226110c27e3dSHong Zhang b->j = uj; 226210c27e3dSHong Zhang b->i = ui; 226310c27e3dSHong Zhang b->diag = 0; 226410c27e3dSHong Zhang b->ilen = 0; 226510c27e3dSHong Zhang b->imax = 0; 226610c27e3dSHong Zhang b->row = perm; 22679bfd6278SHong Zhang b->col = perm; 22689bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 22699bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 22709bfd6278SHong Zhang b->icol = iperm; 227110c27e3dSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 22721393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2273719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 22741393bc97SHong Zhang b->maxnz = b->nz = ui[am]; 227510c27e3dSHong Zhang 2276719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 2277719d5645SBarry Smith (fact)->info.fill_ratio_given = fill; 22781393bc97SHong Zhang if (ai[am] != 0) { 2279719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 228010c27e3dSHong Zhang } else { 2281719d5645SBarry Smith (fact)->info.fill_ratio_needed = 0.0; 228210c27e3dSHong Zhang } 2283719d5645SBarry Smith (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 2284f76d2b81SHong Zhang PetscFunctionReturn(0); 2285f76d2b81SHong Zhang } 2286599ef60dSHong Zhang 2287599ef60dSHong Zhang #undef __FUNCT__ 22881d098868SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_iludt" 22891d098868SHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_iludt(Mat A,Vec bb,Vec xx) 2290599ef60dSHong Zhang { 22911d098868SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 22921d098868SHong Zhang PetscErrorCode ierr; 22931d098868SHong Zhang PetscInt n = A->rmap->n; 22948fc3a347SHong Zhang const PetscInt *ai = a->i,*aj = a->j,*vi; 22951d098868SHong Zhang PetscScalar *x,sum; 22961d098868SHong Zhang const PetscScalar *b; 22971d098868SHong Zhang const MatScalar *aa = a->a,*v; 22988fc3a347SHong Zhang PetscInt i,nz; 22991d098868SHong Zhang 23001d098868SHong Zhang PetscFunctionBegin; 23011d098868SHong Zhang if (!n) PetscFunctionReturn(0); 23021d098868SHong Zhang 23031d098868SHong Zhang ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 23041d098868SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 23051d098868SHong Zhang 23061d098868SHong Zhang /* forward solve the lower triangular */ 23071d098868SHong Zhang x[0] = b[0]; 23081d098868SHong Zhang v = aa; 23091d098868SHong Zhang vi = aj; 23101d098868SHong Zhang for (i=1; i<n; i++) { 23111d098868SHong Zhang nz = ai[i+1] - ai[i]; 23121d098868SHong Zhang sum = b[i]; 2313003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 2314e09e08ccSBarry Smith /* while (nz--) sum -= *v++ * x[*vi++];*/ 2315e0e43300SBarry Smith v += nz; 2316e0e43300SBarry Smith vi += nz; 23171d098868SHong Zhang x[i] = sum; 23181d098868SHong Zhang } 23191d098868SHong Zhang 23201d098868SHong Zhang /* backward solve the upper triangular */ 23218fc3a347SHong Zhang v = aa + ai[n+1]; 23228fc3a347SHong Zhang vi = aj + ai[n+1]; 23231d098868SHong Zhang for (i=n-1; i>=0; i--){ 23248fc3a347SHong Zhang nz = ai[2*n-i +1] - ai[2*n-i]-1; 23251d098868SHong Zhang sum = x[i]; 2326003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 2327e09e08ccSBarry Smith /* while (nz--) sum -= *v++ * x[*vi++]; */ 2328e0e43300SBarry Smith v += nz; 23298fc3a347SHong Zhang vi += nz; vi++; 2330fe31b70dSHong Zhang x[i] = *v++ *sum; /* x[i]=aa[adiag[i]]*sum; v++; */ 23311d098868SHong Zhang } 23321d098868SHong Zhang 23331d098868SHong Zhang ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 23341d098868SHong Zhang ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 23351d098868SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 23361d098868SHong Zhang PetscFunctionReturn(0); 23371d098868SHong Zhang } 23381d098868SHong Zhang 23391d098868SHong Zhang #undef __FUNCT__ 23401d098868SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_iludt" 23411d098868SHong Zhang PetscErrorCode MatSolve_SeqAIJ_iludt(Mat A,Vec bb,Vec xx) 23421d098868SHong Zhang { 23431d098868SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 23441d098868SHong Zhang IS iscol = a->col,isrow = a->row; 23451d098868SHong Zhang PetscErrorCode ierr; 23461d098868SHong Zhang PetscInt i,n=A->rmap->n,*vi,*ai = a->i,*aj = a->j,*adiag=a->diag; 23471d098868SHong Zhang PetscInt nz; 23481d098868SHong Zhang const PetscInt *rout,*cout,*r,*c; 23491d098868SHong Zhang PetscScalar *x,*tmp,*tmps; 23501d098868SHong Zhang const PetscScalar *b; 23511d098868SHong Zhang const MatScalar *aa = a->a,*v; 23521d098868SHong Zhang 23531d098868SHong Zhang PetscFunctionBegin; 23541d098868SHong Zhang if (!n) PetscFunctionReturn(0); 23551d098868SHong Zhang 23561d098868SHong Zhang ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 23571d098868SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 23581d098868SHong Zhang tmp = a->solve_work; 23591d098868SHong Zhang 23601d098868SHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 23611d098868SHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 23621d098868SHong Zhang 23631d098868SHong Zhang /* forward solve the lower triangular */ 23641d098868SHong Zhang tmp[0] = b[*r++]; 23651d098868SHong Zhang tmps = tmp; 23661d098868SHong Zhang v = aa; 23671d098868SHong Zhang vi = aj; 23681d098868SHong Zhang for (i=1; i<n; i++) { 23691d098868SHong Zhang nz = ai[i+1] - ai[i]; 23701d098868SHong Zhang tmp[i] = b[*r++]; 2371003131ecSBarry Smith PetscSparseDenseMinusDot(tmp[i],tmps,v,vi,nz); 23721d098868SHong Zhang v += nz; vi += nz; 23731d098868SHong Zhang } 23741d098868SHong Zhang 23751d098868SHong Zhang /* backward solve the upper triangular */ 23761d098868SHong Zhang v = aa + adiag[n] + 1; 23771d098868SHong Zhang vi = aj + adiag[n] + 1; 23781d098868SHong Zhang for (i=n-1; i>=0; i--){ 23791d098868SHong Zhang nz = adiag[i] - adiag[i+1] - 1; 2380003131ecSBarry Smith PetscSparseDenseMinusDot(tmp[i],tmps,v,vi,nz); 23811d098868SHong Zhang x[*c--] = tmp[i] = tmp[i]*aa[adiag[i]]; 23821d098868SHong Zhang v += nz+1; vi += nz+1; 23831d098868SHong Zhang } 23841d098868SHong Zhang 23851d098868SHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 23861d098868SHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 23871d098868SHong Zhang ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 23881d098868SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 23891d098868SHong Zhang ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr); 23901d098868SHong Zhang PetscFunctionReturn(0); 23911d098868SHong Zhang } 23921d098868SHong Zhang 23931d098868SHong Zhang #undef __FUNCT__ 23941d098868SHong Zhang #define __FUNCT__ "MatILUDTFactor_SeqAIJ" 23951d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact) 23961d098868SHong Zhang { 23971d098868SHong Zhang Mat B = *fact; 2398599ef60dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b; 2399599ef60dSHong Zhang IS isicol; 2400599ef60dSHong Zhang PetscErrorCode ierr; 24011d098868SHong Zhang const PetscInt *r,*ic; 24021d098868SHong Zhang PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag; 2403dc3a2fd3SHong Zhang PetscInt *bi,*bj,*bdiag,*bdiag_rev; 2404f61a948aSLisandro Dalcin PetscInt row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au; 24051d098868SHong Zhang PetscInt nlnk,*lnk; 24061d098868SHong Zhang PetscBT lnkbt; 24071d098868SHong Zhang PetscTruth row_identity,icol_identity,both_identity; 24088aee2decSHong Zhang MatScalar *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp; 24091d098868SHong Zhang const PetscInt *ics; 24101d098868SHong Zhang PetscInt j,nz,*pj,*bjtmp,k,ncut,*jtmp; 2411f61a948aSLisandro Dalcin PetscReal dt=info->dt,dtcol=info->dtcol,shift=info->shiftinblocks; 2412f61a948aSLisandro Dalcin PetscInt dtcount=(PetscInt)info->dtcount,nnz_max; 2413599ef60dSHong Zhang PetscTruth missing; 2414599ef60dSHong Zhang 2415599ef60dSHong Zhang PetscFunctionBegin; 2416f61a948aSLisandro Dalcin 2417f61a948aSLisandro Dalcin if (dt == PETSC_DEFAULT) dt = 0.005; 2418f61a948aSLisandro Dalcin if (dtcol == PETSC_DEFAULT) dtcol = 0.01; /* XXX unused! */ 2419f61a948aSLisandro Dalcin if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax); 2420f61a948aSLisandro Dalcin 24211d098868SHong Zhang /* ------- symbolic factorization, can be reused ---------*/ 24221d098868SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 24231d098868SHong Zhang if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 24241d098868SHong Zhang adiag=a->diag; 2425599ef60dSHong Zhang 2426599ef60dSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 2427599ef60dSHong Zhang 2428599ef60dSHong Zhang /* bdiag is location of diagonal in factor */ 2429a6226a1aSHong Zhang ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 2430a6226a1aSHong Zhang bdiag_rev = bdiag + n+1; 2431599ef60dSHong Zhang 24321d098868SHong Zhang /* allocate row pointers bi */ 24338fc3a347SHong Zhang ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 24341d098868SHong Zhang 24351d098868SHong Zhang /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */ 2436393d3a68SHong Zhang if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */ 24371d098868SHong Zhang nnz_max = ai[n]+2*n*dtcount+2; 24388fc3a347SHong Zhang 24398fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 24408fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(MatScalar),&ba);CHKERRQ(ierr); 2441599ef60dSHong Zhang 2442599ef60dSHong Zhang /* put together the new matrix */ 2443599ef60dSHong Zhang ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 2444599ef60dSHong Zhang ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 2445599ef60dSHong Zhang b = (Mat_SeqAIJ*)(B)->data; 2446599ef60dSHong Zhang b->free_a = PETSC_TRUE; 2447599ef60dSHong Zhang b->free_ij = PETSC_TRUE; 2448599ef60dSHong Zhang b->singlemalloc = PETSC_FALSE; 2449599ef60dSHong Zhang b->a = ba; 2450599ef60dSHong Zhang b->j = bj; 2451599ef60dSHong Zhang b->i = bi; 2452599ef60dSHong Zhang b->diag = bdiag; 2453599ef60dSHong Zhang b->ilen = 0; 2454599ef60dSHong Zhang b->imax = 0; 2455599ef60dSHong Zhang b->row = isrow; 2456599ef60dSHong Zhang b->col = iscol; 2457599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 2458599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 2459599ef60dSHong Zhang b->icol = isicol; 2460599ef60dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2461599ef60dSHong Zhang 24621d098868SHong Zhang ierr = PetscLogObjectMemory(B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 24631d098868SHong Zhang b->maxnz = nnz_max; 2464599ef60dSHong Zhang 2465599ef60dSHong Zhang (B)->factor = MAT_FACTOR_ILUDT; 2466599ef60dSHong Zhang (B)->info.factor_mallocs = 0; 24671d098868SHong Zhang (B)->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]); 24681d098868SHong Zhang CHKMEMQ; 24691d098868SHong Zhang /* ------- end of symbolic factorization ---------*/ 2470599ef60dSHong Zhang 2471599ef60dSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 2472599ef60dSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 2473599ef60dSHong Zhang ics = ic; 2474599ef60dSHong Zhang 2475599ef60dSHong Zhang /* linked list for storing column indices of the active row */ 2476599ef60dSHong Zhang nlnk = n + 1; 2477599ef60dSHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 24781d098868SHong Zhang 24791d098868SHong Zhang /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */ 24801d098868SHong Zhang ierr = PetscMalloc((2*n+1)*sizeof(PetscInt),&im);CHKERRQ(ierr); 24811d098868SHong Zhang jtmp = im + n; 24821d098868SHong Zhang /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */ 24838369b28dSHong Zhang ierr = PetscMalloc((2*n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 24848aee2decSHong Zhang ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 24858369b28dSHong Zhang vtmp = rtmp + n; 2486599ef60dSHong Zhang 2487599ef60dSHong Zhang bi[0] = 0; 24888fc3a347SHong Zhang bdiag[0] = nnz_max-1; /* location of diag[0] in factor B */ 2489dc3a2fd3SHong Zhang bdiag_rev[n] = bdiag[0]; 24908fc3a347SHong Zhang bi[2*n+1] = bdiag[0]+1; /* endof bj and ba array */ 2491599ef60dSHong Zhang for (i=0; i<n; i++) { 2492599ef60dSHong Zhang /* copy initial fill into linked list */ 2493599ef60dSHong Zhang nzi = 0; /* nonzeros for active row i */ 24948369b28dSHong Zhang nzi = ai[r[i]+1] - ai[r[i]]; 24958369b28dSHong 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); 24968369b28dSHong Zhang nzi_al = adiag[r[i]] - ai[r[i]]; 24978369b28dSHong Zhang nzi_au = ai[r[i]+1] - adiag[r[i]] -1; 2498599ef60dSHong Zhang ajtmp = aj + ai[r[i]]; 24998369b28dSHong Zhang ierr = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 2500599ef60dSHong Zhang 2501599ef60dSHong Zhang /* load in initial (unfactored row) */ 25021d098868SHong Zhang aatmp = a->a + ai[r[i]]; 25038369b28dSHong Zhang for (j=0; j<nzi; j++) { 25041d098868SHong Zhang rtmp[ics[*ajtmp++]] = *aatmp++; 2505599ef60dSHong Zhang } 2506599ef60dSHong Zhang 2507599ef60dSHong Zhang /* add pivot rows into linked list */ 2508599ef60dSHong Zhang row = lnk[n]; 2509599ef60dSHong Zhang while (row < i ) { 25101d098868SHong Zhang nzi_bl = bi[row+1] - bi[row] + 1; 25111d098868SHong Zhang bjtmp = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */ 25121d098868SHong Zhang ierr = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr); 2513599ef60dSHong Zhang nzi += nlnk; 2514599ef60dSHong Zhang row = lnk[row]; 2515599ef60dSHong Zhang } 2516599ef60dSHong Zhang 25178369b28dSHong Zhang /* copy data from lnk into jtmp, then initialize lnk */ 25188369b28dSHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr); 2519599ef60dSHong Zhang 2520599ef60dSHong Zhang /* numerical factorization */ 25218369b28dSHong Zhang bjtmp = jtmp; 2522599ef60dSHong Zhang row = *bjtmp++; /* 1st pivot row */ 2523599ef60dSHong Zhang while ( row < i ) { 2524599ef60dSHong Zhang pc = rtmp + row; 25253c2a7987SHong Zhang pv = ba + bdiag[row]; /* 1./(diag of the pivot row) */ 25263c2a7987SHong Zhang multiplier = (*pc) * (*pv); 2527599ef60dSHong Zhang *pc = multiplier; 25283c2a7987SHong Zhang if (PetscAbsScalar(*pc) > dt){ /* apply tolerance dropping rule */ 25291d098868SHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 25301d098868SHong Zhang pv = ba + bdiag[row+1] + 1; 25311d098868SHong Zhang /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */ 25321d098868SHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 25331d098868SHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 2534dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 2535599ef60dSHong Zhang } 2536599ef60dSHong Zhang row = *bjtmp++; 2537599ef60dSHong Zhang } 2538599ef60dSHong Zhang 25398369b28dSHong Zhang /* copy sparse rtmp into contiguous vtmp; separate L and U part */ 25408aee2decSHong Zhang diag_tmp = rtmp[i]; /* save diagonal value - may not needed?? */ 25418369b28dSHong Zhang nzi_bl = 0; j = 0; 25428369b28dSHong Zhang while (jtmp[j] < i){ /* Note: jtmp is sorted */ 25438aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 25448369b28dSHong Zhang nzi_bl++; j++; 25458369b28dSHong Zhang } 25468369b28dSHong Zhang nzi_bu = nzi - nzi_bl -1; 25478369b28dSHong Zhang while (j < nzi){ 25488aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 25498369b28dSHong Zhang j++; 25508369b28dSHong Zhang } 25518369b28dSHong Zhang 2552599ef60dSHong Zhang bjtmp = bj + bi[i]; 2553599ef60dSHong Zhang batmp = ba + bi[i]; 25548369b28dSHong Zhang /* apply level dropping rule to L part */ 25558369b28dSHong Zhang ncut = nzi_al + dtcount; 25568369b28dSHong Zhang if (ncut < nzi_bl){ 25578369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr); 25588369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr); 2559599ef60dSHong Zhang } else { 25608369b28dSHong Zhang ncut = nzi_bl; 25618369b28dSHong Zhang } 25628369b28dSHong Zhang for (j=0; j<ncut; j++){ 25638369b28dSHong Zhang bjtmp[j] = jtmp[j]; 25648369b28dSHong Zhang batmp[j] = vtmp[j]; 25656da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */ 25668369b28dSHong Zhang } 25671d098868SHong Zhang bi[i+1] = bi[i] + ncut; 25688369b28dSHong Zhang nzi = ncut + 1; 25698369b28dSHong Zhang 25708369b28dSHong Zhang /* apply level dropping rule to U part */ 25718369b28dSHong Zhang ncut = nzi_au + dtcount; 25728369b28dSHong Zhang if (ncut < nzi_bu){ 25738369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr); 25748369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr); 25758369b28dSHong Zhang } else { 25768369b28dSHong Zhang ncut = nzi_bu; 25778369b28dSHong Zhang } 25788369b28dSHong Zhang nzi += ncut; 25791d098868SHong Zhang 25801d098868SHong Zhang /* mark bdiagonal */ 25811d098868SHong Zhang bdiag[i+1] = bdiag[i] - (ncut + 1); 2582dc3a2fd3SHong Zhang bdiag_rev[n-i-1] = bdiag[i+1]; 25838fc3a347SHong Zhang bi[2*n - i] = bi[2*n - i +1] - (ncut + 1); 25841d098868SHong Zhang bjtmp = bj + bdiag[i]; 25851d098868SHong Zhang batmp = ba + bdiag[i]; 25861d098868SHong Zhang *bjtmp = i; 25878aee2decSHong Zhang *batmp = diag_tmp; /* rtmp[i]; */ 2588c9c72f8fSHong Zhang if (*batmp == 0.0) { 2589c9c72f8fSHong Zhang *batmp = dt+shift; 2590bd1bc851SBarry Smith /* printf(" row %d add shift %g\n",i,shift); */ 2591c9c72f8fSHong Zhang } 2592b78a477dSHong Zhang *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */ 25936da515a1SHong Zhang /* printf(" (%d,%g),",*bjtmp,*batmp); */ 25941d098868SHong Zhang 25951d098868SHong Zhang bjtmp = bj + bdiag[i+1]+1; 25961d098868SHong Zhang batmp = ba + bdiag[i+1]+1; 25978369b28dSHong Zhang for (k=0; k<ncut; k++){ 25981d098868SHong Zhang bjtmp[k] = jtmp[nzi_bl+1+k]; 25991d098868SHong Zhang batmp[k] = vtmp[nzi_bl+1+k]; 26006da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */ 2601599ef60dSHong Zhang } 26026da515a1SHong Zhang /* printf("\n"); */ 2603599ef60dSHong Zhang 2604599ef60dSHong Zhang im[i] = nzi; /* used by PetscLLAddSortedLU() */ 2605599ef60dSHong Zhang /* 26061d098868SHong Zhang printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]); 26078369b28dSHong Zhang printf(" ----------------------------\n"); 2608599ef60dSHong Zhang */ 26098369b28dSHong Zhang } /* for (i=0; i<n; i++) */ 26101d098868SHong Zhang /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */ 26111d098868SHong 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]); 2612599ef60dSHong Zhang 2613599ef60dSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 2614599ef60dSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 2615599ef60dSHong Zhang 2616599ef60dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2617599ef60dSHong Zhang ierr = PetscFree(im);CHKERRQ(ierr); 2618599ef60dSHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 2619599ef60dSHong Zhang 2620599ef60dSHong Zhang ierr = PetscLogFlops(B->cmap->n);CHKERRQ(ierr); 26211d098868SHong Zhang b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n]; 2622599ef60dSHong Zhang 2623599ef60dSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 2624599ef60dSHong Zhang ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr); 2625599ef60dSHong Zhang both_identity = (PetscTruth) (row_identity && icol_identity); 2626599ef60dSHong Zhang if (row_identity && icol_identity) { 26271d098868SHong Zhang B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_iludt; 2628599ef60dSHong Zhang } else { 26291d098868SHong Zhang B->ops->solve = MatSolve_SeqAIJ_iludt; 2630599ef60dSHong Zhang } 2631599ef60dSHong Zhang 2632b78a477dSHong Zhang B->ops->lufactorsymbolic = MatILUDTFactorSymbolic_SeqAIJ; 2633b78a477dSHong Zhang B->ops->lufactornumeric = MatILUDTFactorNumeric_SeqAIJ; 26341d098868SHong Zhang B->ops->solveadd = 0; 26351d098868SHong Zhang B->ops->solvetranspose = 0; 26361d098868SHong Zhang B->ops->solvetransposeadd = 0; 26371d098868SHong Zhang B->ops->matsolve = 0; 2638599ef60dSHong Zhang B->assembled = PETSC_TRUE; 2639599ef60dSHong Zhang B->preallocated = PETSC_TRUE; 2640599ef60dSHong Zhang PetscFunctionReturn(0); 2641599ef60dSHong Zhang } 2642599ef60dSHong Zhang 26433c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */ 26443c2a7987SHong Zhang #undef __FUNCT__ 26453c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorSymbolic_SeqAIJ" 26463c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info) 26473c2a7987SHong Zhang { 26483c2a7987SHong Zhang PetscErrorCode ierr; 26493c2a7987SHong Zhang 26503c2a7987SHong Zhang PetscFunctionBegin; 26513c2a7987SHong Zhang ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr); 26523c2a7987SHong Zhang 26533c2a7987SHong Zhang fact->ops->lufactornumeric = MatILUDTFactorNumeric_SeqAIJ; 26543c2a7987SHong Zhang PetscFunctionReturn(0); 26553c2a7987SHong Zhang } 26563c2a7987SHong Zhang 26573c2a7987SHong Zhang /* 26583c2a7987SHong Zhang same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors 26593c2a7987SHong Zhang - intend to replace existing MatLUFactorNumeric_SeqAIJ() 26603c2a7987SHong Zhang */ 26613c2a7987SHong Zhang #undef __FUNCT__ 26623c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorNumeric_SeqAIJ" 26633c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info) 26643c2a7987SHong Zhang { 26653c2a7987SHong Zhang Mat C=fact; 26663c2a7987SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 26673c2a7987SHong Zhang IS isrow = b->row,isicol = b->icol; 26683c2a7987SHong Zhang PetscErrorCode ierr; 26693c2a7987SHong Zhang const PetscInt *r,*ic,*ics; 2670b78a477dSHong Zhang PetscInt i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 2671b78a477dSHong Zhang PetscInt *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj; 26723c2a7987SHong Zhang MatScalar *rtmp,*pc,multiplier,*v,*pv,*aa=a->a; 2673b78a477dSHong Zhang PetscReal dt=info->dt,shift=info->shiftinblocks; 2674b78a477dSHong Zhang PetscTruth row_identity, col_identity; 26753c2a7987SHong Zhang 26763c2a7987SHong Zhang PetscFunctionBegin; 26773c2a7987SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 26783c2a7987SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 2679b78a477dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 26803c2a7987SHong Zhang ics = ic; 26813c2a7987SHong Zhang 26823c2a7987SHong Zhang for (i=0; i<n; i++){ 2683b78a477dSHong Zhang /* initialize rtmp array */ 2684b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of nozeros in L(i,:) */ 26853c2a7987SHong Zhang bjtmp = bj + bi[i]; 2686b78a477dSHong Zhang for (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0; 2687b78a477dSHong Zhang rtmp[i] = 0.0; 2688b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */ 2689b78a477dSHong Zhang bjtmp = bj + bdiag[i+1] + 1; 2690b78a477dSHong Zhang for (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0; 26913c2a7987SHong Zhang 2692b78a477dSHong Zhang /* load in initial unfactored row of A */ 26936da515a1SHong Zhang /* printf("row %d\n",i); */ 26943c2a7987SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 26953c2a7987SHong Zhang ajtmp = aj + ai[r[i]]; 26963c2a7987SHong Zhang v = aa + ai[r[i]]; 26973c2a7987SHong Zhang for (j=0; j<nz; j++) { 2698b78a477dSHong Zhang rtmp[ics[*ajtmp++]] = v[j]; 26996da515a1SHong Zhang /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */ 27003c2a7987SHong Zhang } 27016da515a1SHong Zhang /* printf("\n"); */ 27023c2a7987SHong Zhang 2703b78a477dSHong Zhang /* numerical factorization */ 2704b78a477dSHong Zhang bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */ 2705b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of entries in L(i,:) */ 2706b78a477dSHong Zhang k = 0; 2707b78a477dSHong Zhang while (k < nzl){ 27083c2a7987SHong Zhang row = *bjtmp++; 27096da515a1SHong Zhang /* printf(" prow %d\n",row); */ 27103c2a7987SHong Zhang pc = rtmp + row; 2711b78a477dSHong Zhang pv = b->a + bdiag[row]; /* 1./(diag of the pivot row) */ 2712b78a477dSHong Zhang multiplier = (*pc) * (*pv); 27133c2a7987SHong Zhang *pc = multiplier; 2714b78a477dSHong Zhang if (PetscAbsScalar(multiplier) > dt){ 2715b78a477dSHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 2716b78a477dSHong Zhang pv = b->a + bdiag[row+1] + 1; 2717b78a477dSHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 2718b78a477dSHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 27196da515a1SHong Zhang /* ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); */ 27203c2a7987SHong Zhang } 2721b78a477dSHong Zhang k++; 27223c2a7987SHong Zhang } 27233c2a7987SHong Zhang 2724b78a477dSHong Zhang /* finished row so stick it into b->a */ 2725b78a477dSHong Zhang /* L-part */ 2726b78a477dSHong Zhang pv = b->a + bi[i] ; 2727b78a477dSHong Zhang pj = bj + bi[i] ; 2728b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; 2729b78a477dSHong Zhang for (j=0; j<nzl; j++) { 2730b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 27316da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 27323c2a7987SHong Zhang } 2733b78a477dSHong Zhang 2734b78a477dSHong Zhang /* diagonal: invert diagonal entries for simplier triangular solves */ 2735b78a477dSHong Zhang if (rtmp[i] == 0.0) rtmp[i] = dt+shift; 2736b78a477dSHong Zhang b->a[bdiag[i]] = 1.0/rtmp[i]; 27376da515a1SHong Zhang /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */ 2738b78a477dSHong Zhang 2739b78a477dSHong Zhang /* U-part */ 2740b78a477dSHong Zhang pv = b->a + bdiag[i+1] + 1; 2741b78a477dSHong Zhang pj = bj + bdiag[i+1] + 1; 2742b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1] - 1; 2743b78a477dSHong Zhang for (j=0; j<nzu; j++) { 2744b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 27456da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 2746b78a477dSHong Zhang } 27476da515a1SHong Zhang /* printf("\n"); */ 2748b78a477dSHong Zhang } 2749b78a477dSHong Zhang 27503c2a7987SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 27513c2a7987SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 27523c2a7987SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 2753b78a477dSHong Zhang 27543c2a7987SHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 27553c2a7987SHong Zhang ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 27563c2a7987SHong Zhang if (row_identity && col_identity) { 2757b78a477dSHong Zhang C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_iludt; 27583c2a7987SHong Zhang } else { 2759b78a477dSHong Zhang C->ops->solve = MatSolve_SeqAIJ_iludt; 27603c2a7987SHong Zhang } 2761b78a477dSHong Zhang C->ops->solveadd = 0; 2762b78a477dSHong Zhang C->ops->solvetranspose = 0; 2763b78a477dSHong Zhang C->ops->solvetransposeadd = 0; 2764b78a477dSHong Zhang C->ops->matsolve = 0; 27653c2a7987SHong Zhang C->assembled = PETSC_TRUE; 27663c2a7987SHong Zhang C->preallocated = PETSC_TRUE; 27673c2a7987SHong Zhang ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 27683c2a7987SHong Zhang PetscFunctionReturn(0); 27693c2a7987SHong Zhang } 2770