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; 147ce3d78c0SShri Abhyankar PetscTruth newdatastruct=PETSC_FALSE; 148289bc588SBarry Smith 1493a40ed3dSBarry Smith PetscFunctionBegin; 150ce3d78c0SShri Abhyankar ierr = PetscOptionsGetTruth(PETSC_NULL,"-lu_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr); 151ce3d78c0SShri Abhyankar if(newdatastruct){ 152ce3d78c0SShri Abhyankar ierr = MatLUFactorSymbolic_SeqAIJ_newdatastruct(B,A,isrow,iscol,info);CHKERRQ(ierr); 153ce3d78c0SShri Abhyankar PetscFunctionReturn(0); 154ce3d78c0SShri Abhyankar } 155ce3d78c0SShri 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 287ce3d78c0SShri Abhyankar #undef __FUNCT__ 288ce3d78c0SShri Abhyankar #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ_newdatastruct" 289ce3d78c0SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ_newdatastruct(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 290ce3d78c0SShri Abhyankar { 291ce3d78c0SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 292ce3d78c0SShri Abhyankar IS isicol; 293ce3d78c0SShri Abhyankar PetscErrorCode ierr; 294ce3d78c0SShri Abhyankar const PetscInt *r,*ic; 295ce3d78c0SShri Abhyankar PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 296ce3d78c0SShri Abhyankar PetscInt *bi,*bj,*ajtmp; 297ce3d78c0SShri Abhyankar PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 298ce3d78c0SShri Abhyankar PetscReal f; 299ce3d78c0SShri Abhyankar PetscInt nlnk,*lnk,k,**bi_ptr; 300ce3d78c0SShri Abhyankar PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 301ce3d78c0SShri Abhyankar PetscBT lnkbt; 302ce3d78c0SShri Abhyankar 303ce3d78c0SShri Abhyankar PetscFunctionBegin; 304ce3d78c0SShri Abhyankar if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square"); 305ce3d78c0SShri Abhyankar ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 306ce3d78c0SShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 307ce3d78c0SShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 308ce3d78c0SShri Abhyankar 309ce3d78c0SShri Abhyankar /* get new row pointers */ 310ce3d78c0SShri Abhyankar ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 311ce3d78c0SShri Abhyankar bi[0] = 0; 312ce3d78c0SShri Abhyankar 313ce3d78c0SShri Abhyankar /* bdiag is location of diagonal in factor */ 314ce3d78c0SShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 315ce3d78c0SShri Abhyankar bdiag[0] = 0; 316ce3d78c0SShri Abhyankar 317ce3d78c0SShri Abhyankar /* linked list for storing column indices of the active row */ 318ce3d78c0SShri Abhyankar nlnk = n + 1; 319ce3d78c0SShri Abhyankar ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 320ce3d78c0SShri Abhyankar 321ce3d78c0SShri Abhyankar ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 322ce3d78c0SShri Abhyankar 323ce3d78c0SShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 324ce3d78c0SShri Abhyankar f = info->fill; 325ce3d78c0SShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 326ce3d78c0SShri Abhyankar current_space = free_space; 327ce3d78c0SShri Abhyankar 328ce3d78c0SShri Abhyankar for (i=0; i<n; i++) { 329ce3d78c0SShri Abhyankar /* copy previous fill into linked list */ 330ce3d78c0SShri Abhyankar nzi = 0; 331ce3d78c0SShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 332ce3d78c0SShri 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); 333ce3d78c0SShri Abhyankar ajtmp = aj + ai[r[i]]; 334ce3d78c0SShri Abhyankar ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 335ce3d78c0SShri Abhyankar nzi += nlnk; 336ce3d78c0SShri Abhyankar 337ce3d78c0SShri Abhyankar /* add pivot rows into linked list */ 338ce3d78c0SShri Abhyankar row = lnk[n]; 339ce3d78c0SShri Abhyankar while (row < i){ 34070319c84SHong Zhang nzbd = bdiag[row] + 1; /* num of entries in the row with column index <= row */ 341ce3d78c0SShri Abhyankar ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 342ce3d78c0SShri Abhyankar ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 343ce3d78c0SShri Abhyankar nzi += nlnk; 344ce3d78c0SShri Abhyankar row = lnk[row]; 345ce3d78c0SShri Abhyankar } 346ce3d78c0SShri Abhyankar bi[i+1] = bi[i] + nzi; 347ce3d78c0SShri Abhyankar im[i] = nzi; 348ce3d78c0SShri Abhyankar 349ce3d78c0SShri Abhyankar /* mark bdiag */ 350ce3d78c0SShri Abhyankar nzbd = 0; 351ce3d78c0SShri Abhyankar nnz = nzi; 352ce3d78c0SShri Abhyankar k = lnk[n]; 353ce3d78c0SShri Abhyankar while (nnz-- && k < i){ 354ce3d78c0SShri Abhyankar nzbd++; 355ce3d78c0SShri Abhyankar k = lnk[k]; 356ce3d78c0SShri Abhyankar } 35770319c84SHong Zhang bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_newdatastruct() */ 358ce3d78c0SShri Abhyankar 359ce3d78c0SShri Abhyankar /* if free space is not available, make more free space */ 360ce3d78c0SShri Abhyankar if (current_space->local_remaining<nzi) { 361ce3d78c0SShri Abhyankar nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */ 362ce3d78c0SShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 363ce3d78c0SShri Abhyankar reallocs++; 364ce3d78c0SShri Abhyankar } 365ce3d78c0SShri Abhyankar 366ce3d78c0SShri Abhyankar /* copy data into free space, then initialize lnk */ 367ce3d78c0SShri Abhyankar ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 368ce3d78c0SShri Abhyankar bi_ptr[i] = current_space->array; 369ce3d78c0SShri Abhyankar current_space->array += nzi; 370ce3d78c0SShri Abhyankar current_space->local_used += nzi; 371ce3d78c0SShri Abhyankar current_space->local_remaining -= nzi; 372ce3d78c0SShri Abhyankar } 373ce3d78c0SShri Abhyankar #if defined(PETSC_USE_INFO) 374ce3d78c0SShri Abhyankar if (ai[n] != 0) { 375ce3d78c0SShri Abhyankar PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 376ce3d78c0SShri Abhyankar ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 377ce3d78c0SShri Abhyankar ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 378ce3d78c0SShri Abhyankar ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 379ce3d78c0SShri Abhyankar ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 380ce3d78c0SShri Abhyankar } else { 381ce3d78c0SShri Abhyankar ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 382ce3d78c0SShri Abhyankar } 383ce3d78c0SShri Abhyankar #endif 384ce3d78c0SShri Abhyankar 385ce3d78c0SShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 386ce3d78c0SShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 387ce3d78c0SShri Abhyankar 388ce3d78c0SShri Abhyankar /* destroy list of free space and other temporary array(s) */ 389ce3d78c0SShri Abhyankar ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 390ce3d78c0SShri Abhyankar ierr = PetscFreeSpaceContiguous_newdatastruct(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 391ce3d78c0SShri Abhyankar ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 392ce3d78c0SShri Abhyankar ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 393ce3d78c0SShri Abhyankar 394ce3d78c0SShri Abhyankar /* put together the new matrix */ 395ce3d78c0SShri Abhyankar ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 396ce3d78c0SShri Abhyankar ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 397ce3d78c0SShri Abhyankar b = (Mat_SeqAIJ*)(B)->data; 398ce3d78c0SShri Abhyankar b->free_a = PETSC_TRUE; 399ce3d78c0SShri Abhyankar b->free_ij = PETSC_TRUE; 400ce3d78c0SShri Abhyankar b->singlemalloc = PETSC_FALSE; 401ce3d78c0SShri Abhyankar ierr = PetscMalloc((bi[2*n+1])*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 402ce3d78c0SShri Abhyankar b->j = bj; 403ce3d78c0SShri Abhyankar b->i = bi; 404ce3d78c0SShri Abhyankar b->diag = bdiag; 405ce3d78c0SShri Abhyankar b->ilen = 0; 406ce3d78c0SShri Abhyankar b->imax = 0; 407ce3d78c0SShri Abhyankar b->row = isrow; 408ce3d78c0SShri Abhyankar b->col = iscol; 409ce3d78c0SShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 410ce3d78c0SShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 411ce3d78c0SShri Abhyankar b->icol = isicol; 412ce3d78c0SShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 413ce3d78c0SShri Abhyankar 414ce3d78c0SShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 415ce3d78c0SShri Abhyankar ierr = PetscLogObjectMemory(B,bi[2*n+1]*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 416ce3d78c0SShri Abhyankar b->maxnz = b->nz = bi[2*n+1] ; 417ce3d78c0SShri Abhyankar 418ce3d78c0SShri Abhyankar (B)->factor = MAT_FACTOR_LU; 419ce3d78c0SShri Abhyankar (B)->info.factor_mallocs = reallocs; 420ce3d78c0SShri Abhyankar (B)->info.fill_ratio_given = f; 421ce3d78c0SShri Abhyankar 422ce3d78c0SShri Abhyankar if (ai[n]) { 423894a2336SHong Zhang (B)->info.fill_ratio_needed = ((PetscReal)bi[2*n+1])/((PetscReal)ai[n]); 424ce3d78c0SShri Abhyankar } else { 425ce3d78c0SShri Abhyankar (B)->info.fill_ratio_needed = 0.0; 426ce3d78c0SShri Abhyankar } 427ce3d78c0SShri Abhyankar (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct; 428ce3d78c0SShri Abhyankar PetscFunctionReturn(0); 429ce3d78c0SShri Abhyankar } 430ce3d78c0SShri Abhyankar 4315b5bc046SBarry Smith /* 4325b5bc046SBarry Smith Trouble in factorization, should we dump the original matrix? 4335b5bc046SBarry Smith */ 4345b5bc046SBarry Smith #undef __FUNCT__ 4355b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix" 4365b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A) 4375b5bc046SBarry Smith { 4385b5bc046SBarry Smith PetscErrorCode ierr; 43990d69ab7SBarry Smith PetscTruth flg = PETSC_FALSE; 4405b5bc046SBarry Smith 4415b5bc046SBarry Smith PetscFunctionBegin; 44290d69ab7SBarry Smith ierr = PetscOptionsGetTruth(PETSC_NULL,"-mat_factor_dump_on_error",&flg,PETSC_NULL);CHKERRQ(ierr); 4435b5bc046SBarry Smith if (flg) { 4445b5bc046SBarry Smith PetscViewer viewer; 4455b5bc046SBarry Smith char filename[PETSC_MAX_PATH_LEN]; 4465b5bc046SBarry Smith 4475b5bc046SBarry Smith ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr); 4487adad957SLisandro Dalcin ierr = PetscViewerBinaryOpen(((PetscObject)A)->comm,filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr); 4495b5bc046SBarry Smith ierr = MatView(A,viewer);CHKERRQ(ierr); 4505b5bc046SBarry Smith ierr = PetscViewerDestroy(viewer);CHKERRQ(ierr); 4515b5bc046SBarry Smith } 4525b5bc046SBarry Smith PetscFunctionReturn(0); 4535b5bc046SBarry Smith } 4545b5bc046SBarry Smith 455db4efbfdSBarry Smith extern PetscErrorCode MatSolve_Inode(Mat,Vec,Vec); 456db4efbfdSBarry Smith 4570a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 458894a2336SHong Zhang extern PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_newdatastruct(Mat,Vec,Vec); 459894a2336SHong Zhang extern PetscErrorCode MatSolve_SeqAIJ_newdatastruct(Mat,Vec,Vec); 460c9c72f8fSHong Zhang 461c9c72f8fSHong Zhang #undef __FUNCT__ 462c9c72f8fSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_newdatastruct" 463c9c72f8fSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_newdatastruct(Mat B,Mat A,const MatFactorInfo *info) 464c9c72f8fSHong Zhang { 465c9c72f8fSHong Zhang Mat C=B; 466c9c72f8fSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 467c9c72f8fSHong Zhang IS isrow = b->row,isicol = b->icol; 468c9c72f8fSHong Zhang PetscErrorCode ierr; 469c9c72f8fSHong Zhang const PetscInt *r,*ic,*ics; 470c9c72f8fSHong Zhang PetscInt i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 471c9c72f8fSHong Zhang PetscInt *ajtmp,*bjtmp,nz,nzL,row,*bdiag=b->diag,*pj; 472c9c72f8fSHong Zhang MatScalar *rtmp,*pc,multiplier,*v,*pv,*aa=a->a; 473c9c72f8fSHong Zhang PetscReal shift=info->shiftinblocks; 474c9c72f8fSHong Zhang PetscTruth row_identity,col_identity; 475c9c72f8fSHong Zhang 476c9c72f8fSHong Zhang PetscFunctionBegin; 477c9c72f8fSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 478c9c72f8fSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 479c9c72f8fSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 480c9c72f8fSHong Zhang ics = ic; 481c9c72f8fSHong Zhang 482c9c72f8fSHong Zhang for (i=0; i<n; i++){ 483c9c72f8fSHong Zhang /* zero rtmp */ 484c9c72f8fSHong Zhang /* L part */ 485c9c72f8fSHong Zhang nz = bi[i+1] - bi[i]; 486c9c72f8fSHong Zhang bjtmp = bj + bi[i]; 487c9c72f8fSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 488c9c72f8fSHong Zhang 489c9c72f8fSHong Zhang /* U part */ 4908fc3a347SHong Zhang nz = bi[2*n-i+1] - bi[2*n-i]; 4918fc3a347SHong Zhang bjtmp = bj + bi[2*n-i]; 492c9c72f8fSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 493c9c72f8fSHong Zhang 494c9c72f8fSHong Zhang /* load in initial (unfactored row) */ 495c9c72f8fSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 496c9c72f8fSHong Zhang ajtmp = aj + ai[r[i]]; 497c9c72f8fSHong Zhang v = aa + ai[r[i]]; 498c9c72f8fSHong Zhang for (j=0; j<nz; j++) { 499c9c72f8fSHong Zhang rtmp[ics[ajtmp[j]]] = v[j]; 500c9c72f8fSHong Zhang } 501c9c72f8fSHong Zhang if (rtmp[ics[r[i]]] == 0.0){ 502c9c72f8fSHong Zhang rtmp[ics[r[i]]] += shift; /* shift the diagonal of the matrix */ 503c9c72f8fSHong Zhang /* printf("row %d, shift %g\n",i,shift); */ 504c9c72f8fSHong Zhang } 505c9c72f8fSHong Zhang 506c9c72f8fSHong Zhang /* elimination */ 507c9c72f8fSHong Zhang bjtmp = bj + bi[i]; 508c9c72f8fSHong Zhang row = *bjtmp++; 509c9c72f8fSHong Zhang nzL = bi[i+1] - bi[i]; 510c9c72f8fSHong Zhang k = 0; 511c9c72f8fSHong Zhang while (k < nzL) { 512c9c72f8fSHong Zhang pc = rtmp + row; 513c9c72f8fSHong Zhang if (*pc != 0.0) { 514c9c72f8fSHong Zhang pv = b->a + bdiag[row]; 515c9c72f8fSHong Zhang multiplier = *pc * (*pv); 516c9c72f8fSHong Zhang *pc = multiplier; 5178fc3a347SHong Zhang pj = b->j + bi[2*n-row]; /* begining of U(row,:) */ 5188fc3a347SHong Zhang pv = b->a + bi[2*n-row]; 5198fc3a347SHong Zhang nz = bi[2*n-row+1] - bi[2*n-row] - 1; /* num of entries in U(row,:), excluding diag */ 520c9c72f8fSHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 521c9c72f8fSHong Zhang ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 522c9c72f8fSHong Zhang } 523c9c72f8fSHong Zhang row = *bjtmp++; k++; 524c9c72f8fSHong Zhang } 525c9c72f8fSHong Zhang 526c9c72f8fSHong Zhang /* finished row so stick it into b->a */ 527c9c72f8fSHong Zhang /* L part */ 528c9c72f8fSHong Zhang pv = b->a + bi[i] ; 529c9c72f8fSHong Zhang pj = b->j + bi[i] ; 530c9c72f8fSHong Zhang nz = bi[i+1] - bi[i]; 531c9c72f8fSHong Zhang for (j=0; j<nz; j++) { 532c9c72f8fSHong Zhang pv[j] = rtmp[pj[j]]; 533c9c72f8fSHong Zhang } 534c9c72f8fSHong Zhang 535c9c72f8fSHong Zhang /* Mark diagonal and invert diagonal for simplier triangular solves */ 536c9c72f8fSHong Zhang pv = b->a + bdiag[i]; 537c9c72f8fSHong Zhang pj = b->j + bdiag[i]; 538c9c72f8fSHong Zhang /* if (*pj != i)SETERRQ2(PETSC_ERR_SUP,"row %d != *pj %d",i,*pj) */ 539c9c72f8fSHong Zhang *pv = 1.0/rtmp[*pj]; 540c9c72f8fSHong Zhang 541c9c72f8fSHong Zhang /* U part */ 5428fc3a347SHong Zhang pv = b->a + bi[2*n-i]; 5438fc3a347SHong Zhang pj = b->j + bi[2*n-i]; 5448fc3a347SHong Zhang nz = bi[2*n-i+1] - bi[2*n-i] - 1; 5458fc3a347SHong Zhang for (j=0; j<nz; j++) pv[j] = rtmp[pj[j]]; 546c9c72f8fSHong Zhang } 547c9c72f8fSHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 548c9c72f8fSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 549c9c72f8fSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 550c9c72f8fSHong Zhang 551c9c72f8fSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 552c9c72f8fSHong Zhang ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 553c9c72f8fSHong Zhang if (row_identity && col_identity) { 554894a2336SHong Zhang C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct; 555c9c72f8fSHong Zhang } else { 556894a2336SHong Zhang C->ops->solve = MatSolve_SeqAIJ_newdatastruct; 557c9c72f8fSHong Zhang } 558c9c72f8fSHong Zhang 559c9c72f8fSHong Zhang C->ops->solveadd = 0; 560c9c72f8fSHong Zhang C->ops->solvetranspose = 0; 561c9c72f8fSHong Zhang C->ops->solvetransposeadd = 0; 562c9c72f8fSHong Zhang C->ops->matsolve = 0; 563c9c72f8fSHong Zhang C->assembled = PETSC_TRUE; 564c9c72f8fSHong Zhang C->preallocated = PETSC_TRUE; 565c9c72f8fSHong Zhang ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 566c9c72f8fSHong Zhang PetscFunctionReturn(0); 567c9c72f8fSHong Zhang } 568c9c72f8fSHong Zhang 5694a2ae208SSatish Balay #undef __FUNCT__ 5704a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ" 5710481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 572289bc588SBarry Smith { 573719d5645SBarry Smith Mat C=B; 574416022c9SBarry Smith Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 57582bf6240SBarry Smith IS isrow = b->row,isicol = b->icol; 5766849ba73SBarry Smith PetscErrorCode ierr; 5775d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 578d278edc7SBarry Smith PetscInt nz,row,i,j,n=A->rmap->n,diag; 579d278edc7SBarry Smith const PetscInt *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 580d278edc7SBarry Smith const PetscInt *ajtmp,*bjtmp,*diag_offset = b->diag,*pj; 581d278edc7SBarry Smith MatScalar *pv,*rtmp,*pc,multiplier,d; 582d278edc7SBarry Smith const MatScalar *v,*aa=a->a; 58333f9893dSHong Zhang PetscReal rs=0.0; 584b3bf805bSHong Zhang LUShift_Ctx sctx; 58542898933SHong Zhang PetscInt newshift,*ddiag; 586289bc588SBarry Smith 5873a40ed3dSBarry Smith PetscFunctionBegin; 58878b31e54SBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 58978b31e54SBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 590b78a477dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 5919f7d0b68SHong Zhang ics = ic; 592289bc588SBarry Smith 593ada7143aSSatish Balay sctx.shift_top = 0; 594ada7143aSSatish Balay sctx.nshift_max = 0; 595ada7143aSSatish Balay sctx.shift_lo = 0; 596ada7143aSSatish Balay sctx.shift_hi = 0; 5970481f469SBarry Smith sctx.shift_fraction = 0; 598ada7143aSSatish Balay 599118f5789SBarry Smith /* if both shift schemes are chosen by user, only use info->shiftpd */ 6001a890ab1SHong Zhang if (info->shiftpd) { /* set sctx.shift_top=max{rs} */ 60142898933SHong Zhang ddiag = a->diag; 6029f7d0b68SHong Zhang sctx.shift_top = info->zeropivot; 6036cc28720Svictorle for (i=0; i<n; i++) { 6049f95998fSHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 6059f7d0b68SHong Zhang d = (aa)[ddiag[i]]; 6061a890ab1SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 6079f7d0b68SHong Zhang v = aa+ai[i]; 6089f7d0b68SHong Zhang nz = ai[i+1] - ai[i]; 609e24cfe64SHong Zhang for (j=0; j<nz; j++) 6101a890ab1SHong Zhang rs += PetscAbsScalar(v[j]); 6111a890ab1SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 6126cc28720Svictorle } 613118f5789SBarry Smith sctx.shift_top *= 1.1; 614d2276718SHong Zhang sctx.nshift_max = 5; 615d2276718SHong Zhang sctx.shift_lo = 0.; 616d2276718SHong Zhang sctx.shift_hi = 1.; 617a0a356efSHong Zhang } 618a0a356efSHong Zhang 6199f7d0b68SHong Zhang sctx.shift_amount = 0.0; 620a0a356efSHong Zhang sctx.nshift = 0; 621085256b3SBarry Smith do { 622d2276718SHong Zhang sctx.lushift = PETSC_FALSE; 623289bc588SBarry Smith for (i=0; i<n; i++){ 624b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 625b3bf805bSHong Zhang bjtmp = bj + bi[i]; 6269f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 627289bc588SBarry Smith 628289bc588SBarry Smith /* load in initial (unfactored row) */ 6299f7d0b68SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 6309f7d0b68SHong Zhang ajtmp = aj + ai[r[i]]; 6319f7d0b68SHong Zhang v = aa + ai[r[i]]; 632085256b3SBarry Smith for (j=0; j<nz; j++) { 633b3bf805bSHong Zhang rtmp[ics[ajtmp[j]]] = v[j]; 634085256b3SBarry Smith } 635d2276718SHong Zhang rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 636bd1bc851SBarry Smith /* if (sctx.shift_amount > 0.0) printf("row %d, shift %g\n",i,sctx.shift_amount); */ 637289bc588SBarry Smith 638b3bf805bSHong Zhang row = *bjtmp++; 639f2582111SSatish Balay while (row < i) { 6408c37ef55SBarry Smith pc = rtmp + row; 6418c37ef55SBarry Smith if (*pc != 0.0) { 642010a20caSHong Zhang pv = b->a + diag_offset[row]; 643010a20caSHong Zhang pj = b->j + diag_offset[row] + 1; 64435aab85fSBarry Smith multiplier = *pc / *pv++; 6458c37ef55SBarry Smith *pc = multiplier; 646b3bf805bSHong Zhang nz = bi[row+1] - diag_offset[row] - 1; 6479f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 648dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 649289bc588SBarry Smith } 650b3bf805bSHong Zhang row = *bjtmp++; 651289bc588SBarry Smith } 652416022c9SBarry Smith /* finished row so stick it into b->a */ 653b3bf805bSHong Zhang pv = b->a + bi[i] ; 654b3bf805bSHong Zhang pj = b->j + bi[i] ; 655b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 656b3bf805bSHong Zhang diag = diag_offset[i] - bi[i]; 657e57ff13aSBarry Smith rs = 0.0; 658d3d32019SBarry Smith for (j=0; j<nz; j++) { 6599f7d0b68SHong Zhang pv[j] = rtmp[pj[j]]; 6609f7d0b68SHong Zhang rs += PetscAbsScalar(pv[j]); 661d3d32019SBarry Smith } 662e57ff13aSBarry Smith rs -= PetscAbsScalar(pv[diag]); 663d2276718SHong Zhang 664b3bf805bSHong Zhang /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */ 665d2276718SHong Zhang sctx.rs = rs; 666d2276718SHong Zhang sctx.pv = pv[diag]; 667c6b1f410SHong Zhang ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); 6685b5bc046SBarry Smith if (newshift == 1) break; 66935aab85fSBarry Smith } 670d2276718SHong Zhang 6710481f469SBarry Smith if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 6726cc28720Svictorle /* 6736c037d1bSvictorle * if no shift in this attempt & shifting & started shifting & can refine, 6746cc28720Svictorle * then try lower shift 6756cc28720Svictorle */ 6760481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 6770481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 6780481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 679d2276718SHong Zhang sctx.lushift = PETSC_TRUE; 680d2276718SHong Zhang sctx.nshift++; 6816cc28720Svictorle } 682d2276718SHong Zhang } while (sctx.lushift); 6830f11f9aeSSatish Balay 68435aab85fSBarry Smith /* invert diagonal entries for simplier triangular solves */ 68535aab85fSBarry Smith for (i=0; i<n; i++) { 686010a20caSHong Zhang b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]]; 68735aab85fSBarry Smith } 688606d414cSSatish Balay ierr = PetscFree(rtmp);CHKERRQ(ierr); 68978b31e54SBarry Smith ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 69078b31e54SBarry Smith ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 691db4efbfdSBarry Smith if (b->inode.use) { 692db4efbfdSBarry Smith C->ops->solve = MatSolve_Inode; 693db4efbfdSBarry Smith } else { 6948d8c7c43SBarry Smith PetscTruth row_identity, col_identity; 6958d8c7c43SBarry Smith ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 6968d8c7c43SBarry Smith ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 6978d8c7c43SBarry Smith if (row_identity && col_identity) { 6988d8c7c43SBarry Smith C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 6998d8c7c43SBarry Smith } else { 700db4efbfdSBarry Smith C->ops->solve = MatSolve_SeqAIJ; 701db4efbfdSBarry Smith } 7028d8c7c43SBarry Smith } 703db4efbfdSBarry Smith C->ops->solveadd = MatSolveAdd_SeqAIJ; 704db4efbfdSBarry Smith C->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 705db4efbfdSBarry Smith C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 706de26497bSBarry Smith C->ops->matsolve = MatMatSolve_SeqAIJ; 707c456f294SBarry Smith C->assembled = PETSC_TRUE; 7085c9eb25fSBarry Smith C->preallocated = PETSC_TRUE; 709d0f46423SBarry Smith ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 710d2276718SHong Zhang if (sctx.nshift){ 711e738e422SBarry Smith if (info->shiftpd) { 7120481f469SBarry 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); 713e738e422SBarry Smith } else if (info->shiftnz) { 714e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 7156cc28720Svictorle } 7160697b6caSHong Zhang } 7173a40ed3dSBarry Smith PetscFunctionReturn(0); 718289bc588SBarry Smith } 7190889b5dcSvictorle 720137fb511SHong Zhang /* 721137fb511SHong Zhang This routine implements inplace ILU(0) with row or/and column permutations. 722137fb511SHong Zhang Input: 723137fb511SHong Zhang A - original matrix 724137fb511SHong Zhang Output; 725137fb511SHong Zhang A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i] 726137fb511SHong Zhang a->j (col index) is permuted by the inverse of colperm, then sorted 727137fb511SHong Zhang a->a reordered accordingly with a->j 728137fb511SHong Zhang a->diag (ptr to diagonal elements) is updated. 729137fb511SHong Zhang */ 730137fb511SHong Zhang #undef __FUNCT__ 731137fb511SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_InplaceWithPerm" 7320481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info) 733137fb511SHong Zhang { 734b051339dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 735b051339dSHong Zhang IS isrow = a->row,isicol = a->icol; 736137fb511SHong Zhang PetscErrorCode ierr; 7375d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 7385d0c19d7SBarry Smith PetscInt i,j,n=A->rmap->n,*ai=a->i,*aj=a->j; 7395d0c19d7SBarry Smith PetscInt *ajtmp,nz,row; 740b051339dSHong Zhang PetscInt *diag = a->diag,nbdiag,*pj; 741a77337e4SBarry Smith PetscScalar *rtmp,*pc,multiplier,d; 742a77337e4SBarry Smith MatScalar *v,*pv; 743137fb511SHong Zhang PetscReal rs; 744137fb511SHong Zhang LUShift_Ctx sctx; 745b051339dSHong Zhang PetscInt newshift; 746137fb511SHong Zhang 747137fb511SHong Zhang PetscFunctionBegin; 748719d5645SBarry Smith if (A != B) SETERRQ(PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address"); 749137fb511SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 750137fb511SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 751137fb511SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr); 752137fb511SHong Zhang ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 753b051339dSHong Zhang ics = ic; 754137fb511SHong Zhang 755137fb511SHong Zhang sctx.shift_top = 0; 756137fb511SHong Zhang sctx.nshift_max = 0; 757137fb511SHong Zhang sctx.shift_lo = 0; 758137fb511SHong Zhang sctx.shift_hi = 0; 7590481f469SBarry Smith sctx.shift_fraction = 0; 760137fb511SHong Zhang 761137fb511SHong Zhang /* if both shift schemes are chosen by user, only use info->shiftpd */ 762137fb511SHong Zhang if (info->shiftpd) { /* set sctx.shift_top=max{rs} */ 763137fb511SHong Zhang sctx.shift_top = 0; 764137fb511SHong Zhang for (i=0; i<n; i++) { 765137fb511SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 766b051339dSHong Zhang d = (a->a)[diag[i]]; 767137fb511SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 768b051339dSHong Zhang v = a->a+ai[i]; 769b051339dSHong Zhang nz = ai[i+1] - ai[i]; 770137fb511SHong Zhang for (j=0; j<nz; j++) 771137fb511SHong Zhang rs += PetscAbsScalar(v[j]); 772137fb511SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 773137fb511SHong Zhang } 774137fb511SHong Zhang if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot; 775137fb511SHong Zhang sctx.shift_top *= 1.1; 776137fb511SHong Zhang sctx.nshift_max = 5; 777137fb511SHong Zhang sctx.shift_lo = 0.; 778137fb511SHong Zhang sctx.shift_hi = 1.; 779137fb511SHong Zhang } 780137fb511SHong Zhang 781137fb511SHong Zhang sctx.shift_amount = 0; 782137fb511SHong Zhang sctx.nshift = 0; 783137fb511SHong Zhang do { 784137fb511SHong Zhang sctx.lushift = PETSC_FALSE; 785137fb511SHong Zhang for (i=0; i<n; i++){ 786b051339dSHong Zhang /* load in initial unfactored row */ 787b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 788b051339dSHong Zhang ajtmp = aj + ai[r[i]]; 789b051339dSHong Zhang v = a->a + ai[r[i]]; 790137fb511SHong Zhang /* sort permuted ajtmp and values v accordingly */ 791b051339dSHong Zhang for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]]; 792137fb511SHong Zhang ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr); 793137fb511SHong Zhang 794b051339dSHong Zhang diag[r[i]] = ai[r[i]]; 795137fb511SHong Zhang for (j=0; j<nz; j++) { 796137fb511SHong Zhang rtmp[ajtmp[j]] = v[j]; 797b051339dSHong Zhang if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */ 798137fb511SHong Zhang } 799137fb511SHong Zhang rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 800137fb511SHong Zhang 801b051339dSHong Zhang row = *ajtmp++; 802137fb511SHong Zhang while (row < i) { 803137fb511SHong Zhang pc = rtmp + row; 804137fb511SHong Zhang if (*pc != 0.0) { 805b051339dSHong Zhang pv = a->a + diag[r[row]]; 806b051339dSHong Zhang pj = aj + diag[r[row]] + 1; 807137fb511SHong Zhang 808137fb511SHong Zhang multiplier = *pc / *pv++; 809137fb511SHong Zhang *pc = multiplier; 810b051339dSHong Zhang nz = ai[r[row]+1] - diag[r[row]] - 1; 811b051339dSHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 812dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 813137fb511SHong Zhang } 814b051339dSHong Zhang row = *ajtmp++; 815137fb511SHong Zhang } 816b051339dSHong Zhang /* finished row so overwrite it onto a->a */ 817b051339dSHong Zhang pv = a->a + ai[r[i]] ; 818b051339dSHong Zhang pj = aj + ai[r[i]] ; 819b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 820b051339dSHong Zhang nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */ 821137fb511SHong Zhang 822137fb511SHong Zhang rs = 0.0; 823137fb511SHong Zhang for (j=0; j<nz; j++) { 824b051339dSHong Zhang pv[j] = rtmp[pj[j]]; 825b051339dSHong Zhang if (j != nbdiag) rs += PetscAbsScalar(pv[j]); 826137fb511SHong Zhang } 827137fb511SHong Zhang 828137fb511SHong Zhang /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */ 829137fb511SHong Zhang sctx.rs = rs; 830b051339dSHong Zhang sctx.pv = pv[nbdiag]; 831c6b1f410SHong Zhang ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); 832137fb511SHong Zhang if (newshift == 1) break; 833137fb511SHong Zhang } 834137fb511SHong Zhang 8350481f469SBarry Smith if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 836137fb511SHong Zhang /* 837137fb511SHong Zhang * if no shift in this attempt & shifting & started shifting & can refine, 838137fb511SHong Zhang * then try lower shift 839137fb511SHong Zhang */ 8400481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 8410481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 8420481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 843137fb511SHong Zhang sctx.lushift = PETSC_TRUE; 844137fb511SHong Zhang sctx.nshift++; 845137fb511SHong Zhang } 846137fb511SHong Zhang } while (sctx.lushift); 847137fb511SHong Zhang 848137fb511SHong Zhang /* invert diagonal entries for simplier triangular solves */ 849137fb511SHong Zhang for (i=0; i<n; i++) { 850b051339dSHong Zhang a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]]; 851137fb511SHong Zhang } 852137fb511SHong Zhang 853137fb511SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 854137fb511SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 855137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 856b051339dSHong Zhang A->ops->solve = MatSolve_SeqAIJ_InplaceWithPerm; 857db4efbfdSBarry Smith A->ops->solveadd = MatSolveAdd_SeqAIJ; 858db4efbfdSBarry Smith A->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 859db4efbfdSBarry Smith A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 860b051339dSHong Zhang A->assembled = PETSC_TRUE; 8615c9eb25fSBarry Smith A->preallocated = PETSC_TRUE; 862d0f46423SBarry Smith ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr); 863137fb511SHong Zhang if (sctx.nshift){ 864e738e422SBarry Smith if (info->shiftpd) { 8650481f469SBarry 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); 866e738e422SBarry Smith } else if (info->shiftnz) { 867e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 868137fb511SHong Zhang } 869137fb511SHong Zhang } 870137fb511SHong Zhang PetscFunctionReturn(0); 871137fb511SHong Zhang } 872137fb511SHong Zhang 8730a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 8744a2ae208SSatish Balay #undef __FUNCT__ 8754a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ" 8760481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info) 877da3a660dSBarry Smith { 878dfbe8321SBarry Smith PetscErrorCode ierr; 879416022c9SBarry Smith Mat C; 880416022c9SBarry Smith 8813a40ed3dSBarry Smith PetscFunctionBegin; 88243244d56SBarry Smith ierr = MatGetFactor(A,MAT_SOLVER_PETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr); 883719d5645SBarry Smith ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr); 884719d5645SBarry Smith ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr); 885db4efbfdSBarry Smith A->ops->solve = C->ops->solve; 886db4efbfdSBarry Smith A->ops->solvetranspose = C->ops->solvetranspose; 887273d9f13SBarry Smith ierr = MatHeaderCopy(A,C);CHKERRQ(ierr); 88852e6d16bSBarry Smith ierr = PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr); 8893a40ed3dSBarry Smith PetscFunctionReturn(0); 890da3a660dSBarry Smith } 8910a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 8921d098868SHong Zhang 893889a8dedSBarry Smith 8944a2ae208SSatish Balay #undef __FUNCT__ 8954a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ" 896dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx) 8978c37ef55SBarry Smith { 898416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 899416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 9006849ba73SBarry Smith PetscErrorCode ierr; 9015d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 9025d0c19d7SBarry Smith PetscInt nz; 9035d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 904dd6ea824SBarry Smith PetscScalar *x,*tmp,*tmps,sum; 905d9fead3dSBarry Smith const PetscScalar *b; 906dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 9078c37ef55SBarry Smith 9083a40ed3dSBarry Smith PetscFunctionBegin; 9093a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 91091bf9adeSBarry Smith 911d9fead3dSBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 9121ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 913416022c9SBarry Smith tmp = a->solve_work; 9148c37ef55SBarry Smith 9153b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 9163b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 9178c37ef55SBarry Smith 9188c37ef55SBarry Smith /* forward solve the lower triangular */ 9198c37ef55SBarry Smith tmp[0] = b[*r++]; 920010a20caSHong Zhang tmps = tmp; 9218c37ef55SBarry Smith for (i=1; i<n; i++) { 922010a20caSHong Zhang v = aa + ai[i] ; 923010a20caSHong Zhang vi = aj + ai[i] ; 92453e7425aSBarry Smith nz = a->diag[i] - ai[i]; 92553e7425aSBarry Smith sum = b[*r++]; 926003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 9278c37ef55SBarry Smith tmp[i] = sum; 9288c37ef55SBarry Smith } 9298c37ef55SBarry Smith 9308c37ef55SBarry Smith /* backward solve the upper triangular */ 9318c37ef55SBarry Smith for (i=n-1; i>=0; i--){ 932010a20caSHong Zhang v = aa + a->diag[i] + 1; 933010a20caSHong Zhang vi = aj + a->diag[i] + 1; 934416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 9358c37ef55SBarry Smith sum = tmp[i]; 936003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 937010a20caSHong Zhang x[*c--] = tmp[i] = sum*aa[a->diag[i]]; 9388c37ef55SBarry Smith } 9398c37ef55SBarry Smith 9403b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 9413b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 942d9fead3dSBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 9431ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 944dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 9453a40ed3dSBarry Smith PetscFunctionReturn(0); 9468c37ef55SBarry Smith } 947026e39d0SSatish Balay 9482bebee5dSHong Zhang #undef __FUNCT__ 9492bebee5dSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ" 9502bebee5dSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X) 9512bebee5dSHong Zhang { 9522bebee5dSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 9532bebee5dSHong Zhang IS iscol = a->col,isrow = a->row; 9542bebee5dSHong Zhang PetscErrorCode ierr; 9555d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 9565d0c19d7SBarry Smith PetscInt nz,neq; 9575d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 958dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 959dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 960db4efbfdSBarry Smith PetscTruth bisdense,xisdense; 9612bebee5dSHong Zhang 9622bebee5dSHong Zhang PetscFunctionBegin; 9632bebee5dSHong Zhang if (!n) PetscFunctionReturn(0); 9642bebee5dSHong Zhang 965db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr); 966db4efbfdSBarry Smith if (!bisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix"); 967db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr); 968db4efbfdSBarry Smith if (!xisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix"); 969db4efbfdSBarry Smith 9702bebee5dSHong Zhang ierr = MatGetArray(B,&b);CHKERRQ(ierr); 9712bebee5dSHong Zhang ierr = MatGetArray(X,&x);CHKERRQ(ierr); 9722bebee5dSHong Zhang 9732bebee5dSHong Zhang tmp = a->solve_work; 9742bebee5dSHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 9752bebee5dSHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 9762bebee5dSHong Zhang 977a36b22ccSSatish Balay for (neq=0; neq<B->cmap->n; neq++){ 9782bebee5dSHong Zhang /* forward solve the lower triangular */ 9792bebee5dSHong Zhang tmp[0] = b[r[0]]; 9802bebee5dSHong Zhang tmps = tmp; 9812bebee5dSHong Zhang for (i=1; i<n; i++) { 9822bebee5dSHong Zhang v = aa + ai[i] ; 9832bebee5dSHong Zhang vi = aj + ai[i] ; 9842bebee5dSHong Zhang nz = a->diag[i] - ai[i]; 9852bebee5dSHong Zhang sum = b[r[i]]; 986003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 9872bebee5dSHong Zhang tmp[i] = sum; 9882bebee5dSHong Zhang } 9892bebee5dSHong Zhang /* backward solve the upper triangular */ 9902bebee5dSHong Zhang for (i=n-1; i>=0; i--){ 9912bebee5dSHong Zhang v = aa + a->diag[i] + 1; 9922bebee5dSHong Zhang vi = aj + a->diag[i] + 1; 9932bebee5dSHong Zhang nz = ai[i+1] - a->diag[i] - 1; 9942bebee5dSHong Zhang sum = tmp[i]; 995003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 9962bebee5dSHong Zhang x[c[i]] = tmp[i] = sum*aa[a->diag[i]]; 9972bebee5dSHong Zhang } 9982bebee5dSHong Zhang 9992bebee5dSHong Zhang b += n; 10002bebee5dSHong Zhang x += n; 10012bebee5dSHong Zhang } 10022bebee5dSHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10032bebee5dSHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 10042bebee5dSHong Zhang ierr = MatRestoreArray(B,&b);CHKERRQ(ierr); 10052bebee5dSHong Zhang ierr = MatRestoreArray(X,&x);CHKERRQ(ierr); 1006dc0b31edSSatish Balay ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr); 10072bebee5dSHong Zhang PetscFunctionReturn(0); 10082bebee5dSHong Zhang } 10092bebee5dSHong Zhang 1010137fb511SHong Zhang #undef __FUNCT__ 1011137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm" 1012137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx) 1013137fb511SHong Zhang { 1014137fb511SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1015137fb511SHong Zhang IS iscol = a->col,isrow = a->row; 1016137fb511SHong Zhang PetscErrorCode ierr; 10175d0c19d7SBarry Smith const PetscInt *r,*c,*rout,*cout; 10185d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 10195d0c19d7SBarry Smith PetscInt nz,row; 1020dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1021dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1022137fb511SHong Zhang 1023137fb511SHong Zhang PetscFunctionBegin; 1024137fb511SHong Zhang if (!n) PetscFunctionReturn(0); 1025137fb511SHong Zhang 1026137fb511SHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 1027137fb511SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1028137fb511SHong Zhang tmp = a->solve_work; 1029137fb511SHong Zhang 1030137fb511SHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1031137fb511SHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1032137fb511SHong Zhang 1033137fb511SHong Zhang /* forward solve the lower triangular */ 1034137fb511SHong Zhang tmp[0] = b[*r++]; 1035137fb511SHong Zhang tmps = tmp; 1036137fb511SHong Zhang for (row=1; row<n; row++) { 1037137fb511SHong Zhang i = rout[row]; /* permuted row */ 1038137fb511SHong Zhang v = aa + ai[i] ; 1039137fb511SHong Zhang vi = aj + ai[i] ; 1040137fb511SHong Zhang nz = a->diag[i] - ai[i]; 1041137fb511SHong Zhang sum = b[*r++]; 1042003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1043137fb511SHong Zhang tmp[row] = sum; 1044137fb511SHong Zhang } 1045137fb511SHong Zhang 1046137fb511SHong Zhang /* backward solve the upper triangular */ 1047137fb511SHong Zhang for (row=n-1; row>=0; row--){ 1048137fb511SHong Zhang i = rout[row]; /* permuted row */ 1049137fb511SHong Zhang v = aa + a->diag[i] + 1; 1050137fb511SHong Zhang vi = aj + a->diag[i] + 1; 1051137fb511SHong Zhang nz = ai[i+1] - a->diag[i] - 1; 1052137fb511SHong Zhang sum = tmp[row]; 1053003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1054137fb511SHong Zhang x[*c--] = tmp[row] = sum*aa[a->diag[i]]; 1055137fb511SHong Zhang } 1056137fb511SHong Zhang 1057137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1058137fb511SHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1059137fb511SHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 1060137fb511SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1061dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 1062137fb511SHong Zhang PetscFunctionReturn(0); 1063137fb511SHong Zhang } 1064137fb511SHong Zhang 1065930ae53cSBarry Smith /* ----------------------------------------------------------- */ 1066a4005a5dSBarry Smith #include "../src/mat/impls/aij/seq/ftn-kernels/fsolve.h" 10674a2ae208SSatish Balay #undef __FUNCT__ 10684a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering" 1069dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx) 1070930ae53cSBarry Smith { 1071930ae53cSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 10726849ba73SBarry Smith PetscErrorCode ierr; 1073d0f46423SBarry Smith PetscInt n = A->rmap->n; 1074b377110cSBarry Smith const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag; 1075356650c2SBarry Smith PetscScalar *x; 1076356650c2SBarry Smith const PetscScalar *b; 1077dd6ea824SBarry Smith const MatScalar *aa = a->a; 1078aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 1079356650c2SBarry Smith PetscInt adiag_i,i,nz,ai_i; 1080b377110cSBarry Smith const PetscInt *vi; 1081dd6ea824SBarry Smith const MatScalar *v; 1082dd6ea824SBarry Smith PetscScalar sum; 1083d85afc42SSatish Balay #endif 1084930ae53cSBarry Smith 10853a40ed3dSBarry Smith PetscFunctionBegin; 10863a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 1087930ae53cSBarry Smith 1088356650c2SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 10891ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1090930ae53cSBarry Smith 1091aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 10921c4feecaSSatish Balay fortransolveaij_(&n,x,ai,aj,adiag,aa,b); 10931c4feecaSSatish Balay #else 1094930ae53cSBarry Smith /* forward solve the lower triangular */ 1095930ae53cSBarry Smith x[0] = b[0]; 1096930ae53cSBarry Smith for (i=1; i<n; i++) { 1097930ae53cSBarry Smith ai_i = ai[i]; 1098930ae53cSBarry Smith v = aa + ai_i; 1099930ae53cSBarry Smith vi = aj + ai_i; 1100930ae53cSBarry Smith nz = adiag[i] - ai_i; 1101930ae53cSBarry Smith sum = b[i]; 1102003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1103930ae53cSBarry Smith x[i] = sum; 1104930ae53cSBarry Smith } 1105930ae53cSBarry Smith 1106930ae53cSBarry Smith /* backward solve the upper triangular */ 1107930ae53cSBarry Smith for (i=n-1; i>=0; i--){ 1108930ae53cSBarry Smith adiag_i = adiag[i]; 1109d708749eSBarry Smith v = aa + adiag_i + 1; 1110d708749eSBarry Smith vi = aj + adiag_i + 1; 1111930ae53cSBarry Smith nz = ai[i+1] - adiag_i - 1; 1112930ae53cSBarry Smith sum = x[i]; 1113003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1114930ae53cSBarry Smith x[i] = sum*aa[adiag_i]; 1115930ae53cSBarry Smith } 11161c4feecaSSatish Balay #endif 1117dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 1118356650c2SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 11191ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 11203a40ed3dSBarry Smith PetscFunctionReturn(0); 1121930ae53cSBarry Smith } 1122930ae53cSBarry Smith 11234a2ae208SSatish Balay #undef __FUNCT__ 11244a2ae208SSatish Balay #define __FUNCT__ "MatSolveAdd_SeqAIJ" 1125dfbe8321SBarry Smith PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx) 1126da3a660dSBarry Smith { 1127416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1128416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 11296849ba73SBarry Smith PetscErrorCode ierr; 11305d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 11315d0c19d7SBarry Smith PetscInt nz; 11325d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 1133dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,sum; 1134dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1135da3a660dSBarry Smith 11363a40ed3dSBarry Smith PetscFunctionBegin; 113778b31e54SBarry Smith if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 1138da3a660dSBarry Smith 11391ebc52fbSHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 11401ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1141416022c9SBarry Smith tmp = a->solve_work; 1142da3a660dSBarry Smith 11433b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 11443b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1145da3a660dSBarry Smith 1146da3a660dSBarry Smith /* forward solve the lower triangular */ 1147da3a660dSBarry Smith tmp[0] = b[*r++]; 1148da3a660dSBarry Smith for (i=1; i<n; i++) { 1149010a20caSHong Zhang v = aa + ai[i] ; 1150010a20caSHong Zhang vi = aj + ai[i] ; 1151416022c9SBarry Smith nz = a->diag[i] - ai[i]; 1152da3a660dSBarry Smith sum = b[*r++]; 1153010a20caSHong Zhang while (nz--) sum -= *v++ * tmp[*vi++ ]; 1154da3a660dSBarry Smith tmp[i] = sum; 1155da3a660dSBarry Smith } 1156da3a660dSBarry Smith 1157da3a660dSBarry Smith /* backward solve the upper triangular */ 1158da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1159010a20caSHong Zhang v = aa + a->diag[i] + 1; 1160010a20caSHong Zhang vi = aj + a->diag[i] + 1; 1161416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 1162da3a660dSBarry Smith sum = tmp[i]; 1163010a20caSHong Zhang while (nz--) sum -= *v++ * tmp[*vi++ ]; 1164010a20caSHong Zhang tmp[i] = sum*aa[a->diag[i]]; 1165da3a660dSBarry Smith x[*c--] += tmp[i]; 1166da3a660dSBarry Smith } 1167da3a660dSBarry Smith 11683b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 11693b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 11701ebc52fbSHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 11711ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1172dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 1173898183ecSLois Curfman McInnes 11743a40ed3dSBarry Smith PetscFunctionReturn(0); 1175da3a660dSBarry Smith } 1176da3a660dSBarry Smith /* -------------------------------------------------------------------*/ 11774a2ae208SSatish Balay #undef __FUNCT__ 11784a2ae208SSatish Balay #define __FUNCT__ "MatSolveTranspose_SeqAIJ" 1179dfbe8321SBarry Smith PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx) 1180da3a660dSBarry Smith { 1181416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 11822235e667SBarry Smith IS iscol = a->col,isrow = a->row; 11836849ba73SBarry Smith PetscErrorCode ierr; 11845d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 11855d0c19d7SBarry Smith PetscInt i,n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 11865d0c19d7SBarry Smith PetscInt nz,*diag = a->diag; 1187dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,s1; 1188dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1189da3a660dSBarry Smith 11903a40ed3dSBarry Smith PetscFunctionBegin; 11911ebc52fbSHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 11921ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1193416022c9SBarry Smith tmp = a->solve_work; 1194da3a660dSBarry Smith 11952235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 11962235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1197da3a660dSBarry Smith 1198da3a660dSBarry Smith /* copy the b into temp work space according to permutation */ 11992235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1200da3a660dSBarry Smith 1201da3a660dSBarry Smith /* forward solve the U^T */ 1202da3a660dSBarry Smith for (i=0; i<n; i++) { 1203010a20caSHong Zhang v = aa + diag[i] ; 1204010a20caSHong Zhang vi = aj + diag[i] + 1; 1205f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 1206c38d4ed2SBarry Smith s1 = tmp[i]; 1207ef66eb69SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 1208da3a660dSBarry Smith while (nz--) { 1209010a20caSHong Zhang tmp[*vi++ ] -= (*v++)*s1; 1210da3a660dSBarry Smith } 1211c38d4ed2SBarry Smith tmp[i] = s1; 1212da3a660dSBarry Smith } 1213da3a660dSBarry Smith 1214da3a660dSBarry Smith /* backward solve the L^T */ 1215da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1216010a20caSHong Zhang v = aa + diag[i] - 1 ; 1217010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1218f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 1219f1af5d2fSBarry Smith s1 = tmp[i]; 1220da3a660dSBarry Smith while (nz--) { 1221010a20caSHong Zhang tmp[*vi-- ] -= (*v--)*s1; 1222da3a660dSBarry Smith } 1223da3a660dSBarry Smith } 1224da3a660dSBarry Smith 1225da3a660dSBarry Smith /* copy tmp into x according to permutation */ 1226da3a660dSBarry Smith for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1227da3a660dSBarry Smith 12282235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 12292235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 12301ebc52fbSHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 12311ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1232da3a660dSBarry Smith 1233dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 12343a40ed3dSBarry Smith PetscFunctionReturn(0); 1235da3a660dSBarry Smith } 1236da3a660dSBarry Smith 12374a2ae208SSatish Balay #undef __FUNCT__ 12384a2ae208SSatish Balay #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ" 1239dfbe8321SBarry Smith PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx) 1240da3a660dSBarry Smith { 1241416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 12422235e667SBarry Smith IS iscol = a->col,isrow = a->row; 12436849ba73SBarry Smith PetscErrorCode ierr; 12445d0c19d7SBarry Smith const PetscInt *r,*c,*rout,*cout; 12455d0c19d7SBarry Smith PetscInt i,n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 12465d0c19d7SBarry Smith PetscInt nz,*diag = a->diag; 1247dd6ea824SBarry Smith PetscScalar *x,*b,*tmp; 1248dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 12496abc6512SBarry Smith 12503a40ed3dSBarry Smith PetscFunctionBegin; 125123bc6035SMatthew Knepley if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 12526abc6512SBarry Smith 12531ebc52fbSHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 12541ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1255416022c9SBarry Smith tmp = a->solve_work; 12566abc6512SBarry Smith 12572235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 12582235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 12596abc6512SBarry Smith 12606abc6512SBarry Smith /* copy the b into temp work space according to permutation */ 12612235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 12626abc6512SBarry Smith 12636abc6512SBarry Smith /* forward solve the U^T */ 12646abc6512SBarry Smith for (i=0; i<n; i++) { 1265010a20caSHong Zhang v = aa + diag[i] ; 1266010a20caSHong Zhang vi = aj + diag[i] + 1; 1267f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 12686abc6512SBarry Smith tmp[i] *= *v++; 12696abc6512SBarry Smith while (nz--) { 1270010a20caSHong Zhang tmp[*vi++ ] -= (*v++)*tmp[i]; 12716abc6512SBarry Smith } 12726abc6512SBarry Smith } 12736abc6512SBarry Smith 12746abc6512SBarry Smith /* backward solve the L^T */ 12756abc6512SBarry Smith for (i=n-1; i>=0; i--){ 1276010a20caSHong Zhang v = aa + diag[i] - 1 ; 1277010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1278f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 12796abc6512SBarry Smith while (nz--) { 1280010a20caSHong Zhang tmp[*vi-- ] -= (*v--)*tmp[i]; 12816abc6512SBarry Smith } 12826abc6512SBarry Smith } 12836abc6512SBarry Smith 12846abc6512SBarry Smith /* copy tmp into x according to permutation */ 12856abc6512SBarry Smith for (i=0; i<n; i++) x[r[i]] += tmp[i]; 12866abc6512SBarry Smith 12872235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 12882235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 12891ebc52fbSHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 12901ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 12916abc6512SBarry Smith 1292dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 12933a40ed3dSBarry Smith PetscFunctionReturn(0); 1294da3a660dSBarry Smith } 12959e25ed09SBarry Smith /* ----------------------------------------------------------------*/ 12963c5fc038SBarry Smith EXTERN PetscErrorCode Mat_CheckInode(Mat,PetscTruth); 1297f77e22a1SHong Zhang EXTERN PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,Mat,MatDuplicateOption,PetscTruth); 129808480c60SBarry Smith 12998fc3a347SHong Zhang /* 13008fc3a347SHong Zhang ilu(0) with natural ordering under new data structure. 13018fc3a347SHong Zhang Factored arrays bj and ba are stored as 13028fc3a347SHong Zhang L(0,:), L(1,:), ...,L(n-1,:), U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:) 13038fc3a347SHong Zhang 13048fc3a347SHong Zhang bi=fact->i is an array of size 2n+2, in which 13058fc3a347SHong Zhang bi+ 13068fc3a347SHong Zhang bi[i] -> 1st entry of L(i,:),i=0,...,i-1 1307f77e22a1SHong Zhang bi[n] -> points to L(n-1,:)+1 13088fc3a347SHong Zhang bi[n+1] -> 1st entry of U(n-1,:) 13098fc3a347SHong Zhang bi[2n-i] -> 1st entry of U(i,:) 13108fc3a347SHong Zhang bi[2n-i+1] -> end of U(i,:)+1, the 1st entry of U(i-1,:) 1311f77e22a1SHong Zhang bi[2n] -> 1st entry of U(0,:) 1312f77e22a1SHong Zhang bi[2n+1] -> points to U(0,:)+1 13138fc3a347SHong Zhang 13148fc3a347SHong Zhang U(i,:) contains diag[i] as its last entry, i.e., 13158fc3a347SHong Zhang U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 13168fc3a347SHong Zhang */ 1317c9c72f8fSHong Zhang #undef __FUNCT__ 13188fc3a347SHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct" 13198fc3a347SHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1320c9c72f8fSHong Zhang { 1321c9c72f8fSHong Zhang 1322c9c72f8fSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1323c9c72f8fSHong Zhang PetscErrorCode ierr; 1324c9c72f8fSHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag; 13258fc3a347SHong Zhang PetscInt i,j,nz,*bi,*bj,*bdiag; 1326c9c72f8fSHong Zhang 1327c9c72f8fSHong Zhang PetscFunctionBegin; 1328f77e22a1SHong Zhang ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr); 1329c9c72f8fSHong Zhang b = (Mat_SeqAIJ*)(fact)->data; 1330c9c72f8fSHong Zhang 1331f77e22a1SHong Zhang /* allocate matrix arrays for new data structure */ 1332f77e22a1SHong Zhang ierr = PetscMalloc3(ai[n]+1,PetscScalar,&b->a,ai[n]+1,PetscInt,&b->j,2*n+2,PetscInt,&b->i);CHKERRQ(ierr); 1333f77e22a1SHong Zhang ierr = PetscLogObjectMemory(fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(2*n+2)*sizeof(PetscInt));CHKERRQ(ierr); 1334f77e22a1SHong Zhang b->singlemalloc = PETSC_TRUE; 1335f77e22a1SHong Zhang if (!b->diag){ 13368fc3a347SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&b->diag);CHKERRQ(ierr); 1337f77e22a1SHong Zhang } 1338c9c72f8fSHong Zhang bdiag = b->diag; 1339c9c72f8fSHong Zhang 1340c9c72f8fSHong Zhang if (n > 0) { 134116a2bf60SHong Zhang ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr); 1342c9c72f8fSHong Zhang } 1343c9c72f8fSHong Zhang 1344c9c72f8fSHong Zhang /* set bi and bj with new data structure */ 1345c9c72f8fSHong Zhang bi = b->i; 1346c9c72f8fSHong Zhang bj = b->j; 1347c9c72f8fSHong Zhang 1348c9c72f8fSHong Zhang /* L part */ 1349c9c72f8fSHong Zhang bi[0] = 0; 1350c9c72f8fSHong Zhang for (i=0; i<n; i++){ 1351c9c72f8fSHong Zhang nz = adiag[i] - ai[i]; 1352c9c72f8fSHong Zhang bi[i+1] = bi[i] + nz; 1353c9c72f8fSHong Zhang aj = a->j + ai[i]; 1354c9c72f8fSHong Zhang for (j=0; j<nz; j++){ 1355c9c72f8fSHong Zhang *bj = aj[j]; bj++; 1356c9c72f8fSHong Zhang } 1357c9c72f8fSHong Zhang } 1358c9c72f8fSHong Zhang 1359c9c72f8fSHong Zhang /* U part */ 13608fc3a347SHong Zhang bi[n+1] = bi[n]; 1361c9c72f8fSHong Zhang for (i=n-1; i>=0; i--){ 1362c9c72f8fSHong Zhang nz = ai[i+1] - adiag[i] - 1; 13638fc3a347SHong Zhang bi[2*n-i+1] = bi[2*n-i] + nz + 1; 1364c9c72f8fSHong Zhang aj = a->j + adiag[i] + 1; 1365c9c72f8fSHong Zhang for (j=0; j<nz; j++){ 1366c9c72f8fSHong Zhang *bj = aj[j]; bj++; 1367c9c72f8fSHong Zhang } 1368c9c72f8fSHong Zhang /* diag[i] */ 1369c9c72f8fSHong Zhang *bj = i; bj++; 13708fc3a347SHong Zhang bdiag[i] = bi[2*n-i+1]-1; 1371c9c72f8fSHong Zhang } 1372c9c72f8fSHong Zhang PetscFunctionReturn(0); 1373c9c72f8fSHong Zhang } 1374c9c72f8fSHong Zhang 13754a2ae208SSatish Balay #undef __FUNCT__ 13766516239fSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_newdatastruct" 13776516239fSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_newdatastruct(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 13789e25ed09SBarry Smith { 1379416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 13809e25ed09SBarry Smith IS isicol; 13816849ba73SBarry Smith PetscErrorCode ierr; 13825d0c19d7SBarry Smith const PetscInt *r,*ic; 13835d0c19d7SBarry Smith PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,d; 13845a8e39fbSHong Zhang PetscInt *bi,*cols,nnz,*cols_lvl; 1385052f0c41SBarry Smith PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 13865a8e39fbSHong Zhang PetscInt i,levels,diagonal_fill; 138777c4ece6SBarry Smith PetscTruth col_identity,row_identity; 1388329f5518SBarry Smith PetscReal f; 13895a8e39fbSHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 13905a8e39fbSHong Zhang PetscBT lnkbt; 13915a8e39fbSHong Zhang PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 1392a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1393a1a86e44SBarry Smith PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 139409f38230SBarry Smith PetscTruth missing; 13959e25ed09SBarry Smith 13963a40ed3dSBarry Smith PetscFunctionBegin; 13976516239fSHong Zhang //printf("MatILUFactorSymbolic_SeqAIJ_newdatastruct ...\n"); 1398d0f46423SBarry 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); 13996cf997b0SBarry Smith f = info->fill; 140038baddfdSBarry Smith levels = (PetscInt)info->levels; 140138baddfdSBarry Smith diagonal_fill = (PetscInt)info->diagonal_fill; 14024c49b128SBarry Smith ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 140382bf6240SBarry Smith 1404be0abb6dSBarry Smith ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 1405be0abb6dSBarry Smith ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 1406c9c72f8fSHong Zhang 14076516239fSHong Zhang if (!levels && row_identity && col_identity) { 14086516239fSHong Zhang /* special case: ilu(0) with natural ordering */ 14098fc3a347SHong Zhang ierr = MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct(fact,A,isrow,iscol,info);CHKERRQ(ierr); 1410c9c72f8fSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct; 14116516239fSHong Zhang 14126516239fSHong Zhang fact->factor = MAT_FACTOR_ILU; 14136516239fSHong Zhang (fact)->info.factor_mallocs = 0; 14146516239fSHong Zhang (fact)->info.fill_ratio_given = info->fill; 14156516239fSHong Zhang (fact)->info.fill_ratio_needed = 1.0; 14166516239fSHong Zhang b = (Mat_SeqAIJ*)(fact)->data; 14176516239fSHong Zhang ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 14186516239fSHong Zhang if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 14196516239fSHong Zhang b->row = isrow; 14206516239fSHong Zhang b->col = iscol; 14216516239fSHong Zhang b->icol = isicol; 14226516239fSHong Zhang ierr = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 14236516239fSHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 14246516239fSHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 14256516239fSHong Zhang /* ierr = MatILUFactorSymbolic_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); */ 14266516239fSHong Zhang PetscFunctionReturn(0); 14276516239fSHong Zhang } 14286516239fSHong Zhang 14296516239fSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 14306516239fSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 14316516239fSHong Zhang 14326516239fSHong Zhang /* get new row pointers */ 14336516239fSHong Zhang ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 14346516239fSHong Zhang bi[0] = 0; 14356516239fSHong Zhang /* bdiag is location of diagonal in factor */ 14366516239fSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 14376516239fSHong Zhang bdiag[0] = 0; 14386516239fSHong Zhang 14396516239fSHong Zhang ierr = PetscMalloc((2*n+1)*sizeof(PetscInt**),&bj_ptr);CHKERRQ(ierr); 14406516239fSHong Zhang bjlvl_ptr = (PetscInt**)(bj_ptr + n); 14416516239fSHong Zhang 14426516239fSHong Zhang /* create a linked list for storing column indices of the active row */ 14436516239fSHong Zhang nlnk = n + 1; 14446516239fSHong Zhang ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 14456516239fSHong Zhang 14466516239fSHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 14476516239fSHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 14486516239fSHong Zhang current_space = free_space; 14496516239fSHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 14506516239fSHong Zhang current_space_lvl = free_space_lvl; 14516516239fSHong Zhang 14526516239fSHong Zhang for (i=0; i<n; i++) { 14536516239fSHong Zhang nzi = 0; 14546516239fSHong Zhang /* copy current row into linked list */ 14556516239fSHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 14566516239fSHong 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); 14576516239fSHong Zhang cols = aj + ai[r[i]]; 14586516239fSHong Zhang lnk[i] = -1; /* marker to indicate if diagonal exists */ 14596516239fSHong Zhang ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 14606516239fSHong Zhang nzi += nlnk; 14616516239fSHong Zhang 14626516239fSHong Zhang /* make sure diagonal entry is included */ 14636516239fSHong Zhang if (diagonal_fill && lnk[i] == -1) { 14646516239fSHong Zhang fm = n; 14656516239fSHong Zhang while (lnk[fm] < i) fm = lnk[fm]; 14666516239fSHong Zhang lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 14676516239fSHong Zhang lnk[fm] = i; 14686516239fSHong Zhang lnk_lvl[i] = 0; 14696516239fSHong Zhang nzi++; dcount++; 14706516239fSHong Zhang } 14716516239fSHong Zhang 14726516239fSHong Zhang /* add pivot rows into the active row */ 14736516239fSHong Zhang nzbd = 0; 14746516239fSHong Zhang prow = lnk[n]; 14756516239fSHong Zhang while (prow < i) { 14766516239fSHong Zhang nnz = bdiag[prow]; 14776516239fSHong Zhang cols = bj_ptr[prow] + nnz + 1; 14786516239fSHong Zhang cols_lvl = bjlvl_ptr[prow] + nnz + 1; 14796516239fSHong Zhang nnz = bi[prow+1] - bi[prow] - nnz - 1; 14806516239fSHong Zhang ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 14816516239fSHong Zhang nzi += nlnk; 14826516239fSHong Zhang prow = lnk[prow]; 14836516239fSHong Zhang nzbd++; 14846516239fSHong Zhang } 14856516239fSHong Zhang bdiag[i] = nzbd; 14866516239fSHong Zhang bi[i+1] = bi[i] + nzi; 14876516239fSHong Zhang 14886516239fSHong Zhang /* if free space is not available, make more free space */ 14896516239fSHong Zhang if (current_space->local_remaining<nzi) { 14906516239fSHong Zhang nnz = 2*nzi*(n - i); /* estimated and max additional space needed */ 14916516239fSHong Zhang ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 14926516239fSHong Zhang ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 14936516239fSHong Zhang reallocs++; 14946516239fSHong Zhang } 14956516239fSHong Zhang 14966516239fSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 14976516239fSHong Zhang ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 14986516239fSHong Zhang bj_ptr[i] = current_space->array; 14996516239fSHong Zhang bjlvl_ptr[i] = current_space_lvl->array; 15006516239fSHong Zhang 15016516239fSHong Zhang /* make sure the active row i has diagonal entry */ 15026516239fSHong Zhang if (*(bj_ptr[i]+bdiag[i]) != i) { 15036516239fSHong Zhang SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\ 15046516239fSHong Zhang try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i); 15056516239fSHong Zhang } 15066516239fSHong Zhang 15076516239fSHong Zhang current_space->array += nzi; 15086516239fSHong Zhang current_space->local_used += nzi; 15096516239fSHong Zhang current_space->local_remaining -= nzi; 15106516239fSHong Zhang current_space_lvl->array += nzi; 15116516239fSHong Zhang current_space_lvl->local_used += nzi; 15126516239fSHong Zhang current_space_lvl->local_remaining -= nzi; 15136516239fSHong Zhang } 15146516239fSHong Zhang 15156516239fSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 15166516239fSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 15176516239fSHong Zhang 15186516239fSHong Zhang /* destroy list of free space and other temporary arrays */ 15196516239fSHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 15206516239fSHong Zhang 15216516239fSHong Zhang /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */ 15226516239fSHong Zhang ierr = PetscFreeSpaceContiguous_newdatastruct(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 15236516239fSHong Zhang 15246516239fSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 15256516239fSHong Zhang ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 15266516239fSHong Zhang ierr = PetscFree(bj_ptr);CHKERRQ(ierr); 15276516239fSHong Zhang 15286516239fSHong Zhang #if defined(PETSC_USE_INFO) 15296516239fSHong Zhang { 15306516239fSHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 15316516239fSHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 15326516239fSHong Zhang ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 15336516239fSHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 15346516239fSHong Zhang ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 15356516239fSHong Zhang if (diagonal_fill) { 15366516239fSHong Zhang ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 15376516239fSHong Zhang } 15386516239fSHong Zhang } 15396516239fSHong Zhang #endif 15406516239fSHong Zhang 15416516239fSHong Zhang /* put together the new matrix */ 15426516239fSHong Zhang ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 15436516239fSHong Zhang ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 15446516239fSHong Zhang b = (Mat_SeqAIJ*)(fact)->data; 15456516239fSHong Zhang b->free_a = PETSC_TRUE; 15466516239fSHong Zhang b->free_ij = PETSC_TRUE; 15476516239fSHong Zhang b->singlemalloc = PETSC_FALSE; 15486516239fSHong Zhang ierr = PetscMalloc( (bi[2*n+1] )*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 15496516239fSHong Zhang b->j = bj; 15506516239fSHong Zhang b->i = bi; 15516516239fSHong Zhang b->diag = bdiag; 15526516239fSHong Zhang b->ilen = 0; 15536516239fSHong Zhang b->imax = 0; 15546516239fSHong Zhang b->row = isrow; 15556516239fSHong Zhang b->col = iscol; 15566516239fSHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 15576516239fSHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 15586516239fSHong Zhang b->icol = isicol; 15596516239fSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 15606516239fSHong Zhang /* In b structure: Free imax, ilen, old a, old j. 15616516239fSHong Zhang Allocate bdiag, solve_work, new a, new j */ 15626516239fSHong Zhang ierr = PetscLogObjectMemory(fact,bi[2*n+1] * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 15636516239fSHong Zhang b->maxnz = b->nz = bi[2*n+1] ; 15646516239fSHong Zhang (fact)->info.factor_mallocs = reallocs; 15656516239fSHong Zhang (fact)->info.fill_ratio_given = f; 15666516239fSHong Zhang (fact)->info.fill_ratio_needed = ((PetscReal)bi[2*n+1])/((PetscReal)ai[n]); 15676516239fSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct; 15686516239fSHong Zhang /* ierr = MatILUFactorSymbolic_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); */ 15696516239fSHong Zhang PetscFunctionReturn(0); 15706516239fSHong Zhang } 15716516239fSHong Zhang 15726516239fSHong Zhang #undef __FUNCT__ 15736516239fSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ" 15746516239fSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 15756516239fSHong Zhang { 15766516239fSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 15776516239fSHong Zhang IS isicol; 15786516239fSHong Zhang PetscErrorCode ierr; 15796516239fSHong Zhang const PetscInt *r,*ic; 15806516239fSHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,d; 15816516239fSHong Zhang PetscInt *bi,*cols,nnz,*cols_lvl; 15826516239fSHong Zhang PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 15836516239fSHong Zhang PetscInt i,levels,diagonal_fill; 15846516239fSHong Zhang PetscTruth col_identity,row_identity; 15856516239fSHong Zhang PetscReal f; 15866516239fSHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 15876516239fSHong Zhang PetscBT lnkbt; 15886516239fSHong Zhang PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 15896516239fSHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 15906516239fSHong Zhang PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 15916516239fSHong Zhang PetscTruth missing; 15926516239fSHong Zhang PetscTruth newdatastruct=PETSC_FALSE; 15936516239fSHong Zhang 15946516239fSHong Zhang PetscFunctionBegin; 15956516239fSHong Zhang ierr = PetscOptionsGetTruth(PETSC_NULL,"-ilu_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr); 15966516239fSHong Zhang if (newdatastruct){ 15976516239fSHong Zhang ierr = MatILUFactorSymbolic_SeqAIJ_newdatastruct(fact,A,isrow,iscol,info);CHKERRQ(ierr); 15986516239fSHong Zhang PetscFunctionReturn(0); 15996516239fSHong Zhang } 16006516239fSHong Zhang 16016516239fSHong 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); 16026516239fSHong Zhang f = info->fill; 16036516239fSHong Zhang levels = (PetscInt)info->levels; 16046516239fSHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 16056516239fSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 16066516239fSHong Zhang 16076516239fSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 16086516239fSHong Zhang ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 16096516239fSHong Zhang if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */ 1610f77e22a1SHong Zhang ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr); 1611c9c72f8fSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 1612c9c72f8fSHong Zhang 1613719d5645SBarry Smith fact->factor = MAT_FACTOR_ILU; 1614719d5645SBarry Smith (fact)->info.factor_mallocs = 0; 1615719d5645SBarry Smith (fact)->info.fill_ratio_given = info->fill; 1616719d5645SBarry Smith (fact)->info.fill_ratio_needed = 1.0; 1617719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 16188ef3462fSBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 161909f38230SBarry Smith if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 162008480c60SBarry Smith b->row = isrow; 162108480c60SBarry Smith b->col = iscol; 162282bf6240SBarry Smith b->icol = isicol; 1623719d5645SBarry Smith ierr = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1624c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1625c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1626719d5645SBarry Smith ierr = MatILUFactorSymbolic_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); 16273a40ed3dSBarry Smith PetscFunctionReturn(0); 162808480c60SBarry Smith } 162908480c60SBarry Smith 1630898183ecSLois Curfman McInnes ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1631898183ecSLois Curfman McInnes ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 16329e25ed09SBarry Smith 16339e25ed09SBarry Smith /* get new row pointers */ 16345a8e39fbSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 16355a8e39fbSHong Zhang bi[0] = 0; 16365a8e39fbSHong Zhang /* bdiag is location of diagonal in factor */ 16375a8e39fbSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 16385a8e39fbSHong Zhang bdiag[0] = 0; 16396cf997b0SBarry Smith 16405a8e39fbSHong Zhang ierr = PetscMalloc((2*n+1)*sizeof(PetscInt**),&bj_ptr);CHKERRQ(ierr); 16415a8e39fbSHong Zhang bjlvl_ptr = (PetscInt**)(bj_ptr + n); 16425a8e39fbSHong Zhang 16435a8e39fbSHong Zhang /* create a linked list for storing column indices of the active row */ 16445a8e39fbSHong Zhang nlnk = n + 1; 16455a8e39fbSHong Zhang ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 16465a8e39fbSHong Zhang 16475a8e39fbSHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 1648a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 16495a8e39fbSHong Zhang current_space = free_space; 1650a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 16515a8e39fbSHong Zhang current_space_lvl = free_space_lvl; 16525a8e39fbSHong Zhang 16535a8e39fbSHong Zhang for (i=0; i<n; i++) { 16545a8e39fbSHong Zhang nzi = 0; 16556cf997b0SBarry Smith /* copy current row into linked list */ 16565a8e39fbSHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 16573b4a8b6dSBarry 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); 16585a8e39fbSHong Zhang cols = aj + ai[r[i]]; 16595a8e39fbSHong Zhang lnk[i] = -1; /* marker to indicate if diagonal exists */ 16605a8e39fbSHong Zhang ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 16615a8e39fbSHong Zhang nzi += nlnk; 16626cf997b0SBarry Smith 16636cf997b0SBarry Smith /* make sure diagonal entry is included */ 16645a8e39fbSHong Zhang if (diagonal_fill && lnk[i] == -1) { 16656cf997b0SBarry Smith fm = n; 16665a8e39fbSHong Zhang while (lnk[fm] < i) fm = lnk[fm]; 16675a8e39fbSHong Zhang lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 16685a8e39fbSHong Zhang lnk[fm] = i; 16695a8e39fbSHong Zhang lnk_lvl[i] = 0; 16705a8e39fbSHong Zhang nzi++; dcount++; 16716cf997b0SBarry Smith } 16726cf997b0SBarry Smith 16735a8e39fbSHong Zhang /* add pivot rows into the active row */ 16745a8e39fbSHong Zhang nzbd = 0; 16755a8e39fbSHong Zhang prow = lnk[n]; 16765a8e39fbSHong Zhang while (prow < i) { 16775a8e39fbSHong Zhang nnz = bdiag[prow]; 16785a8e39fbSHong Zhang cols = bj_ptr[prow] + nnz + 1; 16795a8e39fbSHong Zhang cols_lvl = bjlvl_ptr[prow] + nnz + 1; 16805a8e39fbSHong Zhang nnz = bi[prow+1] - bi[prow] - nnz - 1; 16815a8e39fbSHong Zhang ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 16825a8e39fbSHong Zhang nzi += nlnk; 16835a8e39fbSHong Zhang prow = lnk[prow]; 16845a8e39fbSHong Zhang nzbd++; 168556beaf04SBarry Smith } 16865a8e39fbSHong Zhang bdiag[i] = nzbd; 16875a8e39fbSHong Zhang bi[i+1] = bi[i] + nzi; 1688ecf371e4SBarry Smith 16895a8e39fbSHong Zhang /* if free space is not available, make more free space */ 16905a8e39fbSHong Zhang if (current_space->local_remaining<nzi) { 16915a8e39fbSHong Zhang nnz = nzi*(n - i); /* estimated and max additional space needed */ 1692a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 1693a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 16945a8e39fbSHong Zhang reallocs++; 16955a8e39fbSHong Zhang } 1696ecf371e4SBarry Smith 16975a8e39fbSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 16985a8e39fbSHong Zhang ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 16995a8e39fbSHong Zhang bj_ptr[i] = current_space->array; 17005a8e39fbSHong Zhang bjlvl_ptr[i] = current_space_lvl->array; 17015a8e39fbSHong Zhang 17025a8e39fbSHong Zhang /* make sure the active row i has diagonal entry */ 17035a8e39fbSHong Zhang if (*(bj_ptr[i]+bdiag[i]) != i) { 170477431f27SBarry Smith SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\ 1705d7ee0231SBarry Smith try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i); 17066cf997b0SBarry Smith } 17075a8e39fbSHong Zhang 17085a8e39fbSHong Zhang current_space->array += nzi; 17095a8e39fbSHong Zhang current_space->local_used += nzi; 17105a8e39fbSHong Zhang current_space->local_remaining -= nzi; 17115a8e39fbSHong Zhang current_space_lvl->array += nzi; 17125a8e39fbSHong Zhang current_space_lvl->local_used += nzi; 17135a8e39fbSHong Zhang current_space_lvl->local_remaining -= nzi; 17149e25ed09SBarry Smith } 17155a8e39fbSHong Zhang 1716898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 1717898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 17185a8e39fbSHong Zhang 17195a8e39fbSHong Zhang /* destroy list of free space and other temporary arrays */ 17205a8e39fbSHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 17216516239fSHong Zhang ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */ 17225a8e39fbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 1723a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 17245a8e39fbSHong Zhang ierr = PetscFree(bj_ptr);CHKERRQ(ierr); 17259e25ed09SBarry Smith 17266cf91177SBarry Smith #if defined(PETSC_USE_INFO) 1727f64065bbSBarry Smith { 17285a8e39fbSHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 1729ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 1730ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 1731ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 1732ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 1733335d9088SBarry Smith if (diagonal_fill) { 1734ae15b995SBarry Smith ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 1735335d9088SBarry Smith } 1736f64065bbSBarry Smith } 173763ba0a88SBarry Smith #endif 1738bbb6d6a8SBarry Smith 17399e25ed09SBarry Smith /* put together the new matrix */ 174007dbf95fSBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 1741719d5645SBarry Smith ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 1742719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 1743e6b907acSBarry Smith b->free_a = PETSC_TRUE; 1744e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 17457c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 1746052f0c41SBarry Smith ierr = PetscMalloc( (bi[n] )*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 17475a8e39fbSHong Zhang b->j = bj; 17485a8e39fbSHong Zhang b->i = bi; 17495a8e39fbSHong Zhang for (i=0; i<n; i++) bdiag[i] += bi[i]; 17505a8e39fbSHong Zhang b->diag = bdiag; 1751416022c9SBarry Smith b->ilen = 0; 1752416022c9SBarry Smith b->imax = 0; 1753416022c9SBarry Smith b->row = isrow; 1754416022c9SBarry Smith b->col = iscol; 1755c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1756c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 175782bf6240SBarry Smith b->icol = isicol; 175887828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1759416022c9SBarry Smith /* In b structure: Free imax, ilen, old a, old j. 17605a8e39fbSHong Zhang Allocate bdiag, solve_work, new a, new j */ 1761719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 17625a8e39fbSHong Zhang b->maxnz = b->nz = bi[n] ; 1763719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 1764719d5645SBarry Smith (fact)->info.fill_ratio_given = f; 1765719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 1766719d5645SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 1767719d5645SBarry Smith ierr = MatILUFactorSymbolic_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); 17683a40ed3dSBarry Smith PetscFunctionReturn(0); 17699e25ed09SBarry Smith } 1770d5d45c9bSBarry Smith 17717c4f633dSBarry Smith #include "../src/mat/impls/sbaij/seq/sbaij.h" 1772a6175056SHong Zhang #undef __FUNCT__ 1773*5f44c854SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ_newdatastruct" 1774*5f44c854SHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_newdatastruct(Mat B,Mat A,const MatFactorInfo *info) 1775*5f44c854SHong Zhang { 1776*5f44c854SHong Zhang Mat C = B; 1777*5f44c854SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 1778*5f44c854SHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 1779*5f44c854SHong Zhang IS ip=b->row,iip = b->icol; 1780*5f44c854SHong Zhang PetscErrorCode ierr; 1781*5f44c854SHong Zhang const PetscInt *rip,*riip; 1782*5f44c854SHong Zhang PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp; 1783*5f44c854SHong Zhang PetscInt *ai=a->i,*aj=a->j; 1784*5f44c854SHong Zhang PetscInt k,jmin,jmax,*c2r,*il,col,nexti,ili,nz; 1785*5f44c854SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 1786*5f44c854SHong Zhang PetscReal zeropivot,shiftnz; 1787*5f44c854SHong Zhang PetscReal shiftpd; 1788*5f44c854SHong Zhang PetscTruth perm_identity; 1789*5f44c854SHong Zhang 1790*5f44c854SHong Zhang PetscFunctionBegin; 1791*5f44c854SHong Zhang 1792*5f44c854SHong Zhang shiftnz = info->shiftnz; 1793*5f44c854SHong Zhang shiftpd = info->shiftpd; 1794*5f44c854SHong Zhang zeropivot = info->zeropivot; 1795*5f44c854SHong Zhang 1796*5f44c854SHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 1797*5f44c854SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 1798*5f44c854SHong Zhang 1799*5f44c854SHong Zhang /* allocate working arrays 1800*5f44c854SHong Zhang c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col 1801*5f44c854SHong Zhang il: for active k row, il[i] gives the index of the 1st nonzero entry in U[i,k:n-1] in bj and ba arrays 1802*5f44c854SHong Zhang */ 1803*5f44c854SHong Zhang nz = (2*mbs+1)*sizeof(PetscInt)+mbs*sizeof(MatScalar); 1804*5f44c854SHong Zhang ierr = PetscMalloc(nz,&il);CHKERRQ(ierr); 1805*5f44c854SHong Zhang c2r = il + mbs; 1806*5f44c854SHong Zhang rtmp = (MatScalar*)(c2r + mbs); 1807*5f44c854SHong Zhang 1808*5f44c854SHong Zhang for (i=0; i<mbs; i++) { 1809*5f44c854SHong Zhang c2r[i] = mbs; 1810*5f44c854SHong Zhang } 1811*5f44c854SHong Zhang il[0] = 0; 1812*5f44c854SHong Zhang 1813*5f44c854SHong Zhang for (k = 0; k<mbs; k++){ 1814*5f44c854SHong Zhang /* zero rtmp */ 1815*5f44c854SHong Zhang nz = bi[k+1] - bi[k]; 1816*5f44c854SHong Zhang bjtmp = bj + bi[k]; 1817*5f44c854SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 1818*5f44c854SHong Zhang 1819*5f44c854SHong Zhang /* load in initial unfactored row */ 1820*5f44c854SHong Zhang bval = ba + bi[k]; 1821*5f44c854SHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 1822*5f44c854SHong Zhang for (j = jmin; j < jmax; j++){ 1823*5f44c854SHong Zhang col = riip[aj[j]]; 1824*5f44c854SHong Zhang if (col >= k){ /* only take upper triangular entry */ 1825*5f44c854SHong Zhang rtmp[col] = aa[j]; 1826*5f44c854SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 1827*5f44c854SHong Zhang } 1828*5f44c854SHong Zhang } 1829*5f44c854SHong Zhang /* shift the diagonal of the matrix */ 1830*5f44c854SHong Zhang 1831*5f44c854SHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 1832*5f44c854SHong Zhang dk = rtmp[k]; 1833*5f44c854SHong Zhang i = c2r[k]; /* first row to be added to k_th row */ 1834*5f44c854SHong Zhang 1835*5f44c854SHong Zhang while (i < k){ 1836*5f44c854SHong Zhang nexti = c2r[i]; /* next row to be added to k_th row */ 1837*5f44c854SHong Zhang 1838*5f44c854SHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 1839*5f44c854SHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 1840*5f44c854SHong Zhang uikdi = - ba[ili]*ba[bdiag[i]]; /* diagonal(k) */ 1841*5f44c854SHong Zhang dk += uikdi*ba[ili]; /* update diag[k] */ 1842*5f44c854SHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 1843*5f44c854SHong Zhang 1844*5f44c854SHong Zhang /* add multiple of row i to k-th row */ 1845*5f44c854SHong Zhang jmin = ili + 1; jmax = bi[i+1]; 1846*5f44c854SHong Zhang if (jmin < jmax){ 1847*5f44c854SHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 1848*5f44c854SHong Zhang /* update il and c2r for row i */ 1849*5f44c854SHong Zhang il[i] = jmin; 1850*5f44c854SHong Zhang j = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i; 1851*5f44c854SHong Zhang } 1852*5f44c854SHong Zhang i = nexti; 1853*5f44c854SHong Zhang } 1854*5f44c854SHong Zhang 1855*5f44c854SHong Zhang /* copy data into U(k,:) */ 1856*5f44c854SHong Zhang ba[bdiag[k]] = 1.0/dk; /* U(k,k) */ 1857*5f44c854SHong Zhang jmin = bi[k]; jmax = bi[k+1]-1; 1858*5f44c854SHong Zhang if (jmin < jmax) { 1859*5f44c854SHong Zhang for (j=jmin; j<jmax; j++){ 1860*5f44c854SHong Zhang col = bj[j]; ba[j] = rtmp[col]; 1861*5f44c854SHong Zhang } 1862*5f44c854SHong Zhang /* add the k-th row into il and c2r */ 1863*5f44c854SHong Zhang il[k] = jmin; 1864*5f44c854SHong Zhang i = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k; 1865*5f44c854SHong Zhang } 1866*5f44c854SHong Zhang } 1867*5f44c854SHong Zhang 1868*5f44c854SHong Zhang ierr = PetscFree(il);CHKERRQ(ierr); 1869*5f44c854SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 1870*5f44c854SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 1871*5f44c854SHong Zhang 1872*5f44c854SHong Zhang ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 1873*5f44c854SHong Zhang if (perm_identity){ 1874*5f44c854SHong Zhang (B)->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering_newdatastruct; 1875*5f44c854SHong Zhang (B)->ops->solvetranspose = 0; 1876*5f44c854SHong Zhang (B)->ops->forwardsolve = 0; 1877*5f44c854SHong Zhang (B)->ops->backwardsolve = 0; 1878*5f44c854SHong Zhang } else { 1879*5f44c854SHong Zhang (B)->ops->solve = 0; 1880*5f44c854SHong Zhang (B)->ops->solvetranspose = 0; 1881*5f44c854SHong Zhang (B)->ops->forwardsolve = 0; 1882*5f44c854SHong Zhang (B)->ops->backwardsolve = 0; 1883*5f44c854SHong Zhang } 1884*5f44c854SHong Zhang 1885*5f44c854SHong Zhang C->assembled = PETSC_TRUE; 1886*5f44c854SHong Zhang C->preallocated = PETSC_TRUE; 1887*5f44c854SHong Zhang ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 1888*5f44c854SHong Zhang PetscFunctionReturn(0); 1889*5f44c854SHong Zhang } 1890*5f44c854SHong Zhang 1891*5f44c854SHong Zhang #undef __FUNCT__ 1892a6175056SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ" 18930481f469SBarry Smith PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 1894a6175056SHong Zhang { 1895719d5645SBarry Smith Mat C = B; 18960968510aSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 18972ed133dbSHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 18989bfd6278SHong Zhang IS ip=b->row,iip = b->icol; 18992ed133dbSHong Zhang PetscErrorCode ierr; 19005d0c19d7SBarry Smith const PetscInt *rip,*riip; 19015d0c19d7SBarry Smith PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol; 19022ed133dbSHong Zhang PetscInt *ai=a->i,*aj=a->j; 19032ed133dbSHong Zhang PetscInt k,jmin,jmax,*jl,*il,col,nexti,ili,nz; 1904622e2028SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 1905540703acSHong Zhang PetscReal zeropivot,rs,shiftnz; 1906fbf22428SSatish Balay PetscReal shiftpd; 1907540703acSHong Zhang ChShift_Ctx sctx; 1908540703acSHong Zhang PetscInt newshift; 1909db4efbfdSBarry Smith PetscTruth perm_identity; 1910d5d45c9bSBarry Smith 1911a6175056SHong Zhang PetscFunctionBegin; 1912540703acSHong Zhang shiftnz = info->shiftnz; 1913540703acSHong Zhang shiftpd = info->shiftpd; 1914ee45ca4aSHong Zhang zeropivot = info->zeropivot; 1915ee45ca4aSHong Zhang 19162ed133dbSHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 19179bfd6278SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 19182ed133dbSHong Zhang 19192ed133dbSHong Zhang /* initialization */ 19202ed133dbSHong Zhang nz = (2*mbs+1)*sizeof(PetscInt)+mbs*sizeof(MatScalar); 19212ed133dbSHong Zhang ierr = PetscMalloc(nz,&il);CHKERRQ(ierr); 19222ed133dbSHong Zhang jl = il + mbs; 19232ed133dbSHong Zhang rtmp = (MatScalar*)(jl + mbs); 19242ed133dbSHong Zhang 1925540703acSHong Zhang sctx.shift_amount = 0; 1926540703acSHong Zhang sctx.nshift = 0; 19272ed133dbSHong Zhang do { 1928540703acSHong Zhang sctx.chshift = PETSC_FALSE; 19292ed133dbSHong Zhang for (i=0; i<mbs; i++) { 19302ed133dbSHong Zhang rtmp[i] = 0.0; jl[i] = mbs; il[0] = 0; 1931a6175056SHong Zhang } 19322ed133dbSHong Zhang 19332ed133dbSHong Zhang for (k = 0; k<mbs; k++){ 1934c05c3958SHong Zhang bval = ba + bi[k]; 19352ed133dbSHong Zhang /* initialize k-th row by the perm[k]-th row of A */ 19362ed133dbSHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 19372ed133dbSHong Zhang for (j = jmin; j < jmax; j++){ 19389bfd6278SHong Zhang col = riip[aj[j]]; 19392ed133dbSHong Zhang if (col >= k){ /* only take upper triangular entry */ 19402ed133dbSHong Zhang rtmp[col] = aa[j]; 1941c05c3958SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 19422ed133dbSHong Zhang } 19432ed133dbSHong Zhang } 194439e3d630SHong Zhang /* shift the diagonal of the matrix */ 1945540703acSHong Zhang if (sctx.nshift) rtmp[k] += sctx.shift_amount; 19462ed133dbSHong Zhang 19472ed133dbSHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 19482ed133dbSHong Zhang dk = rtmp[k]; 19492ed133dbSHong Zhang i = jl[k]; /* first row to be added to k_th row */ 19502ed133dbSHong Zhang 19512ed133dbSHong Zhang while (i < k){ 19522ed133dbSHong Zhang nexti = jl[i]; /* next row to be added to k_th row */ 19532ed133dbSHong Zhang 19542ed133dbSHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 19552ed133dbSHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 19562ed133dbSHong Zhang uikdi = - ba[ili]*ba[bi[i]]; /* diagonal(k) */ 19572ed133dbSHong Zhang dk += uikdi*ba[ili]; 19582ed133dbSHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 19592ed133dbSHong Zhang 19602ed133dbSHong Zhang /* add multiple of row i to k-th row */ 19612ed133dbSHong Zhang jmin = ili + 1; jmax = bi[i+1]; 19622ed133dbSHong Zhang if (jmin < jmax){ 19632ed133dbSHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 19642ed133dbSHong Zhang /* update il and jl for row i */ 19652ed133dbSHong Zhang il[i] = jmin; 19662ed133dbSHong Zhang j = bj[jmin]; jl[i] = jl[j]; jl[j] = i; 19672ed133dbSHong Zhang } 19682ed133dbSHong Zhang i = nexti; 19692ed133dbSHong Zhang } 19702ed133dbSHong Zhang 19710697b6caSHong Zhang /* shift the diagonals when zero pivot is detected */ 19720697b6caSHong Zhang /* compute rs=sum of abs(off-diagonal) */ 19730697b6caSHong Zhang rs = 0.0; 19743c9e8598SHong Zhang jmin = bi[k]+1; 19753c9e8598SHong Zhang nz = bi[k+1] - jmin; 19763c9e8598SHong Zhang bcol = bj + jmin; 19773c9e8598SHong Zhang while (nz--){ 197889f845c8SHong Zhang rs += PetscAbsScalar(rtmp[*bcol]); 197989f845c8SHong Zhang bcol++; 19803c9e8598SHong Zhang } 1981540703acSHong Zhang 1982540703acSHong Zhang sctx.rs = rs; 1983540703acSHong Zhang sctx.pv = dk; 19845b5bc046SBarry Smith ierr = MatCholeskyCheckShift_inline(info,sctx,k,newshift);CHKERRQ(ierr); 1985117f1891Stmunson 1986117f1891Stmunson if (newshift == 1) { 1987117f1891Stmunson if (!sctx.shift_amount) { 1988117f1891Stmunson sctx.shift_amount = 1e-5; 1989117f1891Stmunson } 1990117f1891Stmunson break; 1991117f1891Stmunson } 19923c9e8598SHong Zhang 19933c9e8598SHong Zhang /* copy data into U(k,:) */ 199439e3d630SHong Zhang ba[bi[k]] = 1.0/dk; /* U(k,k) */ 199539e3d630SHong Zhang jmin = bi[k]+1; jmax = bi[k+1]; 199639e3d630SHong Zhang if (jmin < jmax) { 199739e3d630SHong Zhang for (j=jmin; j<jmax; j++){ 199839e3d630SHong Zhang col = bj[j]; ba[j] = rtmp[col]; rtmp[col] = 0.0; 19993c9e8598SHong Zhang } 200039e3d630SHong Zhang /* add the k-th row into il and jl */ 20013c9e8598SHong Zhang il[k] = jmin; 20023c9e8598SHong Zhang i = bj[jmin]; jl[k] = jl[i]; jl[i] = k; 20033c9e8598SHong Zhang } 20043c9e8598SHong Zhang } 2005540703acSHong Zhang } while (sctx.chshift); 20063c9e8598SHong Zhang ierr = PetscFree(il);CHKERRQ(ierr); 20073c9e8598SHong Zhang 200839e3d630SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 20099bfd6278SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 2010db4efbfdSBarry Smith 2011db4efbfdSBarry Smith ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 2012db4efbfdSBarry Smith if (perm_identity){ 2013719d5645SBarry Smith (B)->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering; 2014719d5645SBarry Smith (B)->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering; 2015719d5645SBarry Smith (B)->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering; 2016719d5645SBarry Smith (B)->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering; 2017db4efbfdSBarry Smith } else { 2018719d5645SBarry Smith (B)->ops->solve = MatSolve_SeqSBAIJ_1; 2019719d5645SBarry Smith (B)->ops->solvetranspose = MatSolve_SeqSBAIJ_1; 2020719d5645SBarry Smith (B)->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1; 2021719d5645SBarry Smith (B)->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1; 2022db4efbfdSBarry Smith } 2023db4efbfdSBarry Smith 20243c9e8598SHong Zhang C->assembled = PETSC_TRUE; 20253c9e8598SHong Zhang C->preallocated = PETSC_TRUE; 2026d0f46423SBarry Smith ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 2027540703acSHong Zhang if (sctx.nshift){ 2028540703acSHong Zhang if (shiftnz) { 20291e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 2030540703acSHong Zhang } else if (shiftpd) { 20311e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 20320697b6caSHong Zhang } 20333c9e8598SHong Zhang } 20343c9e8598SHong Zhang PetscFunctionReturn(0); 20353c9e8598SHong Zhang } 2036a6175056SHong Zhang 2037a6175056SHong Zhang #undef __FUNCT__ 2038*5f44c854SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ_newdatastruct" 2039*5f44c854SHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_newdatastruct(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2040a6175056SHong Zhang { 20410968510aSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2042ed59401aSHong Zhang Mat_SeqSBAIJ *b; 2043dfbe8321SBarry Smith PetscErrorCode ierr; 204458ebbce7SBarry Smith PetscTruth perm_identity,missing; 2045*5f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 20465d0c19d7SBarry Smith const PetscInt *rip,*riip; 2047ed59401aSHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 204858ebbce7SBarry Smith PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 20495a8e39fbSHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 2050ed59401aSHong Zhang PetscReal fill=info->fill,levels=info->levels; 2051a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 2052a1a86e44SBarry Smith PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 20537a48dd6fSHong Zhang PetscBT lnkbt; 2054b635d36bSHong Zhang IS iperm; 2055a6175056SHong Zhang 2056a6175056SHong Zhang PetscFunctionBegin; 2057d0f46423SBarry 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); 205858ebbce7SBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 205958ebbce7SBarry Smith if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 2060653a6975SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 2061b635d36bSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 20627a48dd6fSHong Zhang 206339e3d630SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 2064*5f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 206539e3d630SHong Zhang ui[0] = 0; 206639e3d630SHong Zhang 2067b635d36bSHong Zhang /* ICC(0) without matrix ordering: simply copies fill pattern */ 2068622e2028SHong Zhang if (!levels && perm_identity) { 206958ebbce7SBarry Smith 2070ed59401aSHong Zhang for (i=0; i<am; i++) { 207139e3d630SHong Zhang ui[i+1] = ui[i] + ai[i+1] - a->diag[i]; 2072*5f44c854SHong Zhang udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */ 2073*5f44c854SHong Zhang } 2074*5f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2075*5f44c854SHong Zhang cols = uj; 2076*5f44c854SHong Zhang for (i=0; i<am; i++) { 2077*5f44c854SHong Zhang aj = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */ 2078*5f44c854SHong Zhang ncols = ui[i+1] - ui[i] - 1; 2079*5f44c854SHong Zhang for (j=0; j<ncols; j++) *cols++ = aj[j]; 2080*5f44c854SHong Zhang *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */ 2081*5f44c854SHong Zhang } 2082*5f44c854SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 2083*5f44c854SHong Zhang SETERRQ(1,"Not done yet"); 2084*5f44c854SHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 2085*5f44c854SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 2086*5f44c854SHong Zhang 2087*5f44c854SHong Zhang /* initialization */ 2088*5f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 2089*5f44c854SHong Zhang 2090*5f44c854SHong Zhang /* jl: linked list for storing indices of the pivot rows 2091*5f44c854SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 2092*5f44c854SHong Zhang ierr = PetscMalloc((2*am+1)*sizeof(PetscInt)+2*am*sizeof(PetscInt**),&jl);CHKERRQ(ierr); 2093*5f44c854SHong Zhang il = jl + am; 2094*5f44c854SHong Zhang uj_ptr = (PetscInt**)(il + am); 2095*5f44c854SHong Zhang uj_lvl_ptr = (PetscInt**)(uj_ptr + am); 2096*5f44c854SHong Zhang for (i=0; i<am; i++){ 2097*5f44c854SHong Zhang jl[i] = am; il[i] = 0; 2098*5f44c854SHong Zhang } 2099*5f44c854SHong Zhang 2100*5f44c854SHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 2101*5f44c854SHong Zhang nlnk = am + 1; 2102*5f44c854SHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 2103*5f44c854SHong Zhang 2104*5f44c854SHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 2105*5f44c854SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 2106*5f44c854SHong Zhang current_space = free_space; 2107*5f44c854SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr); 2108*5f44c854SHong Zhang current_space_lvl = free_space_lvl; 2109*5f44c854SHong Zhang 2110*5f44c854SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 2111*5f44c854SHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 2112*5f44c854SHong Zhang nzk = 0; 2113*5f44c854SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 2114*5f44c854SHong Zhang if (!ncols) SETERRQ2(PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k); 2115*5f44c854SHong Zhang ncols_upper = 0; 2116*5f44c854SHong Zhang for (j=0; j<ncols; j++){ 2117*5f44c854SHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 2118*5f44c854SHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 2119*5f44c854SHong Zhang ajtmp[ncols_upper] = i; 2120*5f44c854SHong Zhang ncols_upper++; 2121*5f44c854SHong Zhang } 2122*5f44c854SHong Zhang } 2123*5f44c854SHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 2124*5f44c854SHong Zhang nzk += nlnk; 2125*5f44c854SHong Zhang 2126*5f44c854SHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 2127*5f44c854SHong Zhang prow = jl[k]; /* 1st pivot row */ 2128*5f44c854SHong Zhang 2129*5f44c854SHong Zhang while (prow < k){ 2130*5f44c854SHong Zhang nextprow = jl[prow]; 2131*5f44c854SHong Zhang 2132*5f44c854SHong Zhang /* merge prow into k-th row */ 2133*5f44c854SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 2134*5f44c854SHong Zhang jmax = ui[prow+1]; 2135*5f44c854SHong Zhang ncols = jmax-jmin; 2136*5f44c854SHong Zhang i = jmin - ui[prow]; 2137*5f44c854SHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 2138*5f44c854SHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 2139*5f44c854SHong Zhang j = *(uj - 1); 2140*5f44c854SHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 2141*5f44c854SHong Zhang nzk += nlnk; 2142*5f44c854SHong Zhang 2143*5f44c854SHong Zhang /* update il and jl for prow */ 2144*5f44c854SHong Zhang if (jmin < jmax){ 2145*5f44c854SHong Zhang il[prow] = jmin; 2146*5f44c854SHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 2147*5f44c854SHong Zhang } 2148*5f44c854SHong Zhang prow = nextprow; 2149*5f44c854SHong Zhang } 2150*5f44c854SHong Zhang 2151*5f44c854SHong Zhang /* if free space is not available, make more free space */ 2152*5f44c854SHong Zhang if (current_space->local_remaining<nzk) { 2153*5f44c854SHong Zhang i = am - k + 1; /* num of unfactored rows */ 2154*5f44c854SHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 2155*5f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 2156*5f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 2157*5f44c854SHong Zhang reallocs++; 2158*5f44c854SHong Zhang } 2159*5f44c854SHong Zhang 2160*5f44c854SHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 2161*5f44c854SHong Zhang if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 2162*5f44c854SHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 2163*5f44c854SHong Zhang 2164*5f44c854SHong Zhang /* add the k-th row into il and jl */ 2165*5f44c854SHong Zhang if (nzk > 1){ 2166*5f44c854SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 2167*5f44c854SHong Zhang jl[k] = jl[i]; jl[i] = k; 2168*5f44c854SHong Zhang il[k] = ui[k] + 1; 2169*5f44c854SHong Zhang } 2170*5f44c854SHong Zhang uj_ptr[k] = current_space->array; 2171*5f44c854SHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 2172*5f44c854SHong Zhang 2173*5f44c854SHong Zhang current_space->array += nzk; 2174*5f44c854SHong Zhang current_space->local_used += nzk; 2175*5f44c854SHong Zhang current_space->local_remaining -= nzk; 2176*5f44c854SHong Zhang 2177*5f44c854SHong Zhang current_space_lvl->array += nzk; 2178*5f44c854SHong Zhang current_space_lvl->local_used += nzk; 2179*5f44c854SHong Zhang current_space_lvl->local_remaining -= nzk; 2180*5f44c854SHong Zhang 2181*5f44c854SHong Zhang ui[k+1] = ui[k] + nzk; 2182*5f44c854SHong Zhang } 2183*5f44c854SHong Zhang 2184*5f44c854SHong Zhang #if defined(PETSC_USE_INFO) 2185*5f44c854SHong Zhang if (ai[am] != 0) { 2186*5f44c854SHong Zhang PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]); 2187*5f44c854SHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 2188*5f44c854SHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2189*5f44c854SHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 2190*5f44c854SHong Zhang } else { 2191*5f44c854SHong Zhang ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 2192*5f44c854SHong Zhang } 2193*5f44c854SHong Zhang #endif 2194*5f44c854SHong Zhang 2195*5f44c854SHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 2196*5f44c854SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 2197*5f44c854SHong Zhang ierr = PetscFree(jl);CHKERRQ(ierr); 2198*5f44c854SHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 2199*5f44c854SHong Zhang 2200*5f44c854SHong Zhang /* destroy list of free space and other temporary array(s) */ 2201*5f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2202*5f44c854SHong Zhang ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 2203*5f44c854SHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2204*5f44c854SHong Zhang ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 2205*5f44c854SHong Zhang 2206*5f44c854SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 2207*5f44c854SHong Zhang 2208*5f44c854SHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 2209*5f44c854SHong Zhang 2210*5f44c854SHong Zhang b = (Mat_SeqSBAIJ*)(fact)->data; 2211*5f44c854SHong Zhang b->singlemalloc = PETSC_FALSE; 2212*5f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 2213*5f44c854SHong Zhang b->j = uj; 2214*5f44c854SHong Zhang b->i = ui; 2215*5f44c854SHong Zhang b->diag = udiag; 2216*5f44c854SHong Zhang b->ilen = 0; 2217*5f44c854SHong Zhang b->imax = 0; 2218*5f44c854SHong Zhang b->row = perm; 2219*5f44c854SHong Zhang b->col = perm; 2220*5f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2221*5f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2222*5f44c854SHong Zhang b->icol = iperm; 2223*5f44c854SHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 2224*5f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2225*5f44c854SHong Zhang ierr = PetscLogObjectMemory((fact),(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 2226*5f44c854SHong Zhang b->maxnz = b->nz = ui[am]; 2227*5f44c854SHong Zhang b->free_a = PETSC_TRUE; 2228*5f44c854SHong Zhang b->free_ij = PETSC_TRUE; 2229*5f44c854SHong Zhang 2230*5f44c854SHong Zhang (fact)->info.factor_mallocs = reallocs; 2231*5f44c854SHong Zhang (fact)->info.fill_ratio_given = fill; 2232*5f44c854SHong Zhang if (ai[am] != 0) { 2233*5f44c854SHong Zhang (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 2234*5f44c854SHong Zhang } else { 2235*5f44c854SHong Zhang (fact)->info.fill_ratio_needed = 0.0; 2236*5f44c854SHong Zhang } 2237*5f44c854SHong Zhang (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_newdatastruct; 2238*5f44c854SHong Zhang PetscFunctionReturn(0); 2239*5f44c854SHong Zhang } 2240*5f44c854SHong Zhang 2241*5f44c854SHong Zhang #undef __FUNCT__ 2242*5f44c854SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ" 2243*5f44c854SHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2244*5f44c854SHong Zhang { 2245*5f44c854SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2246*5f44c854SHong Zhang Mat_SeqSBAIJ *b; 2247*5f44c854SHong Zhang PetscErrorCode ierr; 2248*5f44c854SHong Zhang PetscTruth perm_identity,missing; 2249*5f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 2250*5f44c854SHong Zhang const PetscInt *rip,*riip; 2251*5f44c854SHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 2252*5f44c854SHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 2253*5f44c854SHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 2254*5f44c854SHong Zhang PetscReal fill=info->fill,levels=info->levels; 2255*5f44c854SHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 2256*5f44c854SHong Zhang PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 2257*5f44c854SHong Zhang PetscBT lnkbt; 2258*5f44c854SHong Zhang IS iperm; 2259*5f44c854SHong Zhang PetscTruth newdatastruct=PETSC_FALSE; 2260*5f44c854SHong Zhang 2261*5f44c854SHong Zhang PetscFunctionBegin; 2262*5f44c854SHong Zhang ierr = PetscOptionsGetTruth(PETSC_NULL,"-icc_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr); 2263*5f44c854SHong Zhang if(newdatastruct){ 2264*5f44c854SHong Zhang ierr = MatICCFactorSymbolic_SeqAIJ_newdatastruct(fact,A,perm,info);CHKERRQ(ierr); 2265*5f44c854SHong Zhang PetscFunctionReturn(0); 2266*5f44c854SHong Zhang } 2267*5f44c854SHong Zhang 2268*5f44c854SHong 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); 2269*5f44c854SHong Zhang ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 2270*5f44c854SHong Zhang if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 2271*5f44c854SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 2272*5f44c854SHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 2273*5f44c854SHong Zhang 2274*5f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 2275*5f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 2276*5f44c854SHong Zhang ui[0] = 0; 2277*5f44c854SHong Zhang 2278*5f44c854SHong Zhang /* ICC(0) without matrix ordering: simply copies fill pattern */ 2279*5f44c854SHong Zhang if (!levels && perm_identity) { 2280*5f44c854SHong Zhang 2281*5f44c854SHong Zhang for (i=0; i<am; i++) { 2282*5f44c854SHong Zhang ui[i+1] = ui[i] + ai[i+1] - a->diag[i]; 2283*5f44c854SHong Zhang udiag[i] = ui[i]; 2284ed59401aSHong Zhang } 228539e3d630SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 228639e3d630SHong Zhang cols = uj; 2287ed59401aSHong Zhang for (i=0; i<am; i++) { 2288ed59401aSHong Zhang aj = a->j + a->diag[i]; 228939e3d630SHong Zhang ncols = ui[i+1] - ui[i]; 229039e3d630SHong Zhang for (j=0; j<ncols; j++) *cols++ = *aj++; 2291ed59401aSHong Zhang } 229239e3d630SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 2293b635d36bSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 2294b635d36bSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 2295b635d36bSHong Zhang 22967a48dd6fSHong Zhang /* initialization */ 22975a8e39fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 22987a48dd6fSHong Zhang 22997a48dd6fSHong Zhang /* jl: linked list for storing indices of the pivot rows 23007a48dd6fSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 23011393bc97SHong Zhang ierr = PetscMalloc((2*am+1)*sizeof(PetscInt)+2*am*sizeof(PetscInt**),&jl);CHKERRQ(ierr); 23027a48dd6fSHong Zhang il = jl + am; 23037a48dd6fSHong Zhang uj_ptr = (PetscInt**)(il + am); 23047a48dd6fSHong Zhang uj_lvl_ptr = (PetscInt**)(uj_ptr + am); 23057a48dd6fSHong Zhang for (i=0; i<am; i++){ 23067a48dd6fSHong Zhang jl[i] = am; il[i] = 0; 23077a48dd6fSHong Zhang } 23087a48dd6fSHong Zhang 23097a48dd6fSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 23107a48dd6fSHong Zhang nlnk = am + 1; 23112ed133dbSHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 23127a48dd6fSHong Zhang 23137a48dd6fSHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 2314a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 23157a48dd6fSHong Zhang current_space = free_space; 2316a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr); 23177a48dd6fSHong Zhang current_space_lvl = free_space_lvl; 23187a48dd6fSHong Zhang 23197a48dd6fSHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 23207a48dd6fSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 23217a48dd6fSHong Zhang nzk = 0; 2322622e2028SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 23233b4a8b6dSBarry 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); 2324622e2028SHong Zhang ncols_upper = 0; 2325622e2028SHong Zhang for (j=0; j<ncols; j++){ 2326b635d36bSHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 2327b635d36bSHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 23285a8e39fbSHong Zhang ajtmp[ncols_upper] = i; 2329622e2028SHong Zhang ncols_upper++; 2330622e2028SHong Zhang } 2331622e2028SHong Zhang } 2332b635d36bSHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 23337a48dd6fSHong Zhang nzk += nlnk; 23347a48dd6fSHong Zhang 23357a48dd6fSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 23367a48dd6fSHong Zhang prow = jl[k]; /* 1st pivot row */ 23377a48dd6fSHong Zhang 23387a48dd6fSHong Zhang while (prow < k){ 23397a48dd6fSHong Zhang nextprow = jl[prow]; 23407a48dd6fSHong Zhang 23417a48dd6fSHong Zhang /* merge prow into k-th row */ 23427a48dd6fSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 23437a48dd6fSHong Zhang jmax = ui[prow+1]; 23447a48dd6fSHong Zhang ncols = jmax-jmin; 2345ed59401aSHong Zhang i = jmin - ui[prow]; 2346ed59401aSHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 23475a8e39fbSHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 23485a8e39fbSHong Zhang j = *(uj - 1); 23495a8e39fbSHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 23507a48dd6fSHong Zhang nzk += nlnk; 23517a48dd6fSHong Zhang 23527a48dd6fSHong Zhang /* update il and jl for prow */ 23537a48dd6fSHong Zhang if (jmin < jmax){ 23547a48dd6fSHong Zhang il[prow] = jmin; 23557a48dd6fSHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 23567a48dd6fSHong Zhang } 23577a48dd6fSHong Zhang prow = nextprow; 23587a48dd6fSHong Zhang } 23597a48dd6fSHong Zhang 23607a48dd6fSHong Zhang /* if free space is not available, make more free space */ 23617a48dd6fSHong Zhang if (current_space->local_remaining<nzk) { 23627a48dd6fSHong Zhang i = am - k + 1; /* num of unfactored rows */ 23637a48dd6fSHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 2364a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 2365a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 23667a48dd6fSHong Zhang reallocs++; 23677a48dd6fSHong Zhang } 23687a48dd6fSHong Zhang 23697a48dd6fSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 2370b635d36bSHong Zhang if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 23712ed133dbSHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 23727a48dd6fSHong Zhang 23737a48dd6fSHong Zhang /* add the k-th row into il and jl */ 237439e3d630SHong Zhang if (nzk > 1){ 23757a48dd6fSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 23767a48dd6fSHong Zhang jl[k] = jl[i]; jl[i] = k; 23777a48dd6fSHong Zhang il[k] = ui[k] + 1; 23787a48dd6fSHong Zhang } 23797a48dd6fSHong Zhang uj_ptr[k] = current_space->array; 23807a48dd6fSHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 23817a48dd6fSHong Zhang 23827a48dd6fSHong Zhang current_space->array += nzk; 23837a48dd6fSHong Zhang current_space->local_used += nzk; 23847a48dd6fSHong Zhang current_space->local_remaining -= nzk; 23857a48dd6fSHong Zhang 23867a48dd6fSHong Zhang current_space_lvl->array += nzk; 23877a48dd6fSHong Zhang current_space_lvl->local_used += nzk; 23887a48dd6fSHong Zhang current_space_lvl->local_remaining -= nzk; 23897a48dd6fSHong Zhang 23907a48dd6fSHong Zhang ui[k+1] = ui[k] + nzk; 23917a48dd6fSHong Zhang } 23927a48dd6fSHong Zhang 23936cf91177SBarry Smith #if defined(PETSC_USE_INFO) 23947a48dd6fSHong Zhang if (ai[am] != 0) { 239539e3d630SHong Zhang PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]); 2396ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 2397ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2398ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 23997a48dd6fSHong Zhang } else { 2400ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 24017a48dd6fSHong Zhang } 240263ba0a88SBarry Smith #endif 24037a48dd6fSHong Zhang 24047a48dd6fSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 2405b635d36bSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 24067a48dd6fSHong Zhang ierr = PetscFree(jl);CHKERRQ(ierr); 24075a8e39fbSHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 24087a48dd6fSHong Zhang 24097a48dd6fSHong Zhang /* destroy list of free space and other temporary array(s) */ 24107a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2411a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 24122ed133dbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2413a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 24147a48dd6fSHong Zhang 241539e3d630SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 241639e3d630SHong Zhang 24177a48dd6fSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 24187a48dd6fSHong Zhang 2419719d5645SBarry Smith b = (Mat_SeqSBAIJ*)(fact)->data; 24207a48dd6fSHong Zhang b->singlemalloc = PETSC_FALSE; 24217a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 24227a48dd6fSHong Zhang b->j = uj; 24237a48dd6fSHong Zhang b->i = ui; 2424*5f44c854SHong Zhang b->diag = udiag; 24257a48dd6fSHong Zhang b->ilen = 0; 24267a48dd6fSHong Zhang b->imax = 0; 24277a48dd6fSHong Zhang b->row = perm; 2428b635d36bSHong Zhang b->col = perm; 2429b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2430b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2431b635d36bSHong Zhang b->icol = iperm; 24327a48dd6fSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 24337a48dd6fSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2434719d5645SBarry Smith ierr = PetscLogObjectMemory((fact),(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 24357a48dd6fSHong Zhang b->maxnz = b->nz = ui[am]; 2436e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2437e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 24387a48dd6fSHong Zhang 2439719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 2440719d5645SBarry Smith (fact)->info.fill_ratio_given = fill; 24417a48dd6fSHong Zhang if (ai[am] != 0) { 2442719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 24437a48dd6fSHong Zhang } else { 2444719d5645SBarry Smith (fact)->info.fill_ratio_needed = 0.0; 24457a48dd6fSHong Zhang } 2446719d5645SBarry Smith (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 2447a6175056SHong Zhang PetscFunctionReturn(0); 2448a6175056SHong Zhang } 2449d5d45c9bSBarry Smith 2450f76d2b81SHong Zhang #undef __FUNCT__ 2451f76d2b81SHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ" 24520481f469SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2453f76d2b81SHong Zhang { 2454f76d2b81SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 245510c27e3dSHong Zhang Mat_SeqSBAIJ *b; 2456dfbe8321SBarry Smith PetscErrorCode ierr; 2457f76d2b81SHong Zhang PetscTruth perm_identity; 245810c27e3dSHong Zhang PetscReal fill = info->fill; 24595d0c19d7SBarry Smith const PetscInt *rip,*riip; 24605d0c19d7SBarry Smith PetscInt i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 246110c27e3dSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 24622ed133dbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr; 2463a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 246410c27e3dSHong Zhang PetscBT lnkbt; 24652ed133dbSHong Zhang IS iperm; 2466f76d2b81SHong Zhang 2467f76d2b81SHong Zhang PetscFunctionBegin; 2468d0f46423SBarry 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); 24692ed133dbSHong Zhang /* check whether perm is the identity mapping */ 2470f76d2b81SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 24712ed133dbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 24722ed133dbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 24739bfd6278SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 247410c27e3dSHong Zhang 247510c27e3dSHong Zhang /* initialization */ 24761393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 247710c27e3dSHong Zhang ui[0] = 0; 247810c27e3dSHong Zhang 247910c27e3dSHong Zhang /* jl: linked list for storing indices of the pivot rows 24801393bc97SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 24811393bc97SHong Zhang ierr = PetscMalloc((3*am+1)*sizeof(PetscInt)+am*sizeof(PetscInt**),&jl);CHKERRQ(ierr); 24821393bc97SHong Zhang il = jl + am; 24831393bc97SHong Zhang cols = il + am; 24841393bc97SHong Zhang ui_ptr = (PetscInt**)(cols + am); 24851393bc97SHong Zhang for (i=0; i<am; i++){ 24861393bc97SHong Zhang jl[i] = am; il[i] = 0; 2487f76d2b81SHong Zhang } 248810c27e3dSHong Zhang 248910c27e3dSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 24901393bc97SHong Zhang nlnk = am + 1; 24911393bc97SHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 249210c27e3dSHong Zhang 24931393bc97SHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 2494a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 249510c27e3dSHong Zhang current_space = free_space; 249610c27e3dSHong Zhang 24971393bc97SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 249810c27e3dSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 249910c27e3dSHong Zhang nzk = 0; 25002ed133dbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 25013b4a8b6dSBarry 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); 25022ed133dbSHong Zhang ncols_upper = 0; 25032ed133dbSHong Zhang for (j=0; j<ncols; j++){ 25049bfd6278SHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 25052ed133dbSHong Zhang if (i >= k){ /* only take upper triangular entry */ 25062ed133dbSHong Zhang cols[ncols_upper] = i; 25072ed133dbSHong Zhang ncols_upper++; 25082ed133dbSHong Zhang } 25092ed133dbSHong Zhang } 25101393bc97SHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 251110c27e3dSHong Zhang nzk += nlnk; 251210c27e3dSHong Zhang 251310c27e3dSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 251410c27e3dSHong Zhang prow = jl[k]; /* 1st pivot row */ 251510c27e3dSHong Zhang 251610c27e3dSHong Zhang while (prow < k){ 251710c27e3dSHong Zhang nextprow = jl[prow]; 251810c27e3dSHong Zhang /* merge prow into k-th row */ 25191393bc97SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 252010c27e3dSHong Zhang jmax = ui[prow+1]; 252110c27e3dSHong Zhang ncols = jmax-jmin; 25221393bc97SHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 25235a8e39fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 252410c27e3dSHong Zhang nzk += nlnk; 252510c27e3dSHong Zhang 252610c27e3dSHong Zhang /* update il and jl for prow */ 252710c27e3dSHong Zhang if (jmin < jmax){ 252810c27e3dSHong Zhang il[prow] = jmin; 25292ed133dbSHong Zhang j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow; 253010c27e3dSHong Zhang } 253110c27e3dSHong Zhang prow = nextprow; 253210c27e3dSHong Zhang } 253310c27e3dSHong Zhang 253410c27e3dSHong Zhang /* if free space is not available, make more free space */ 253510c27e3dSHong Zhang if (current_space->local_remaining<nzk) { 25361393bc97SHong Zhang i = am - k + 1; /* num of unfactored rows */ 253710c27e3dSHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 2538a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 253910c27e3dSHong Zhang reallocs++; 254010c27e3dSHong Zhang } 254110c27e3dSHong Zhang 254210c27e3dSHong Zhang /* copy data into free space, then initialize lnk */ 25431393bc97SHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 254410c27e3dSHong Zhang 254510c27e3dSHong Zhang /* add the k-th row into il and jl */ 254610c27e3dSHong Zhang if (nzk-1 > 0){ 25471393bc97SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 254810c27e3dSHong Zhang jl[k] = jl[i]; jl[i] = k; 254910c27e3dSHong Zhang il[k] = ui[k] + 1; 255010c27e3dSHong Zhang } 25512ed133dbSHong Zhang ui_ptr[k] = current_space->array; 255210c27e3dSHong Zhang current_space->array += nzk; 255310c27e3dSHong Zhang current_space->local_used += nzk; 255410c27e3dSHong Zhang current_space->local_remaining -= nzk; 255510c27e3dSHong Zhang 255610c27e3dSHong Zhang ui[k+1] = ui[k] + nzk; 255710c27e3dSHong Zhang } 255810c27e3dSHong Zhang 25596cf91177SBarry Smith #if defined(PETSC_USE_INFO) 25601393bc97SHong Zhang if (ai[am] != 0) { 25611393bc97SHong Zhang PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]); 2562ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 2563ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2564ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 256510c27e3dSHong Zhang } else { 2566ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 256710c27e3dSHong Zhang } 256863ba0a88SBarry Smith #endif 256910c27e3dSHong Zhang 257010c27e3dSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 25719bfd6278SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 257210c27e3dSHong Zhang ierr = PetscFree(jl);CHKERRQ(ierr); 257310c27e3dSHong Zhang 257410c27e3dSHong Zhang /* destroy list of free space and other temporary array(s) */ 25751393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2576a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 257710c27e3dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 257810c27e3dSHong Zhang 257910c27e3dSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 258010c27e3dSHong Zhang 2581719d5645SBarry Smith b = (Mat_SeqSBAIJ*)(fact)->data; 258210c27e3dSHong Zhang b->singlemalloc = PETSC_FALSE; 2583e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2584e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 25851393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 258610c27e3dSHong Zhang b->j = uj; 258710c27e3dSHong Zhang b->i = ui; 258810c27e3dSHong Zhang b->diag = 0; 258910c27e3dSHong Zhang b->ilen = 0; 259010c27e3dSHong Zhang b->imax = 0; 259110c27e3dSHong Zhang b->row = perm; 25929bfd6278SHong Zhang b->col = perm; 25939bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 25949bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 25959bfd6278SHong Zhang b->icol = iperm; 259610c27e3dSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 25971393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2598719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 25991393bc97SHong Zhang b->maxnz = b->nz = ui[am]; 260010c27e3dSHong Zhang 2601719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 2602719d5645SBarry Smith (fact)->info.fill_ratio_given = fill; 26031393bc97SHong Zhang if (ai[am] != 0) { 2604719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 260510c27e3dSHong Zhang } else { 2606719d5645SBarry Smith (fact)->info.fill_ratio_needed = 0.0; 260710c27e3dSHong Zhang } 2608719d5645SBarry Smith (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 2609f76d2b81SHong Zhang PetscFunctionReturn(0); 2610f76d2b81SHong Zhang } 2611599ef60dSHong Zhang 2612599ef60dSHong Zhang #undef __FUNCT__ 2613894a2336SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_newdatastruct" 2614894a2336SHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_newdatastruct(Mat A,Vec bb,Vec xx) 2615599ef60dSHong Zhang { 26161d098868SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 26171d098868SHong Zhang PetscErrorCode ierr; 26181d098868SHong Zhang PetscInt n = A->rmap->n; 26198fc3a347SHong Zhang const PetscInt *ai = a->i,*aj = a->j,*vi; 26201d098868SHong Zhang PetscScalar *x,sum; 26211d098868SHong Zhang const PetscScalar *b; 26221d098868SHong Zhang const MatScalar *aa = a->a,*v; 26238fc3a347SHong Zhang PetscInt i,nz; 26241d098868SHong Zhang 26251d098868SHong Zhang PetscFunctionBegin; 26261d098868SHong Zhang if (!n) PetscFunctionReturn(0); 26271d098868SHong Zhang 26281d098868SHong Zhang ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 26291d098868SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 26301d098868SHong Zhang 26311d098868SHong Zhang /* forward solve the lower triangular */ 26321d098868SHong Zhang x[0] = b[0]; 26331d098868SHong Zhang v = aa; 26341d098868SHong Zhang vi = aj; 26351d098868SHong Zhang for (i=1; i<n; i++) { 26361d098868SHong Zhang nz = ai[i+1] - ai[i]; 26371d098868SHong Zhang sum = b[i]; 2638003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 2639e09e08ccSBarry Smith /* while (nz--) sum -= *v++ * x[*vi++];*/ 2640e0e43300SBarry Smith v += nz; 2641e0e43300SBarry Smith vi += nz; 26421d098868SHong Zhang x[i] = sum; 26431d098868SHong Zhang } 26441d098868SHong Zhang 26451d098868SHong Zhang /* backward solve the upper triangular */ 26468fc3a347SHong Zhang v = aa + ai[n+1]; 26478fc3a347SHong Zhang vi = aj + ai[n+1]; 26481d098868SHong Zhang for (i=n-1; i>=0; i--){ 26498fc3a347SHong Zhang nz = ai[2*n-i +1] - ai[2*n-i]-1; 26501d098868SHong Zhang sum = x[i]; 2651003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 2652e09e08ccSBarry Smith /* while (nz--) sum -= *v++ * x[*vi++]; */ 2653e0e43300SBarry Smith v += nz; 26548fc3a347SHong Zhang vi += nz; vi++; 2655fe31b70dSHong Zhang x[i] = *v++ *sum; /* x[i]=aa[adiag[i]]*sum; v++; */ 26561d098868SHong Zhang } 26571d098868SHong Zhang 26581d098868SHong Zhang ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 26591d098868SHong Zhang ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 26601d098868SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 26611d098868SHong Zhang PetscFunctionReturn(0); 26621d098868SHong Zhang } 26631d098868SHong Zhang 26641d098868SHong Zhang #undef __FUNCT__ 2665894a2336SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_newdatastruct" 2666894a2336SHong Zhang PetscErrorCode MatSolve_SeqAIJ_newdatastruct(Mat A,Vec bb,Vec xx) 26671d098868SHong Zhang { 26681d098868SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 26691d098868SHong Zhang IS iscol = a->col,isrow = a->row; 26701d098868SHong Zhang PetscErrorCode ierr; 2671894a2336SHong Zhang PetscInt i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,nz,k; 26721d098868SHong Zhang const PetscInt *rout,*cout,*r,*c; 2673894a2336SHong Zhang PetscScalar *x,*tmp,*tmps,sum; 26741d098868SHong Zhang const PetscScalar *b; 26751d098868SHong Zhang const MatScalar *aa = a->a,*v; 26761d098868SHong Zhang 26771d098868SHong Zhang PetscFunctionBegin; 26781d098868SHong Zhang if (!n) PetscFunctionReturn(0); 26791d098868SHong Zhang 26801d098868SHong Zhang ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 26811d098868SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 26821d098868SHong Zhang tmp = a->solve_work; 26831d098868SHong Zhang 26841d098868SHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 2685894a2336SHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 26861d098868SHong Zhang 26871d098868SHong Zhang /* forward solve the lower triangular */ 26881d098868SHong Zhang tmp[0] = b[*r++]; 26891d098868SHong Zhang tmps = tmp; 26901d098868SHong Zhang v = aa; 26911d098868SHong Zhang vi = aj; 26921d098868SHong Zhang for (i=1; i<n; i++) { 26931d098868SHong Zhang nz = ai[i+1] - ai[i]; 2694894a2336SHong Zhang sum = b[*r++]; 2695894a2336SHong Zhang PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 2696894a2336SHong Zhang tmp[i] = sum; 26971d098868SHong Zhang v += nz; vi += nz; 26981d098868SHong Zhang } 26991d098868SHong Zhang 27001d098868SHong Zhang /* backward solve the upper triangular */ 2701894a2336SHong Zhang k = n+1; 2702894a2336SHong Zhang v = aa + ai[k]; /* 1st entry of U(n-1,:) */ 2703894a2336SHong Zhang vi = aj + ai[k]; 27041d098868SHong Zhang for (i=n-1; i>=0; i--){ 2705894a2336SHong Zhang k = 2*n-i; 2706894a2336SHong Zhang nz = ai[k +1] - ai[k] - 1; 2707894a2336SHong Zhang sum = tmp[i]; 2708894a2336SHong Zhang PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 2709894a2336SHong Zhang x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */ 27101d098868SHong Zhang v += nz+1; vi += nz+1; 27111d098868SHong Zhang } 27121d098868SHong Zhang 27131d098868SHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 27141d098868SHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 27151d098868SHong Zhang ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 27161d098868SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 27171d098868SHong Zhang ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr); 27181d098868SHong Zhang PetscFunctionReturn(0); 27191d098868SHong Zhang } 27201d098868SHong Zhang 27211d098868SHong Zhang #undef __FUNCT__ 27221d098868SHong Zhang #define __FUNCT__ "MatILUDTFactor_SeqAIJ" 27231d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact) 27241d098868SHong Zhang { 27251d098868SHong Zhang Mat B = *fact; 2726599ef60dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b; 2727599ef60dSHong Zhang IS isicol; 2728599ef60dSHong Zhang PetscErrorCode ierr; 27291d098868SHong Zhang const PetscInt *r,*ic; 27301d098868SHong Zhang PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag; 2731dc3a2fd3SHong Zhang PetscInt *bi,*bj,*bdiag,*bdiag_rev; 2732f61a948aSLisandro Dalcin PetscInt row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au; 27331d098868SHong Zhang PetscInt nlnk,*lnk; 27341d098868SHong Zhang PetscBT lnkbt; 27351d098868SHong Zhang PetscTruth row_identity,icol_identity,both_identity; 27368aee2decSHong Zhang MatScalar *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp; 27371d098868SHong Zhang const PetscInt *ics; 27381d098868SHong Zhang PetscInt j,nz,*pj,*bjtmp,k,ncut,*jtmp; 2739f61a948aSLisandro Dalcin PetscReal dt=info->dt,dtcol=info->dtcol,shift=info->shiftinblocks; 2740f61a948aSLisandro Dalcin PetscInt dtcount=(PetscInt)info->dtcount,nnz_max; 2741599ef60dSHong Zhang PetscTruth missing; 2742599ef60dSHong Zhang 2743599ef60dSHong Zhang PetscFunctionBegin; 2744f61a948aSLisandro Dalcin 2745f61a948aSLisandro Dalcin if (dt == PETSC_DEFAULT) dt = 0.005; 2746f61a948aSLisandro Dalcin if (dtcol == PETSC_DEFAULT) dtcol = 0.01; /* XXX unused! */ 2747f61a948aSLisandro Dalcin if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax); 2748f61a948aSLisandro Dalcin 27491d098868SHong Zhang /* ------- symbolic factorization, can be reused ---------*/ 27501d098868SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 27511d098868SHong Zhang if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 27521d098868SHong Zhang adiag=a->diag; 2753599ef60dSHong Zhang 2754599ef60dSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 2755599ef60dSHong Zhang 2756599ef60dSHong Zhang /* bdiag is location of diagonal in factor */ 2757a6226a1aSHong Zhang ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 2758a6226a1aSHong Zhang bdiag_rev = bdiag + n+1; 2759599ef60dSHong Zhang 27601d098868SHong Zhang /* allocate row pointers bi */ 27618fc3a347SHong Zhang ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 27621d098868SHong Zhang 27631d098868SHong Zhang /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */ 2764393d3a68SHong Zhang if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */ 27651d098868SHong Zhang nnz_max = ai[n]+2*n*dtcount+2; 27668fc3a347SHong Zhang 27678fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 27688fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(MatScalar),&ba);CHKERRQ(ierr); 2769599ef60dSHong Zhang 2770599ef60dSHong Zhang /* put together the new matrix */ 2771599ef60dSHong Zhang ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 2772599ef60dSHong Zhang ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 2773599ef60dSHong Zhang b = (Mat_SeqAIJ*)(B)->data; 2774599ef60dSHong Zhang b->free_a = PETSC_TRUE; 2775599ef60dSHong Zhang b->free_ij = PETSC_TRUE; 2776599ef60dSHong Zhang b->singlemalloc = PETSC_FALSE; 2777599ef60dSHong Zhang b->a = ba; 2778599ef60dSHong Zhang b->j = bj; 2779599ef60dSHong Zhang b->i = bi; 2780599ef60dSHong Zhang b->diag = bdiag; 2781599ef60dSHong Zhang b->ilen = 0; 2782599ef60dSHong Zhang b->imax = 0; 2783599ef60dSHong Zhang b->row = isrow; 2784599ef60dSHong Zhang b->col = iscol; 2785599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 2786599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 2787599ef60dSHong Zhang b->icol = isicol; 2788599ef60dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2789599ef60dSHong Zhang 27901d098868SHong Zhang ierr = PetscLogObjectMemory(B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 27911d098868SHong Zhang b->maxnz = nnz_max; 2792599ef60dSHong Zhang 2793599ef60dSHong Zhang (B)->factor = MAT_FACTOR_ILUDT; 2794599ef60dSHong Zhang (B)->info.factor_mallocs = 0; 27951d098868SHong Zhang (B)->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]); 27961d098868SHong Zhang CHKMEMQ; 27971d098868SHong Zhang /* ------- end of symbolic factorization ---------*/ 2798599ef60dSHong Zhang 2799599ef60dSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 2800599ef60dSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 2801599ef60dSHong Zhang ics = ic; 2802599ef60dSHong Zhang 2803599ef60dSHong Zhang /* linked list for storing column indices of the active row */ 2804599ef60dSHong Zhang nlnk = n + 1; 2805599ef60dSHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28061d098868SHong Zhang 28071d098868SHong Zhang /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */ 28081d098868SHong Zhang ierr = PetscMalloc((2*n+1)*sizeof(PetscInt),&im);CHKERRQ(ierr); 28091d098868SHong Zhang jtmp = im + n; 28101d098868SHong Zhang /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */ 28118369b28dSHong Zhang ierr = PetscMalloc((2*n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 28128aee2decSHong Zhang ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 28138369b28dSHong Zhang vtmp = rtmp + n; 2814599ef60dSHong Zhang 2815599ef60dSHong Zhang bi[0] = 0; 28168fc3a347SHong Zhang bdiag[0] = nnz_max-1; /* location of diag[0] in factor B */ 2817dc3a2fd3SHong Zhang bdiag_rev[n] = bdiag[0]; 28188fc3a347SHong Zhang bi[2*n+1] = bdiag[0]+1; /* endof bj and ba array */ 2819599ef60dSHong Zhang for (i=0; i<n; i++) { 2820599ef60dSHong Zhang /* copy initial fill into linked list */ 2821599ef60dSHong Zhang nzi = 0; /* nonzeros for active row i */ 28228369b28dSHong Zhang nzi = ai[r[i]+1] - ai[r[i]]; 28238369b28dSHong 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); 28248369b28dSHong Zhang nzi_al = adiag[r[i]] - ai[r[i]]; 28258369b28dSHong Zhang nzi_au = ai[r[i]+1] - adiag[r[i]] -1; 2826599ef60dSHong Zhang ajtmp = aj + ai[r[i]]; 28278369b28dSHong Zhang ierr = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 2828599ef60dSHong Zhang 2829599ef60dSHong Zhang /* load in initial (unfactored row) */ 28301d098868SHong Zhang aatmp = a->a + ai[r[i]]; 28318369b28dSHong Zhang for (j=0; j<nzi; j++) { 28321d098868SHong Zhang rtmp[ics[*ajtmp++]] = *aatmp++; 2833599ef60dSHong Zhang } 2834599ef60dSHong Zhang 2835599ef60dSHong Zhang /* add pivot rows into linked list */ 2836599ef60dSHong Zhang row = lnk[n]; 2837599ef60dSHong Zhang while (row < i ) { 28381d098868SHong Zhang nzi_bl = bi[row+1] - bi[row] + 1; 28391d098868SHong Zhang bjtmp = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */ 28401d098868SHong Zhang ierr = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr); 2841599ef60dSHong Zhang nzi += nlnk; 2842599ef60dSHong Zhang row = lnk[row]; 2843599ef60dSHong Zhang } 2844599ef60dSHong Zhang 28458369b28dSHong Zhang /* copy data from lnk into jtmp, then initialize lnk */ 28468369b28dSHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr); 2847599ef60dSHong Zhang 2848599ef60dSHong Zhang /* numerical factorization */ 28498369b28dSHong Zhang bjtmp = jtmp; 2850599ef60dSHong Zhang row = *bjtmp++; /* 1st pivot row */ 2851599ef60dSHong Zhang while ( row < i ) { 2852599ef60dSHong Zhang pc = rtmp + row; 28533c2a7987SHong Zhang pv = ba + bdiag[row]; /* 1./(diag of the pivot row) */ 28543c2a7987SHong Zhang multiplier = (*pc) * (*pv); 2855599ef60dSHong Zhang *pc = multiplier; 28563c2a7987SHong Zhang if (PetscAbsScalar(*pc) > dt){ /* apply tolerance dropping rule */ 28571d098868SHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 28581d098868SHong Zhang pv = ba + bdiag[row+1] + 1; 28591d098868SHong Zhang /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */ 28601d098868SHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 28611d098868SHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 2862dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 2863599ef60dSHong Zhang } 2864599ef60dSHong Zhang row = *bjtmp++; 2865599ef60dSHong Zhang } 2866599ef60dSHong Zhang 28678369b28dSHong Zhang /* copy sparse rtmp into contiguous vtmp; separate L and U part */ 28688aee2decSHong Zhang diag_tmp = rtmp[i]; /* save diagonal value - may not needed?? */ 28698369b28dSHong Zhang nzi_bl = 0; j = 0; 28708369b28dSHong Zhang while (jtmp[j] < i){ /* Note: jtmp is sorted */ 28718aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 28728369b28dSHong Zhang nzi_bl++; j++; 28738369b28dSHong Zhang } 28748369b28dSHong Zhang nzi_bu = nzi - nzi_bl -1; 28758369b28dSHong Zhang while (j < nzi){ 28768aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 28778369b28dSHong Zhang j++; 28788369b28dSHong Zhang } 28798369b28dSHong Zhang 2880599ef60dSHong Zhang bjtmp = bj + bi[i]; 2881599ef60dSHong Zhang batmp = ba + bi[i]; 28828369b28dSHong Zhang /* apply level dropping rule to L part */ 28838369b28dSHong Zhang ncut = nzi_al + dtcount; 28848369b28dSHong Zhang if (ncut < nzi_bl){ 28858369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr); 28868369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr); 2887599ef60dSHong Zhang } else { 28888369b28dSHong Zhang ncut = nzi_bl; 28898369b28dSHong Zhang } 28908369b28dSHong Zhang for (j=0; j<ncut; j++){ 28918369b28dSHong Zhang bjtmp[j] = jtmp[j]; 28928369b28dSHong Zhang batmp[j] = vtmp[j]; 28936da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */ 28948369b28dSHong Zhang } 28951d098868SHong Zhang bi[i+1] = bi[i] + ncut; 28968369b28dSHong Zhang nzi = ncut + 1; 28978369b28dSHong Zhang 28988369b28dSHong Zhang /* apply level dropping rule to U part */ 28998369b28dSHong Zhang ncut = nzi_au + dtcount; 29008369b28dSHong Zhang if (ncut < nzi_bu){ 29018369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr); 29028369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr); 29038369b28dSHong Zhang } else { 29048369b28dSHong Zhang ncut = nzi_bu; 29058369b28dSHong Zhang } 29068369b28dSHong Zhang nzi += ncut; 29071d098868SHong Zhang 29081d098868SHong Zhang /* mark bdiagonal */ 29091d098868SHong Zhang bdiag[i+1] = bdiag[i] - (ncut + 1); 2910dc3a2fd3SHong Zhang bdiag_rev[n-i-1] = bdiag[i+1]; 29118fc3a347SHong Zhang bi[2*n - i] = bi[2*n - i +1] - (ncut + 1); 29121d098868SHong Zhang bjtmp = bj + bdiag[i]; 29131d098868SHong Zhang batmp = ba + bdiag[i]; 29141d098868SHong Zhang *bjtmp = i; 29158aee2decSHong Zhang *batmp = diag_tmp; /* rtmp[i]; */ 2916c9c72f8fSHong Zhang if (*batmp == 0.0) { 2917c9c72f8fSHong Zhang *batmp = dt+shift; 2918bd1bc851SBarry Smith /* printf(" row %d add shift %g\n",i,shift); */ 2919c9c72f8fSHong Zhang } 2920b78a477dSHong Zhang *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */ 29216da515a1SHong Zhang /* printf(" (%d,%g),",*bjtmp,*batmp); */ 29221d098868SHong Zhang 29231d098868SHong Zhang bjtmp = bj + bdiag[i+1]+1; 29241d098868SHong Zhang batmp = ba + bdiag[i+1]+1; 29258369b28dSHong Zhang for (k=0; k<ncut; k++){ 29261d098868SHong Zhang bjtmp[k] = jtmp[nzi_bl+1+k]; 29271d098868SHong Zhang batmp[k] = vtmp[nzi_bl+1+k]; 29286da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */ 2929599ef60dSHong Zhang } 29306da515a1SHong Zhang /* printf("\n"); */ 2931599ef60dSHong Zhang 2932599ef60dSHong Zhang im[i] = nzi; /* used by PetscLLAddSortedLU() */ 2933599ef60dSHong Zhang /* 29341d098868SHong Zhang printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]); 29358369b28dSHong Zhang printf(" ----------------------------\n"); 2936599ef60dSHong Zhang */ 29378369b28dSHong Zhang } /* for (i=0; i<n; i++) */ 29381d098868SHong Zhang /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */ 29391d098868SHong 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]); 2940599ef60dSHong Zhang 2941599ef60dSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 2942599ef60dSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 2943599ef60dSHong Zhang 2944599ef60dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2945599ef60dSHong Zhang ierr = PetscFree(im);CHKERRQ(ierr); 2946599ef60dSHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 2947599ef60dSHong Zhang 2948599ef60dSHong Zhang ierr = PetscLogFlops(B->cmap->n);CHKERRQ(ierr); 29491d098868SHong Zhang b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n]; 2950599ef60dSHong Zhang 2951599ef60dSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 2952599ef60dSHong Zhang ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr); 2953599ef60dSHong Zhang both_identity = (PetscTruth) (row_identity && icol_identity); 2954599ef60dSHong Zhang if (row_identity && icol_identity) { 2955894a2336SHong Zhang B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct; 2956599ef60dSHong Zhang } else { 2957894a2336SHong Zhang B->ops->solve = MatSolve_SeqAIJ_newdatastruct; 2958599ef60dSHong Zhang } 2959599ef60dSHong Zhang 2960b78a477dSHong Zhang B->ops->lufactorsymbolic = MatILUDTFactorSymbolic_SeqAIJ; 2961b78a477dSHong Zhang B->ops->lufactornumeric = MatILUDTFactorNumeric_SeqAIJ; 29621d098868SHong Zhang B->ops->solveadd = 0; 29631d098868SHong Zhang B->ops->solvetranspose = 0; 29641d098868SHong Zhang B->ops->solvetransposeadd = 0; 29651d098868SHong Zhang B->ops->matsolve = 0; 2966599ef60dSHong Zhang B->assembled = PETSC_TRUE; 2967599ef60dSHong Zhang B->preallocated = PETSC_TRUE; 2968599ef60dSHong Zhang PetscFunctionReturn(0); 2969599ef60dSHong Zhang } 2970599ef60dSHong Zhang 29713c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */ 29723c2a7987SHong Zhang #undef __FUNCT__ 29733c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorSymbolic_SeqAIJ" 29743c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info) 29753c2a7987SHong Zhang { 29763c2a7987SHong Zhang PetscErrorCode ierr; 29773c2a7987SHong Zhang 29783c2a7987SHong Zhang PetscFunctionBegin; 29793c2a7987SHong Zhang ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr); 29803c2a7987SHong Zhang 29813c2a7987SHong Zhang fact->ops->lufactornumeric = MatILUDTFactorNumeric_SeqAIJ; 29823c2a7987SHong Zhang PetscFunctionReturn(0); 29833c2a7987SHong Zhang } 29843c2a7987SHong Zhang 29853c2a7987SHong Zhang /* 29863c2a7987SHong Zhang same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors 29873c2a7987SHong Zhang - intend to replace existing MatLUFactorNumeric_SeqAIJ() 29883c2a7987SHong Zhang */ 29893c2a7987SHong Zhang #undef __FUNCT__ 29903c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorNumeric_SeqAIJ" 29913c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info) 29923c2a7987SHong Zhang { 29933c2a7987SHong Zhang Mat C=fact; 29943c2a7987SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 29953c2a7987SHong Zhang IS isrow = b->row,isicol = b->icol; 29963c2a7987SHong Zhang PetscErrorCode ierr; 29973c2a7987SHong Zhang const PetscInt *r,*ic,*ics; 2998b78a477dSHong Zhang PetscInt i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 2999b78a477dSHong Zhang PetscInt *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj; 30003c2a7987SHong Zhang MatScalar *rtmp,*pc,multiplier,*v,*pv,*aa=a->a; 3001b78a477dSHong Zhang PetscReal dt=info->dt,shift=info->shiftinblocks; 3002b78a477dSHong Zhang PetscTruth row_identity, col_identity; 30033c2a7987SHong Zhang 30043c2a7987SHong Zhang PetscFunctionBegin; 30053c2a7987SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 30063c2a7987SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 3007b78a477dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 30083c2a7987SHong Zhang ics = ic; 30093c2a7987SHong Zhang 30103c2a7987SHong Zhang for (i=0; i<n; i++){ 3011b78a477dSHong Zhang /* initialize rtmp array */ 3012b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of nozeros in L(i,:) */ 30133c2a7987SHong Zhang bjtmp = bj + bi[i]; 3014b78a477dSHong Zhang for (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0; 3015b78a477dSHong Zhang rtmp[i] = 0.0; 3016b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */ 3017b78a477dSHong Zhang bjtmp = bj + bdiag[i+1] + 1; 3018b78a477dSHong Zhang for (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0; 30193c2a7987SHong Zhang 3020b78a477dSHong Zhang /* load in initial unfactored row of A */ 30216da515a1SHong Zhang /* printf("row %d\n",i); */ 30223c2a7987SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 30233c2a7987SHong Zhang ajtmp = aj + ai[r[i]]; 30243c2a7987SHong Zhang v = aa + ai[r[i]]; 30253c2a7987SHong Zhang for (j=0; j<nz; j++) { 3026b78a477dSHong Zhang rtmp[ics[*ajtmp++]] = v[j]; 30276da515a1SHong Zhang /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */ 30283c2a7987SHong Zhang } 30296da515a1SHong Zhang /* printf("\n"); */ 30303c2a7987SHong Zhang 3031b78a477dSHong Zhang /* numerical factorization */ 3032b78a477dSHong Zhang bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */ 3033b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of entries in L(i,:) */ 3034b78a477dSHong Zhang k = 0; 3035b78a477dSHong Zhang while (k < nzl){ 30363c2a7987SHong Zhang row = *bjtmp++; 30376da515a1SHong Zhang /* printf(" prow %d\n",row); */ 30383c2a7987SHong Zhang pc = rtmp + row; 3039b78a477dSHong Zhang pv = b->a + bdiag[row]; /* 1./(diag of the pivot row) */ 3040b78a477dSHong Zhang multiplier = (*pc) * (*pv); 30413c2a7987SHong Zhang *pc = multiplier; 3042b78a477dSHong Zhang if (PetscAbsScalar(multiplier) > dt){ 3043b78a477dSHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 3044b78a477dSHong Zhang pv = b->a + bdiag[row+1] + 1; 3045b78a477dSHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 3046b78a477dSHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 30476da515a1SHong Zhang /* ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); */ 30483c2a7987SHong Zhang } 3049b78a477dSHong Zhang k++; 30503c2a7987SHong Zhang } 30513c2a7987SHong Zhang 3052b78a477dSHong Zhang /* finished row so stick it into b->a */ 3053b78a477dSHong Zhang /* L-part */ 3054b78a477dSHong Zhang pv = b->a + bi[i] ; 3055b78a477dSHong Zhang pj = bj + bi[i] ; 3056b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; 3057b78a477dSHong Zhang for (j=0; j<nzl; j++) { 3058b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 30596da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 30603c2a7987SHong Zhang } 3061b78a477dSHong Zhang 3062b78a477dSHong Zhang /* diagonal: invert diagonal entries for simplier triangular solves */ 3063b78a477dSHong Zhang if (rtmp[i] == 0.0) rtmp[i] = dt+shift; 3064b78a477dSHong Zhang b->a[bdiag[i]] = 1.0/rtmp[i]; 30656da515a1SHong Zhang /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */ 3066b78a477dSHong Zhang 3067b78a477dSHong Zhang /* U-part */ 3068b78a477dSHong Zhang pv = b->a + bdiag[i+1] + 1; 3069b78a477dSHong Zhang pj = bj + bdiag[i+1] + 1; 3070b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1] - 1; 3071b78a477dSHong Zhang for (j=0; j<nzu; j++) { 3072b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 30736da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 3074b78a477dSHong Zhang } 30756da515a1SHong Zhang /* printf("\n"); */ 3076b78a477dSHong Zhang } 3077b78a477dSHong Zhang 30783c2a7987SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 30793c2a7987SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 30803c2a7987SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3081b78a477dSHong Zhang 30823c2a7987SHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 30833c2a7987SHong Zhang ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 30843c2a7987SHong Zhang if (row_identity && col_identity) { 3085894a2336SHong Zhang C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct; 30863c2a7987SHong Zhang } else { 3087894a2336SHong Zhang C->ops->solve = MatSolve_SeqAIJ_newdatastruct; 30883c2a7987SHong Zhang } 3089b78a477dSHong Zhang C->ops->solveadd = 0; 3090b78a477dSHong Zhang C->ops->solvetranspose = 0; 3091b78a477dSHong Zhang C->ops->solvetransposeadd = 0; 3092b78a477dSHong Zhang C->ops->matsolve = 0; 30933c2a7987SHong Zhang C->assembled = PETSC_TRUE; 30943c2a7987SHong Zhang C->preallocated = PETSC_TRUE; 30953c2a7987SHong Zhang ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 30963c2a7987SHong Zhang PetscFunctionReturn(0); 30973c2a7987SHong Zhang } 3098